27 lines
484 B
Go
27 lines
484 B
Go
|
package config
|
||
|
|
|
||
|
|
import (
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestEnvDefaultsToDev(t *testing.T) {
|
||
|
|
c := Config{}
|
||
|
|
assert.Equal(t, DEV, c.Environment())
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestInvalidEnvReturnsDev(t *testing.T) {
|
||
|
|
c := Config{Env: Env("foobar")}
|
||
|
|
assert.Equal(t, DEV, c.Environment())
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestValidEnvReturnsCorrect(t *testing.T) {
|
||
|
|
for _, e := range Envs {
|
||
|
|
t.Run(string(e), func(t *testing.T) {
|
||
|
|
c := Config{Env: e}
|
||
|
|
assert.Equal(t, e, c.Environment())
|
||
|
|
})
|
||
|
|
}
|
||
|
|
}
|