get_uint64_var_test.go

package env

import (
	"os"
	"testing"
)

type uint64FlagTest struct {
	valueIn  string
	valueOut uint64
	fallback uint64
}

var getUint64VarFlagTests = []uint64FlagTest{
	{"0", 0, 7},
	{"42", 42, 7},
	{"13", 13, 7},
	{"255", 255, 7},
	{"65535", 65535, 7},
	{"4294967295", 4294967295, 7},
	{"18446744073709551615", 18446744073709551615, 7},
}

var getUint64VarFlagTestsInvalid = []string{
	"test",
	"!!!",
	"+",
	"-",
	"10.5",
	"-1",
}

// Test option with preset value
func TestGetUint64Var(t *testing.T) {
	for _, tt := range getUint64VarFlagTests {
		t.Run(tt.valueIn, func(t *testing.T) {
			os.Clearenv()

			if err := os.Setenv(variableName, tt.valueIn); err != nil {
				t.Error(err)
			}

			v, err := GetUint64Var(variableName, tt.fallback)

			if err != nil {
				t.Error(err)
			}

			if v != tt.valueOut {
				t.Errorf("Variable %s not equal to value '%v'", variableName, tt.valueOut)
			}
		})
	}
}

// Test option when variable was not found
func TestGetUint64VarDefault(t *testing.T) {
	os.Clearenv()

	v, err := GetUint64Var(variableName, 12)

	if err != nil {
		t.Error(err)
	}

	if v != 12 {
		t.Error("Variable must contain 12")
	}
}

// Test option when variable contain invalid value
func TestGetUint64VarInvalidValue(t *testing.T) {
	for _, v := range getUint64VarFlagTestsInvalid {
		t.Run(v, func(t *testing.T) {
			os.Clearenv()
			if err := os.Setenv(variableName, v); err != nil {
				t.Error(err)
			}

			v, err := GetUint64Var(variableName, 0)

			if err == nil {
				t.Error("Must be error")
			}

			if v != 0 {
				t.Error("Variable must contain 0")
			}
		})
	}
}