package transform

import (
	"path/filepath"
	"testing"

	"github.com/drone/drone/Godeps/_workspace/src/github.com/franela/goblin"
	common "github.com/drone/drone/pkg/types"
)

func Test_Transform(t *testing.T) {

	g := goblin.Goblin(t)
	g.Describe("Transform", func() {

		g.It("Should transform setup step", func() {
			c := &common.Config{}
			c.Build = &common.Step{}
			c.Build.Config = map[string]interface{}{}
			transformSetup(c)
			g.Assert(c.Setup != nil).IsTrue()
			g.Assert(c.Setup.Image).Equal("plugins/drone-build")
			g.Assert(c.Setup.Config).Equal(c.Build.Config)
		})

		g.It("Should transform clone step", func() {
			c := &common.Config{}
			transformClone(c)
			g.Assert(c.Clone != nil).IsTrue()
			g.Assert(c.Clone.Image).Equal("plugins/drone-git")
		})

		g.It("Should transform build", func() {
			c := &common.Config{}
			c.Build = &common.Step{}
			c.Build.Config = map[string]interface{}{}
			c.Build.Config["commands"] = []string{"echo hello"}
			transformBuild(c)
			g.Assert(len(c.Build.Config)).Equal(0)
			g.Assert(c.Build.Entrypoint[0]).Equal("/bin/bash")
			g.Assert(c.Build.Command[0]).Equal("/drone/bin/build.sh")
		})

		g.It("Should transform images", func() {
			c := &common.Config{}
			c.Setup = &common.Step{Image: "foo"}
			c.Clone = &common.Step{Image: "foo/bar"}
			c.Build = &common.Step{Image: "golang"}
			c.Publish = map[string]*common.Step{"google_compute": &common.Step{}}
			c.Deploy = map[string]*common.Step{"amazon": &common.Step{}}
			c.Notify = map[string]*common.Step{"slack": &common.Step{}}
			transformImages(c)

			g.Assert(c.Setup.Image).Equal("plugins/drone-foo")
			g.Assert(c.Clone.Image).Equal("foo/bar")
			g.Assert(c.Build.Image).Equal("golang")
			g.Assert(c.Publish["google_compute"].Image).Equal("plugins/drone-google-compute")
			g.Assert(c.Deploy["amazon"].Image).Equal("plugins/drone-amazon")
			g.Assert(c.Notify["slack"].Image).Equal("plugins/drone-slack")
		})

		g.It("Should transform docker plugin", func() {
			c := &common.Config{}
			c.Publish = map[string]*common.Step{}
			c.Publish["docker"] = &common.Step{Image: "plugins/drone-docker"}
			transformDockerPlugin(c)
			g.Assert(c.Publish["docker"].Privileged).Equal(true)
		})

		g.It("Should remove privileged flag", func() {
			c := &common.Config{}
			c.Setup = &common.Step{Privileged: true}
			c.Clone = &common.Step{Privileged: true}
			c.Build = &common.Step{Privileged: true}
			c.Compose = map[string]*common.Step{"postgres": &common.Step{Privileged: true}}
			c.Publish = map[string]*common.Step{"google": &common.Step{Privileged: true}}
			c.Deploy = map[string]*common.Step{"amazon": &common.Step{Privileged: true}}
			c.Notify = map[string]*common.Step{"slack": &common.Step{Privileged: true}}
			rmPrivileged(c)

			g.Assert(c.Setup.Privileged).Equal(false)
			g.Assert(c.Clone.Privileged).Equal(false)
			g.Assert(c.Build.Privileged).Equal(false)
			g.Assert(c.Compose["postgres"].Privileged).Equal(false)
			g.Assert(c.Publish["google"].Privileged).Equal(false)
			g.Assert(c.Deploy["amazon"].Privileged).Equal(false)
			g.Assert(c.Notify["slack"].Privileged).Equal(false)
		})

		g.It("Should not remove docker plugin privileged flag", func() {
			c := &common.Config{}
			c.Setup = &common.Step{}
			c.Clone = &common.Step{}
			c.Build = &common.Step{}
			c.Publish = map[string]*common.Step{}
			c.Publish["docker"] = &common.Step{Image: "plugins/drone-docker"}
			transformDockerPlugin(c)
			g.Assert(c.Publish["docker"].Privileged).Equal(true)
		})

		g.It("Should remove volumes", func() {
			c := &common.Config{}
			c.Setup = &common.Step{Volumes: []string{"/:/tmp"}}
			c.Clone = &common.Step{Volumes: []string{"/:/tmp"}}
			c.Build = &common.Step{Volumes: []string{"/:/tmp"}}
			c.Compose = map[string]*common.Step{"postgres": &common.Step{Volumes: []string{"/:/tmp"}}}
			c.Publish = map[string]*common.Step{"google": &common.Step{Volumes: []string{"/:/tmp"}}}
			c.Deploy = map[string]*common.Step{"amazon": &common.Step{Volumes: []string{"/:/tmp"}}}
			c.Notify = map[string]*common.Step{"slack": &common.Step{Volumes: []string{"/:/tmp"}}}
			rmVolumes(c)

			g.Assert(len(c.Setup.Volumes)).Equal(0)
			g.Assert(len(c.Clone.Volumes)).Equal(0)
			g.Assert(len(c.Build.Volumes)).Equal(0)
			g.Assert(len(c.Compose["postgres"].Volumes)).Equal(0)
			g.Assert(len(c.Publish["google"].Volumes)).Equal(0)
			g.Assert(len(c.Deploy["amazon"].Volumes)).Equal(0)
			g.Assert(len(c.Notify["slack"].Volumes)).Equal(0)
		})

		g.It("Should remove network", func() {
			c := &common.Config{}
			c.Setup = &common.Step{NetworkMode: "host"}
			c.Clone = &common.Step{NetworkMode: "host"}
			c.Build = &common.Step{NetworkMode: "host"}
			c.Compose = map[string]*common.Step{"postgres": &common.Step{NetworkMode: "host"}}
			c.Publish = map[string]*common.Step{"google": &common.Step{NetworkMode: "host"}}
			c.Deploy = map[string]*common.Step{"amazon": &common.Step{NetworkMode: "host"}}
			c.Notify = map[string]*common.Step{"slack": &common.Step{NetworkMode: "host"}}
			rmNetwork(c)

			g.Assert(c.Setup.NetworkMode).Equal("")
			g.Assert(c.Clone.NetworkMode).Equal("")
			g.Assert(c.Build.NetworkMode).Equal("")
			g.Assert(c.Compose["postgres"].NetworkMode).Equal("")
			g.Assert(c.Publish["google"].NetworkMode).Equal("")
			g.Assert(c.Deploy["amazon"].NetworkMode).Equal("")
			g.Assert(c.Notify["slack"].NetworkMode).Equal("")
		})

		g.It("Should return full qualified image name", func() {
			g.Assert(imageName("microsoft/azure")).Equal("microsoft/azure")
			g.Assert(imageName("azure")).Equal("plugins/drone-azure")
			g.Assert(imageName("azure_storage")).Equal("plugins/drone-azure-storage")
		})

		g.It("Should have cached volumes", func() {
			c := &common.Config{
				Setup: &common.Step{},
				Clone: &common.Step{
					Config: map[string]interface{}{},
				},
				Build: &common.Step{
					Cache: []string{".git", "foo", "bar"},
				},
				Notify:  map[string]*common.Step{},
				Deploy:  map[string]*common.Step{},
				Publish: map[string]*common.Step{},
			}
			r := &common.Repo{
				Link:     "https://github.com/drone/drone",
				FullName: "drone/drone",
			}
			transformWorkspace(c, r)
			transformCache(c, r)

			cacheCount := len(c.Build.Cache)

			test := func(s *common.Step) {
				g.Assert(len(s.Volumes)).Equal(cacheCount)
			}

			testRange := func(s map[string]*common.Step) {
				for _, step := range s {
					test(step)
				}
			}

			test(c.Setup)
			test(c.Clone)
			test(c.Build)
			testRange(c.Publish)
			testRange(c.Deploy)
			testRange(c.Notify)
			testRange(c.Compose)
		})

		g.It("Should have default workspace directory", func() {
			c := &common.Config{
				Clone: &common.Step{
					Config: map[string]interface{}{},
				},
			}
			r := &common.Repo{
				Link:     "https://github.com/drone/drone",
				FullName: "drone/drone",
			}
			transformWorkspace(c, r)

			g.Assert(c.Clone.Config["path"]).Equal(workspaceRoot(r))
		})

		g.It("Should use path for working directory", func() {
			c := &common.Config{
				Clone: &common.Step{
					Config: map[string]interface{}{
						"path": "foo/bar",
					},
				},
			}
			r := &common.Repo{
				Link:     "https://github.com/drone/drone",
				FullName: "drone/drone",
			}
			transformWorkspace(c, r)

			g.Assert(c.Clone.Config["path"]).Equal(filepath.Join(buildRoot, "foo/bar"))
		})
	})
}