env_test.go

package env_test

import (
	"bytes"
	"fmt"
	"net/url"
	"os"
	"strconv"
	"testing"
	"time"

	"github.com/peterhellberg/env"
)

func TestMapClient(t *testing.T) {
	e := env.MapClient(env.Map{
		"FOO": "bar",
		"BAR": "123",
		"BAZ": "true",
	})

	if got, want := e.String("FOO", ""), "bar"; got != want {
		t.Fatalf(`String("FOO", "") = %q, want %q`, got, want)
	}

	if got, want := e.Int("BAR", 0), 123; got != want {
		t.Fatalf(`Int("BAR", 0) = %d, want %d`, got, want)
	}

	if !e.Bool("BAZ", false) {
		t.Fatalf(`!Bool("BAZ", false)`)
	}
}

func TestBool(t *testing.T) {
	tests := []struct {
		env string
		in  bool
		out bool
	}{
		{"t", false, true},
		{"1", false, true},
		{"", false, false},
		{"FALSE", true, false},
	}

	for _, tt := range tests {
		t.Setenv("BOOL", tt.env)

		if got := env.Bool("BOOL", tt.in); got != tt.out {
			t.Errorf(`Bool("BOOL", %v) = %v, want %v`, tt.in, got, tt.out)
		}
	}
}

func TestBoolDefault(t *testing.T) {
	in, out := true, true

	os.Clearenv()

	if got := env.Bool("BOOL_DEFAULT", in); got != out {
		t.Errorf(`Bool("BOOL_DEFAULT", %v) = %v, want %v`, in, got, out)
	}
}

func ExampleBool() {
	os.Setenv("BOOL", "t")

	fmt.Println(env.Bool("BOOL", false))
	// Output: true
}

func TestBytes(t *testing.T) {
	in, out := []byte("baz"), []byte("bar")

	t.Setenv("BYTES", string(out))

	if got := env.Bytes("BYTES", in); !bytes.Equal(got, out) {
		t.Errorf(`Bytes("BYTES", "%s") = %s, want %s`, in, got, out)
	}
}

func TestBytesDefault(t *testing.T) {
	in, out := []byte("baz"), []byte("baz")

	os.Clearenv()

	if got := env.Bytes("BYTES_DEFAULT", in); !bytes.Equal(got, out) {
		t.Errorf(`Bytes("BYTES_DEFAULT", "%s") = %s, want %s`, in, got, out)
	}
}

func ExampleBytes() {
	os.Setenv("BYTES", "foo")

	fmt.Printf("%s", env.Bytes("BYTES", nil))
	// Output: foo
}

func TestDuration(t *testing.T) {
	in, fallback, out := "5s", 10*time.Minute, 5*time.Second

	t.Setenv("DURATION", in)

	if got := env.Duration("DURATION", fallback); got != out {
		t.Errorf(`Duration("DURATION", %#v) = %v, want %v`, in, got, out)
	}
}

func TestDurationDefault(t *testing.T) {
	fallback := 123 * time.Second

	os.Clearenv()

	if got := env.Duration("DURATION_DEFAULT", fallback); got != fallback {
		t.Errorf(`Duration("DURATION_DEFAULT", %#v) = %v, want %v`, fallback, got, fallback)
	}
}

func ExampleDuration() {
	os.Setenv("DURATION", "23s")

	fmt.Printf("%s", env.Duration("DURATION", 0))
	// Output: 23s
}

func TestFloat64(t *testing.T) {
	in, out := float64(1.1), float64(2.5)

	t.Setenv("FLOAT64", "2.5")

	if got := env.Float64("FLOAT64", in); got != out {
		t.Errorf(`Float64("FLOAT64", %v) = %v, want %v`, in, got, out)
	}
}

func TestFloat64Default(t *testing.T) {
	in, out := float64(5.2), float64(5.2)

	os.Clearenv()

	if got := env.Float64("FLOAT64_DEFAULT", in); got != out {
		t.Errorf(`Float64("FLOAT64_DEFAULT", %v) = %v, want %v`, in, got, out)
	}
}

func ExampleFloat64() {
	os.Setenv("FLOAT64", "1.23")

	fmt.Println(env.Float64("FLOAT64", 0))
	// Output: 1.23
}

