package publish import ( "strings" "testing" "github.com/drone/drone/shared/build/buildfile" "github.com/drone/drone/shared/build/repo" "gopkg.in/yaml.v1" ) type PublishToDrone struct { Publish *Publish `yaml:"publish,omitempty"` } func setUpWithDrone(input string) (string, error) { var buildStruct PublishToDrone err := yaml.Unmarshal([]byte(input), &buildStruct) if err != nil { return "", err } bf := buildfile.New() buildStruct.Publish.Write(bf, &repo.Repo{Name: "name"}) return bf.String(), err } // DockerHost and version test (no auth) var dockerHostYaml = ` publish: docker: docker_host: tcp://server:1000 docker_version: 1.3.0 image_name: registry/image ` func TestDockerHost(t *testing.T) { response, err := setUpWithDrone(dockerHostYaml) t.Log(dockerHostYaml) if err != nil { t.Fatalf("Can't unmarshal script: %s\n\n", err.Error()) } expected := "export DOCKER_HOST=tcp://server:1000" if !strings.Contains(response, expected) { t.Fatalf("Response: " + response + " doesn't export correct " + "DOCKER_HOST envvar: expected " + expected + "\n\n") } expected = "https://get.docker.io/builds/Linux/x86_64/docker-1.3.0.tgz" if !strings.Contains(response, expected) { t.Fatalf("Response: " + response + " doesn't download from:" + expected + "\n\n") } } var dockerHostNoVersionYaml = ` publish: docker: docker_host: tcp://server:1000 image_name: registry/image ` func TestDockerHostNoVersion(t *testing.T) { response, err := setUpWithDrone(dockerHostNoVersionYaml) t.Log(dockerHostNoVersionYaml) if err != nil { t.Fatalf("Can't unmarshal script: %s\n\n", err.Error()) } expected := "export DOCKER_HOST=tcp://server:1000" if !strings.Contains(response, expected) { t.Fatalf("Response: " + response + " doesn't export correct " + "DOCKER_HOST envvar: expected " + expected + "\n\n") } download := "https://get.docker.io/builds/Linux/x86_64/docker-latest.tgz" if !strings.Contains(response, download) { t.Fatalf("Response: " + response + " doesn't download from:" + download + "\n\n") } } // Private Registry Test (no auth) var privateRegistryNoAuthYaml = ` publish: docker: dockerfile: file_path docker_host: tcp://server:1000 docker_version: 1.0 registry_login: false image_name: registry/image ` func TestPrivateRegistryNoAuth(t *testing.T) { response, err := setUpWithDrone(privateRegistryNoAuthYaml) t.Log(privateRegistryNoAuthYaml) if err != nil { t.Fatalf("Can't unmarshal script: %s\n\n", err.Error()) } if !strings.Contains(response, "docker build --pull -t registry/image:$(git rev-parse --short HEAD)") { t.Fatalf("Response: " + response + " doesn't contain registry in image-names: expected registry/image\n\n") } } // Private Registry Test (with auth) var privateRegistryAuthYaml = ` publish: docker: dockerfile: file_path docker_host: tcp://server:1000 docker_version: 1.0 registry_login_url: https://registry:8000/v1/ registry_login: true username: username password: password email: email@example.com image_name: registry/image ` func TestPrivateRegistryAuth(t *testing.T) { response, err := setUpWithDrone(privateRegistryAuthYaml) t.Log(privateRegistryAuthYaml) if err != nil { t.Fatalf("Can't unmarshal script: %s\n\n", err.Error()) } if !strings.Contains(response, "docker login -u username -p password -e email@example.com https://registry:8000/v1/") { t.Log("\n\n\n\ndocker login -u username -p xxxxxxxx -e email@example.com https://registry:8000/v1/\n\n\n\n") t.Fatalf("Response: " + response + " doesn't contain private registry login\n\n") } if !strings.Contains(response, "docker build --pull -t registry/image:$(git rev-parse --short HEAD) .") { t.Log("docker build --pull -t registry/image:$(git rev-parse --short HEAD) .") t.Fatalf("Response: " + response + " doesn't contain registry in image-names\n\n") } } // Keep builds Test var keepBuildsYaml = ` publish: docker: docker_host: tcp://server:1000 docker_version: 1.0 keep_build: true username: username password: password email: email@example.com image_name: image ` func TestKeepBuilds(t *testing.T) { response, err := setUpWithDrone(keepBuildsYaml) t.Log(keepBuildsYaml) if err != nil { t.Fatalf("Can't unmarshal script: %s\n\n", err.Error()) } if strings.Contains(response, "docker rmi") { t.Fatalf("Response: " + response + " incorrectly instructs the docker server to remove the builds when it shouldn't\n\n") } } // Custom Tag test var customTagYaml = ` publish: docker: docker_host: tcp://server:1000 docker_version: 1.0 tag: release-0.1 username: username password: password email: email@example.com image_name: username/image ` func TestSingleTag(t *testing.T) { response, err := setUpWithDrone(customTagYaml) t.Log(customTagYaml) if err != nil { t.Fatalf("Can't unmarshal script: %s\n", err.Error()) } if strings.Contains(response, "$(git rev-parse --short HEAD)") { t.Fatalf("Response: " + response + " is tagging images from git-refs when it should use a custom tag\n\n") } if !strings.Contains(response, "docker build --pull -t username/image:release-0.1") { t.Fatalf("Response: " + response + " isn't tagging images using our custom tag\n\n") } if !strings.Contains(response, "docker push username/image:release-0.1") { t.Fatalf("Response: " + response + " doesn't push the custom tagged image\n\n") } if !strings.Contains(response, "docker rmi username/image:release-0.1") { t.Fatalf("Response: " + response + " doesn't remove custom tagged image\n\n") } } var missingFieldsYaml = ` publish: docker: dockerfile: file ` var multipleTagsYaml = ` publish: docker: docker_host: tcp://server:1000 docker_version: 1.0 tags: [release-0.2, release-latest] username: username password: password email: email@example.com image_name: username/image ` func TestTagsNoSingle(t *testing.T) { response, err := setUpWithDrone(multipleTagsYaml) t.Log(multipleTagsYaml) if err != nil { t.Fatalf("Can't unmarshal script: %s\n", err.Error()) } if strings.Contains(response, "$(git rev-parse --short HEAD)") { t.Fatalf("Response: " + response + " is tagging images from git-refs when it should using custom tag\n\n") } if !strings.Contains(response, "docker build --pull -t username/image:release-0.2") { t.Fatalf("Response: " + response + " isn't tagging images using our first custom tag\n\n") } if !strings.Contains(response, "docker tag username/image:release-0.2 username/image:release-latest") { t.Fatalf("Response: " + response + " isn't tagging images using our second custom tag\n\n") } if !strings.Contains(response, "docker push username/image:release-0.2") { t.Fatalf("Response: " + response + " doesn't push the custom tagged image\n\n") } if !strings.Contains(response, "docker rmi username/image:release-0.2") { t.Fatalf("Response: " + response + " doesn't remove custom tagged image\n\n") } if !strings.Contains(response, "docker push username/image:release-latest") { t.Fatalf("Response: " + response + " doesn't push the second custom tagged image\n\n") } if !strings.Contains(response, "docker rmi username/image:release-latest") { t.Fatalf("Response: " + response + " doesn't remove second custom tagged image\n\n") } } var bothTagsYaml = ` publish: docker: docker_host: tcp://server:1000 docker_version: 1.0 tag: release-0.2 tags: [release-0.3, release-latest] username: username password: password email: email@example.com image_name: username/image ` func TestTagsWithSingle(t *testing.T) { response, err := setUpWithDrone(bothTagsYaml) t.Log(bothTagsYaml) if err != nil { t.Fatalf("Can't unmarshal script: %s\n", err.Error()) } if strings.Contains(response, "$(git rev-parse --short HEAD)") { t.Fatalf("Response: " + response + " is tagging images from git-refs when it should using custom tag\n\n") } if !strings.Contains(response, "docker build --pull -t username/image:release-0.3") { t.Fatalf("Response: " + response + " isn't tagging images using our first custom tag\n\n") } if !strings.Contains(response, "docker tag username/image:release-0.3 username/image:release-0.2") { t.Fatalf("Response: " + response + " isn't tagging images using our second custom tag\n\n") } if !strings.Contains(response, "docker tag username/image:release-0.3 username/image:release-latest") { t.Fatalf("Response: " + response + " isn't tagging images using our third custom tag\n\n") } if !strings.Contains(response, "docker push username/image:release-0.2") { t.Fatalf("Response: " + response + " doesn't push the custom tagged image\n\n") } if !strings.Contains(response, "docker rmi username/image:release-0.2") { t.Fatalf("Response: " + response + " doesn't remove custom tagged image\n\n") } if !strings.Contains(response, "docker push username/image:release-0.3") { t.Fatalf("Response: " + response + " doesn't push the custom tagged image\n\n") } if !strings.Contains(response, "docker rmi username/image:release-0.3") { t.Fatalf("Response: " + response + " doesn't remove custom tagged image\n\n") } if !strings.Contains(response, "docker push username/image:release-latest") { t.Fatalf("Response: " + response + " doesn't push the second custom tagged image\n\n") } if !strings.Contains(response, "docker rmi username/image:release-latest") { t.Fatalf("Response: " + response + " doesn't remove second custom tagged image\n\n") } } func TestMissingFields(t *testing.T) { response, err := setUpWithDrone(missingFieldsYaml) t.Log(missingFieldsYaml) if err != nil { t.Fatalf("Can't unmarshal script: %s\n\n", err.Error()) } if !strings.Contains(response, "Missing argument(s)") { t.Fatalf("Response: " + response + " didn't contain missing arguments warning\n\n") } } var validYaml = ` publish: docker: docker_file: file_path docker_host: tcp://server:1000 docker_version: 1.0 username: user password: password email: email image_name: user/image registry_login: true ` func TestValidYaml(t *testing.T) { response, err := setUpWithDrone(validYaml) t.Log(validYaml) if err != nil { t.Fatalf("Can't unmarshal script: %s\n\n", err.Error()) } if !strings.Contains(response, "docker build --pull -t user/image:$(git rev-parse --short HEAD) - <") { t.Fatalf("Response: " + response + "doesn't contain build command for commit hash\n\n") } if !strings.Contains(response, "docker login -u user -p password -e email") { t.Fatalf("Response: " + response + " doesn't contain login command\n\n") } if !strings.Contains(response, "docker push user/image:$(git rev-parse --short HEAD)") { t.Fatalf("Response: " + response + " doesn't contain push command\n\n") } if !strings.Contains(response, "docker rmi user/image:"+ "$(git rev-parse --short HEAD)") { t.Fatalf("Response: " + response + " doesn't contain remove image command\n\n") } } var withoutDockerFileYaml = ` publish: docker: docker_host: tcp://server:1000 docker_version: 1.0 image_name: user/image username: user password: password email: email ` func TestWithoutDockerFile(t *testing.T) { response, err := setUpWithDrone(withoutDockerFileYaml) t.Log(withoutDockerFileYaml) if err != nil { t.Fatalf("Can't unmarshal script: %s\n\n", err.Error()) } if !strings.Contains(response, "docker build --pull -t user/image:$(git rev-parse --short HEAD) .") { t.Fatalf("Response: " + response + " doesn't contain build command\n\n") } }