package yaml

import (
	"testing"

	"github.com/docker/libcompose/yaml"
	"github.com/franela/goblin"
)

func TestParse(t *testing.T) {
	g := goblin.Goblin(t)

	g.Describe("Parser", func() {
		g.Describe("Given a yaml file", func() {

			g.It("Should unmarshal a string", func() {
				out, err := ParseString(sampleYaml)
				if err != nil {
					g.Fail(err)
				}

				g.Assert(out.Workspace.Base).Equal("/go")
				g.Assert(out.Workspace.Path).Equal("src/github.com/octocat/hello-world")
				g.Assert(out.Volumes.Volumes[0].Name).Equal("custom")
				g.Assert(out.Volumes.Volumes[0].Driver).Equal("blockbridge")
				g.Assert(out.Networks.Networks[0].Name).Equal("custom")
				g.Assert(out.Networks.Networks[0].Driver).Equal("overlay")
				g.Assert(out.Services.Containers[0].Name).Equal("database")
				g.Assert(out.Services.Containers[0].Image).Equal("mysql")
				g.Assert(out.Pipeline.Containers[0].Name).Equal("test")
				g.Assert(out.Pipeline.Containers[0].Image).Equal("golang")
				g.Assert(out.Pipeline.Containers[0].Commands).Equal(yaml.Stringorslice{"go install", "go test"})
				g.Assert(out.Pipeline.Containers[1].Name).Equal("build")
				g.Assert(out.Pipeline.Containers[1].Image).Equal("golang")
				g.Assert(out.Pipeline.Containers[1].Commands).Equal(yaml.Stringorslice{"go build"})
				g.Assert(out.Pipeline.Containers[2].Name).Equal("notify")
				g.Assert(out.Pipeline.Containers[2].Image).Equal("slack")
				// g.Assert(out.Pipeline.Containers[2].NetworkMode).Equal("container:name")
				g.Assert(out.Labels["com.example.team"]).Equal("frontend")
				g.Assert(out.Labels["com.example.type"]).Equal("build")
				g.Assert(out.DependsOn[0]).Equal("lint")
				g.Assert(out.DependsOn[1]).Equal("test")
				g.Assert(out.RunsOn[0]).Equal("success")
				g.Assert(out.RunsOn[1]).Equal("failure")
				g.Assert(out.SkipClone).Equal(false)
			})

			g.It("Should handle simple yaml anchors", func() {
				out, err := ParseString(simpleYamlAnchors)
				if err != nil {
					g.Fail(err)
				}
				g.Assert(out.Pipeline.Containers[0].Name).Equal("notify_success")
				g.Assert(out.Pipeline.Containers[0].Image).Equal("plugins/slack")
			})

			g.It("Should unmarshal variables", func() {
				out, err := ParseString(sampleVarYaml)
				if err != nil {
					g.Fail(err)
				}
				g.Assert(out.Pipeline.Containers[0].Name).Equal("notify_fail")
				g.Assert(out.Pipeline.Containers[0].Image).Equal("plugins/slack")
				g.Assert(len(out.Pipeline.Containers[0].Constraints.Event.Include)).Equal(0)
				g.Assert(out.Pipeline.Containers[1].Name).Equal("notify_success")
				g.Assert(out.Pipeline.Containers[1].Image).Equal("plugins/slack")
				g.Assert(out.Pipeline.Containers[1].Constraints.Event.Include).Equal([]string{"success"})
			})
		})
	})
}

var sampleYaml = `
image: hello-world
build:
  context: .
  dockerfile: Dockerfile
workspace:
  path: src/github.com/octocat/hello-world
  base: /go
pipeline:
  test:
    image: golang
    commands:
      - go install
      - go test
  build:
    image: golang
    network_mode: container:name
    commands:
      - go build
    when:
      event: push
  notify:
    image: slack
    channel: dev
    when:
      event: failure
services:
  database:
    image: mysql
networks:
  custom:
    driver: overlay
volumes:
  custom:
    driver: blockbridge
labels:
  com.example.type: "build"
  com.example.team: "frontend"
depends_on:
  - lint
  - test
runs_on:
  - success
  - failure
`

var simpleYamlAnchors = `
vars:
  image: &image plugins/slack
pipeline:
  notify_success:
    image: *image
`

var sampleVarYaml = `
_slack: &SLACK
  image: plugins/slack
pipeline:
  notify_fail: *SLACK
  notify_success:
    << : *SLACK
    when:
      event: success
`
