package deploy

import (
	"strings"
	"testing"

	"github.com/drone/drone/pkg/build/buildfile"

	"launchpad.net/goyaml"
)

// emulate Build struct
type DeployToCF struct {
	Deploy *Deploy `yaml:"deploy,omitempty"`
}

var sampleYmlBasic = `
deploy:
  cloudfoundry:
    target: https://api.example.com
    username: foo
    password: bar
`

var sampleYmlWithOrg = `
deploy:
  cloudfoundry:
    target: https://api.example.com
    username: foo
    password: bar
    org: custom-org
`

var sampleYmlWithAppName = `
deploy:
  cloudfoundry:
    target: https://api.example.com
    username: foo
    password: bar
    app: test-app
`

func setUpWithCF(input string) (string, error) {
	var buildStruct DeployToCF
	err := goyaml.Unmarshal([]byte(input), &buildStruct)
	if err != nil {
		return "", err
	}
	bf := buildfile.New()
	buildStruct.Deploy.Write(bf)
	return bf.String(), err
}

func TestCloudFoundryDeployment(t *testing.T) {
	bscr, err := setUpWithCF(sampleYmlBasic)
	if err != nil {
		t.Fatalf("Can't unmarshal deploy script: %s", err)
	}

	if !strings.Contains(bscr, "cf login -a https://api.example.com -u foo -p bar -o foo -s dev") {
		t.Error("Expect login script to contains default space")
	}

	if !strings.Contains(bscr, "cf push") {
		t.Error("Expect script to contains push")
	}
}

func TestCloudFoundryDeploymentWithOrg(t *testing.T) {
	bscr, err := setUpWithCF(sampleYmlWithOrg)
	if err != nil {
		t.Fatalf("Can't unmarshal deploy script: %s", err)
	}

	if !strings.Contains(bscr, "cf login -a https://api.example.com -u foo -p bar -o custom-org") {
		t.Error("Expect login script to contains organization")
	}
}

func TestCloudFoundryDeploymentWithApp(t *testing.T) {
	bscr, err := setUpWithCF(sampleYmlWithAppName)
	if err != nil {
		t.Fatalf("Can't unmarshal deploy script: %s", err)
	}

	if !strings.Contains(bscr, "cf push test-app") {
		t.Error("Expect login script to contains app name")
	}
}