flag_test.go

package main

import (
	"os"
	"strconv"
	"testing"
)

func TestLookupEnvOrString_EnvVar(t *testing.T) {
	envKey := "TWB_ENV_STRING_TEST_KEY_ENVVALUE"
	envValue := "env-value"
	defaultVal := "default"

	err := os.Setenv(envKey, envValue)
	if err != nil {
		t.Errorf("not able to set environment variable %q: %q", envKey, err)
	}
	defer os.Unsetenv(envKey)

	if v := LookupEnvOrString(envKey, defaultVal); v != envValue {
		t.Errorf("got %q, want %q", v, envValue)
	}
}

func TestLookupEnvOrString_DefaultVal(t *testing.T) {
	envKey := "TWB_ENV_STRING_TEST_KEY_DEFAULTVALUE"
	defaultVal := "default"

	if v := LookupEnvOrString(envKey, defaultVal); v != defaultVal {
		t.Errorf("got %q, want %q", v, defaultVal)
	}
}

func TestLookupEnvOrBool_EnvSet(t *testing.T) {
	envKey := "TWB_ENV_BOOL_TEST_KEY_ENVVALUE"
	envValue := true
	defaultVal := false

	err := os.Setenv(envKey, strconv.FormatBool(envValue))
	if err != nil {
		t.Errorf("not able to set environment variable %q: %q", envKey, err)
	}
	defer os.Unsetenv(envKey)

	if v := LookupEnvOrBool(envKey, defaultVal); v != envValue {
		t.Errorf("got %v, want %v", v, envValue)
	}
}

func TestLookupEnvOrBool_DefaultVal(t *testing.T) {
	envKey := "TWB_ENV_BOOL_TEST_KEY_DEFAULTVALUE"
	defaultVal := true

	if v := LookupEnvOrBool(envKey, defaultVal); v != defaultVal {
		t.Errorf("got %v, want %v", v, defaultVal)
	}
}