env_test.go

package main

import (
	"testing"

	"github.com/stretchr/testify/assert"
)

func TestEnvMap(t *testing.T) {
	t.Setenv("DOCKER_CONSUL_BOOTSTRAP_TEST_FOO", "ignore")
	t.Setenv("DOCKER_CONSUL_BOOTSTRAP_TEST_BAR", "ignored")

	e := NewEnvMap()
	e.Merge(map[string]string{
		"DOCKER_CONSUL_BOOTSTRAP_TEST_FOO":  "changed",
		"DOCKER_CONSUL_BOOTSTRAP_TEST_TEST": "testing",
		"DOCKER_CONSUL_BOOTSTRAP_TEST_BAZ":  "foo",
	})
	e.Merge(map[string]string{
		"DOCKER_CONSUL_BOOTSTRAP_TEST_BAZ": "bar",
	})
	e.Add("DOCKER_CONSUL_BOOTSTRAP_TEST_BAR", "testing")
	e.Add("DOCKER_CONSUL_BOOTSTRAP_TEST_DOLLAR", "testing$a1^h")
	e.Add("DOCKER_CONSUL_BOOTSTRAP_TEST_EXTRA", "abc^d$1234&567")
	e.Add("DOCKER_CONSUL_BOOTSTRAP_EXPAND_ONE", "foo-${DOCKER_CONSUL_BOOTSTRAP_TEST_FOO}-bar")
	e.Add("DOCKER_CONSUL_BOOTSTRAP_EXPAND_TWO", "foo-${DOCKER_CONSUL_BOOTSTRAP_TEST_BAZ}-baz")
	e.Add("  ", "test-empty")

	assert.ElementsMatch(t, []string{
		"DOCKER_CONSUL_BOOTSTRAP_TEST_TEST=testing",
		"DOCKER_CONSUL_BOOTSTRAP_TEST_DOLLAR=testing$a1^h",
		"DOCKER_CONSUL_BOOTSTRAP_TEST_EXTRA=abc^d$1234&567",
		"DOCKER_CONSUL_BOOTSTRAP_TEST_BAZ=bar",
		"DOCKER_CONSUL_BOOTSTRAP_EXPAND_ONE=foo-ignore-bar",
		"DOCKER_CONSUL_BOOTSTRAP_EXPAND_TWO=foo-bar-baz",
	}, e.Environ())

	e.Add("DOCKER_CONSUL_BOOTSTRAP_NEWLINE_ESCAPED", "hello\nworld\n")
	e.Add("DOCKER_CONSUL_BOOTSTRAP_NEWLINE", `foo
bar
baz

test`)

	assert.Equal(t, map[string]string{
		"DOCKER_CONSUL_BOOTSTRAP_TEST_TEST":       "testing",
		"DOCKER_CONSUL_BOOTSTRAP_TEST_DOLLAR":     "testing$a1^h",
		"DOCKER_CONSUL_BOOTSTRAP_TEST_EXTRA":      "abc^d$1234&567",
		"DOCKER_CONSUL_BOOTSTRAP_TEST_BAZ":        "bar",
		"DOCKER_CONSUL_BOOTSTRAP_EXPAND_ONE":      "foo-ignore-bar",
		"DOCKER_CONSUL_BOOTSTRAP_EXPAND_TWO":      "foo-bar-baz",
		"DOCKER_CONSUL_BOOTSTRAP_NEWLINE":         "foo\nbar\nbaz\n\ntest",
		"DOCKER_CONSUL_BOOTSTRAP_NEWLINE_ESCAPED": "hello\nworld\n",
	}, e.Map())
}