mirror of
https://github.com/oauth2-proxy/oauth2-proxy.git
synced 2025-03-23 21:50:48 +02:00
65 lines
1.4 KiB
Go
65 lines
1.4 KiB
Go
package main_test
|
|
|
|
import (
|
|
"os"
|
|
"testing"
|
|
|
|
proxy "github.com/oauth2-proxy/oauth2-proxy"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
type EnvTest struct {
|
|
TestField string `cfg:"target_field" env:"TEST_ENV_FIELD"`
|
|
EnvTestEmbed
|
|
Named EnvTestNamed
|
|
}
|
|
|
|
type EnvTestEmbed struct {
|
|
TestFieldEmbed string `cfg:"target_field_embed" env:"TEST_ENV_FIELD_EMBED"`
|
|
}
|
|
|
|
type EnvTestNamed struct {
|
|
TestFieldNamed string `cfg:"target_field_named" env:"TEST_ENV_FIELD_NAMED"`
|
|
}
|
|
|
|
func TestLoadEnvForStruct(t *testing.T) {
|
|
|
|
cfg := make(proxy.EnvOptions)
|
|
cfg.LoadEnvForStruct(&EnvTest{})
|
|
|
|
_, ok := cfg["target_field"]
|
|
assert.Equal(t, ok, false)
|
|
|
|
os.Setenv("TEST_ENV_FIELD", "1234abcd")
|
|
cfg.LoadEnvForStruct(&EnvTest{})
|
|
v := cfg["target_field"]
|
|
assert.Equal(t, v, "1234abcd")
|
|
}
|
|
|
|
func TestLoadEnvForStructWithEmbeddedFields(t *testing.T) {
|
|
|
|
cfg := make(proxy.EnvOptions)
|
|
cfg.LoadEnvForStruct(&EnvTest{})
|
|
|
|
_, ok := cfg["target_field_embed"]
|
|
assert.Equal(t, ok, false)
|
|
|
|
os.Setenv("TEST_ENV_FIELD_EMBED", "1234abcd")
|
|
cfg.LoadEnvForStruct(&EnvTest{})
|
|
v := cfg["target_field_embed"]
|
|
assert.Equal(t, v, "1234abcd")
|
|
}
|
|
|
|
func TestLoadEnvForStructWithNamedStructFields(t *testing.T) {
|
|
cfg := make(proxy.EnvOptions)
|
|
cfg.LoadEnvForStruct(&EnvTest{})
|
|
|
|
_, ok := cfg["target_field_named"]
|
|
assert.Equal(t, ok, false)
|
|
|
|
os.Setenv("TEST_ENV_FIELD_NAMED", "1234abcd")
|
|
cfg.LoadEnvForStruct(&EnvTest{})
|
|
v := cfg["target_field_named"]
|
|
assert.Equal(t, v, "1234abcd")
|
|
}
|