envs_test.go

package envs

import (
	"os"
	"testing"
)

func Test_Get(t *testing.T) {
	envs := EnvConfig{}
	_ = os.Setenv("TEST_GET_ENVS", "get_envs")
	envs.ReadEnvs()
	if res := envs.Get("TEST_GET_ENVS"); res != "get_envs" {
		t.Errorf("expected 'get_envs', got '%s'", res)
	}
	if res := envs.Get("TEST_GET_ENVS_DEFAULT", "default_string"); res != "default_string" {
		t.Errorf("expected 'default_string', got '%s'", res)
	}
}

func Test_GetBool(t *testing.T) {
	envs := EnvConfig{}
	_ = os.Setenv("TEST_GET_ENVS_BOOL", "true")
	envs.ReadEnvs()
	if res := envs.GetBool("TEST_GET_ENVS_BOOL"); !res {
		t.Errorf("expected true, got %t", res)
	}
	if res := envs.GetBool("TEST_GET_ENVS_BOOL_DEFAULT", true); !res {
		t.Errorf("expected true, got %t", res)
	}
}

func Test_GetFloat(t *testing.T) {
	envs := EnvConfig{}
	_ = os.Setenv("TEST_GET_ENVS_FLOAT", "123.7645")
	envs.ReadEnvs()
	if res := envs.GetFloat("TEST_GET_ENVS_FLOAT"); res != float32(123.7645) {
		t.Errorf("expected 123.7645, got %f", res)
	}
	if res := envs.GetFloat("TEST_GET_ENVS_FLOAT_DEFAULT", 456.123); res != float32(456.123) {
		t.Errorf("expected 456.123, got %f", res)
	}
}

func Test_GetInt(t *testing.T) {
	envs := EnvConfig{}
	_ = os.Setenv("TEST_GET_ENVS_INT", "123")
	envs.ReadEnvs()
	if res := envs.GetInt("TEST_GET_ENVS_INT"); res != 123 {
		t.Errorf("expected 123, got %d", res)
	}
	if res := envs.GetInt("TEST_GET_ENVS_INT_DEFAULT", 876); res != 876 {
		t.Errorf("expected 876, got %d", res)
	}
}

func Test_GetMap(t *testing.T) {
	envs := EnvConfig{}
	_ = os.Setenv("TEST_GET_ENVS_MAP", "key1:value1;key2:value2")
	envs.ReadEnvs()
	res := envs.GetMap("TEST_GET_ENVS_MAP")
	if value := res["key1"]; value != "value1" {
		t.Errorf("expected 'value1', got '%s'", value)
	}
	if value := res["key2"]; value != "value2" {
		t.Errorf("expected 'value2', got '%s'", value)
	}
	res = envs.GetMap("TEST_GET_ENVS_MAP_DEFAULT", map[string]string{"defaultKey": "defaultValue"})
	if value := res["defaultKey"]; value != "defaultValue" {
		t.Errorf("expected 'defaultValue', got '%s'", value)
	}
}

func Test_GetSlice(t *testing.T) {
	envs := EnvConfig{}
	_ = os.Setenv("TEST_GET_ENVS_SLICE", "first,second,third")
	envs.ReadEnvs()
	expected := []string{"first", "second", "third"}
	for i, value := range envs.GetSlice("TEST_GET_ENVS_SLICE") {
		if value != expected[i] {
			t.Errorf("expected '%s', got '%s'", expected[i], value)
		}
	}
	if res := envs.GetSlice("TEST_GET_ENVS_SLICE_DEFAULT", []string{"defaultValue"}); res[0] != "defaultValue" {
		t.Errorf("expected 'defaultValue', got '%s'", res[0])
	}
}

func Test_GetSliceFloat(t *testing.T) {
	envs := EnvConfig{}
	_ = os.Setenv("TEST_GET_ENVS_SLICE_FLOAT", "123.554,64.23,78.09876")
	envs.ReadEnvs()
	expected := []float32{123.554, 64.23, 78.09876}
	for i, value := range envs.GetSliceFloat("TEST_GET_ENVS_SLICE_FLOAT") {
		if value != expected[i] {
			t.Errorf("expected %f, got %f", expected[i], value)
		}
	}
	res := envs.GetSliceFloat("TEST_GET_ENVS_SLICE_FLOAT_DEFAULT", []float32{456.235})
	if res[0] != float32(456.235) {
		t.Errorf("expected 456.235, got %f", res[0])
	}
}

func Test_GetSliceInt(t *testing.T) {
	envs := EnvConfig{}
	_ = os.Setenv("TEST_GET_ENVS_SLICE_INT", "453,8,124567")
	envs.ReadEnvs()
	expected := []int{453, 8, 124567}
	for i, value := range envs.GetSliceInt("TEST_GET_ENVS_SLICE_INT") {
		if value != expected[i] {
			t.Errorf("expected %d, got %d", expected[i], value)
		}
	}
	if res := envs.GetSliceInt("TEST_GET_ENVS_SLICE_INT_DEFAULT", []int{3576}); res[0] != 3576 {
		t.Errorf("expected 3576, got %d", res[0])
	}
}

func Test_readEnvs(t *testing.T) {
	envs := EnvConfig{}

	{ // the .env file exists - envs are empty
		_ = os.WriteFile(".env", []byte{}, 0644)
		envs.ReadEnvs()
		if res := envs.Get("TEST_READ_ENVS_1"); res != "" {
			t.Errorf("expected empty value, got %s", res)
		}
		_ = os.Remove(".env")
	}

	{ // the .env file exists - envs are read
		_ = os.WriteFile(".env", []byte("# Comment\nTEST_READ_ENVS_2=read_envs_2"), 0644)
		envs.ReadEnvs()
		if res := envs.Get("TEST_READ_ENVS_2"); res != "read_envs_2" {
			t.Errorf("expected 'read_envs_2', got %s", res)
		}
		_ = os.Remove(".env")
	}

	{ // a file with custom name exists - envs are read
		_ = os.WriteFile(".custom_name", []byte("TEST_READ_ENVS_3=read_envs_3"), 0644)
		envs.Filepath = ".custom_name"
		envs.ReadEnvs()
		if res := envs.Get("TEST_READ_ENVS_3"); res != "read_envs_3" {
			t.Errorf("expected 'read_envs_3', got %s", res)
		}
		_ = os.Remove(".custom_name")
		envs.Filepath = ""
	}

	{ // the .env file exists and the same env var was set - envs are read
		_ = os.Setenv("TEST_READ_ENVS_4", "read_envs_4_from_env?somePara=value")
		_ = os.WriteFile(".env", []byte("TEST_READ_ENVS_4=read_envs_4"), 0644)
		envs.ReadEnvs()
		if res := envs.Get("TEST_READ_ENVS_4"); res != "read_envs_4_from_env?somePara=value" {
			t.Errorf("expected 'read_envs_4_from_env?somePara=value', got %s", res)
		}
		_ = os.Remove(".env")
	}

	{ // the .env file doesn't exists - envs are empty
		envs.ReadEnvs()
		if res := envs.Get("TEST_READ_ENVS_5"); res != "" {
			t.Errorf("expected empty value, got %s", res)
		}
	}

	{ // the .env file doesn't exists - envs are read
		_ = os.Setenv("TEST_READ_ENVS_6", "read_envs_6")
		envs.ReadEnvs()
		if res := envs.Get("TEST_READ_ENVS_6"); res != "read_envs_6" {
			t.Errorf("expected 'read_envs_6', got %s", res)
		}
	}
}