drone-yaml/yaml/compiler/transform/secret_test.go

package transform

import (
	"testing"

	"github.com/drone/drone-runtime/engine"
	"github.com/google/go-cmp/cmp"
)

func TestWithSecret(t *testing.T) {
	step := &engine.Step{
		Metadata: engine.Metadata{
			UID:  "1",
			Name: "build",
		},
		Envs: map[string]string{},
	}
	spec := &engine.Spec{
		Metadata: engine.Metadata{
			UID:       "acdj0yjqv7uh5hidveg0ggr42x8oj67b",
			Namespace: "pivqfthg1c9hy83ylht1sxx4nygjc7tk",
		},
		Steps: []*engine.Step{step},
	}
	secrets := map[string]string{
		"password": "correct-horse-battery-staple",
	}
	WithSecrets(secrets)(spec)

	want := []*engine.Secret{
		{
			Metadata: engine.Metadata{
				Name:      "password",
				Namespace: "pivqfthg1c9hy83ylht1sxx4nygjc7tk",
			},
			Data: "correct-horse-battery-staple",
		},
	}
	if diff := cmp.Diff(want, spec.Secrets, ignoreMetadata); diff != "" {
		t.Errorf("Unexpected secret transform")
		t.Log(diff)
	}
}

func TestWithSecretFunc(t *testing.T) {
	step := &engine.Step{
		Metadata: engine.Metadata{
			UID:  "1",
			Name: "build",
		},
		Envs: map[string]string{},
		Secrets: []*engine.SecretVar{
			{
				Name: "password",
				Env:  "PASSWORD",
			},
		},
	}
	spec := &engine.Spec{
		Metadata: engine.Metadata{
			UID:       "acdj0yjqv7uh5hidveg0ggr42x8oj67b",
			Namespace: "pivqfthg1c9hy83ylht1sxx4nygjc7tk",
		},
		Steps: []*engine.Step{
			step,
			// this is a step that requests a secret
			// but should be skipped.
			{
				RunPolicy: engine.RunNever,
				Secrets: []*engine.SecretVar{
					{
						Name: "github_token",
						Env:  "GITHUB_TOKEN",
					},
				},
			},
		},
	}

	fn := func(name string) *engine.Secret {
		if name == "github_token" {
			t.Errorf("Requested secret for skipped step")
			return nil
		}
		return &engine.Secret{
			Metadata: engine.Metadata{
				Name: "password",
			},
			Data: "correct-horse-battery-staple",
		}
	}
	WithSecretFunc(fn)(spec)

	want := []*engine.Secret{
		{
			Metadata: engine.Metadata{
				Name:      "password",
				Namespace: "pivqfthg1c9hy83ylht1sxx4nygjc7tk",
			},
			Data: "correct-horse-battery-staple",
		},
	}
	if diff := cmp.Diff(want, spec.Secrets, ignoreMetadata); diff != "" {
		t.Errorf("Unexpected secret transform")
		t.Log(diff)
	}
}