fixed errors when providing docker container name in the parameter

server/docker/docker.go

@@ -93,11 +93,14 @@ func BuildRunContainer(NumPorts int, GPU string, ContainerName string) (*DockerV
 	}
 	RespDocker.ImagePath = config.DefaultDockerFile
 
-	if ContainerName != "" {
+	// We are checking if the container name is not nil and not equal to the default one used
+	// which is docker-ubuntu-sshd
+	if ContainerName != "" && ContainerName != "docker-ubuntu-sshd" {
 		Containers, err := ViewAllContainers()
 		if err != nil {
 			return nil,err
 		}
+
 		for _, dockerContainer := range Containers.DockerContainer {
 			if dockerContainer.ContainerName == ContainerName {
 				RespDocker.ImagePath = config.DockerContainers + ContainerName + "/"
@@ -311,7 +314,6 @@ func ViewAllContainers() (*DockerContainers, error){
 
 	for _, f := range folders {
 		if f.IsDir() {
-			fmt.Print(f.Name())
 			//Declare variable DockerContainer of type struct
             var Container DockerContainer
 

server/docker/docker_test.go

@@ -1,19 +1,24 @@
 package docker
 
 import (
-	"fmt"
 	"testing"
 )
 
 func TestDocker(t *testing.T) {
-	//TODO overwrite with custom docker paths
-	resp,err := BuildRunContainer(2,"true","")
+	// Testing by providing default container name
+	_,err := BuildRunContainer(2,"false","docker-ubuntu-sshd")
+
+	if err != nil {
+		t.Error(err)
+	}
+
+	// Testing if no container name is provided if default is used
+	_,err = BuildRunContainer(2,"false","")
 
 	if err != nil {
 		t.Error(err)
 	}
 
-	fmt.Print(resp.VNCPort)
 }
 
 func TestViewAllContainers(t *testing.T) {