func TestInt(t *testing.T) {
	in, out := 1, 2

	t.Setenv("INT", strconv.Itoa(out))

	if got := env.Int("INT", in); got != out {
		t.Errorf(`Int("INT", %v) = %v, want %v`, in, got, out)
	}
}

func TestIntDefault(t *testing.T) {
	in, out := 3, 3

	os.Clearenv()

	if got := env.Int("INT_DEFAULT", in); got != out {
		t.Errorf(`Int("INT_DEFAULT", %v) = %v, want %v`, in, got, out)
	}
}

func ExampleInt() {
	os.Setenv("INT", "345")

	fmt.Println(env.Int("INT", 0))
	// Output: 345
}

func TestString(t *testing.T) {
	in, out := "baz", "bar"

	t.Setenv("STRING", out)

	if got := env.String("STRING", in); got != out {
		t.Errorf(`String("STRING", "%v") = %v, want %v`, in, got, out)
	}
}

func TestStringDefault(t *testing.T) {
	in, out := "baz", "baz"

	os.Clearenv()

	if got := env.String("STRING_DEFAULT", in); got != out {
		t.Errorf(`String("STRING_DEFAULT", "%v") = %v, want %v`, in, got, out)
	}
}

func ExampleString() {
	os.Setenv("STRING", "foo bar")

	fmt.Println(env.String("STRING", ""))
	// Output: foo bar
}

func TestStrings(t *testing.T) {
	eqStrings := func(a, b []string) bool {
		if len(a) != len(b) {
			return false
		}

		for i := range a {
			if a[i] != b[i] {
				return false
			}
		}

		return true
	}

	tests := []struct {
		env      string
		fallback []string
		want     []string
	}{
		{"foo,bar,baz", []string{}, []string{"foo", "bar", "baz"}},
		{"", []string{"sit", "amet"}, []string{"sit", "amet"}},
	}

	for _, tt := range tests {
		t.Setenv("STRINGS", tt.env)

		if got := env.Strings("STRINGS", tt.fallback); !eqStrings(got, tt.want) {
			t.Errorf(`String("STRINGS", %q) = %q, want %q`, tt.fallback, got, tt.want)
		}
	}
}

func TestStrings_differentSeparator(t *testing.T) {
	eqStrings := func(a, b []string) bool {
		if len(a) != len(b) {
			return false
		}

		for i := range a {
			if a[i] != b[i] {
				return false
			}
		}

		return true
	}

	tests := []struct {
		env      string
		sep      string
		fallback []string
		want     []string
	}{
		{"foo-bar-baz", "-", []string{}, []string{"foo", "bar", "baz"}},
		{"foo/bar/baz", "/", []string{}, []string{"foo", "bar", "baz"}},
		{"", "/", []string{"sit", "amet"}, []string{"sit", "amet"}},
	}

	for _, tt := range tests {
		t.Setenv("STRINGS", tt.env)

		if got := env.Strings("STRINGS", tt.fallback, tt.sep); !eqStrings(got, tt.want) {
			t.Errorf(`String("STRINGS", %q, %q) = %q, want %q`, tt.fallback, tt.sep, got, tt.want)
		}
	}
}

func ExampleStrings() {
	os.Setenv("STRINGS", "foo,bar,baz")

	fmt.Println(env.Strings("STRINGS", []string{}))
	// Output: [foo bar baz]
}

func TestURL(t *testing.T) {
	in, out := &url.URL{Host: "example.com"}, &url.URL{Host: "example.com"}

	t.Setenv("URL", out.String())

	if got := env.URL("URL", in); got.String() != out.String() {
		t.Errorf(`URL("URL", "%v") = %v, want %v`, in, got, out)
	}
}

func TestURLDefault(t *testing.T) {
	in, out := &url.URL{Host: "example.com"}, &url.URL{Host: "example.com"}

	os.Clearenv()

	if got := env.URL("URL_DEFAULT", in); got.String() != out.String() {
		t.Errorf(`URL("URL_DEFAULT", "%v") = %v, want %v`, in, got, out)
	}
}

func ExampleURL() {
	os.Setenv("URL", "http://example.com/foo")

	fmt.Println(env.URL("URL", &url.URL{Host: "fallback"}).String())
	// Output: http://example.com/foo
}