github.com/wuhuizuo/gomplate@v3.5.0+incompatible/vault/auth_test.go (about) 1 package vault 2 3 import ( 4 "os" 5 "testing" 6 7 "github.com/stretchr/testify/assert" 8 ) 9 10 func TestLogin(t *testing.T) { 11 server, v := MockServer(404, "Not Found") 12 defer server.Close() 13 os.Setenv("VAULT_TOKEN", "foo") 14 defer os.Unsetenv("VAULT_TOKEN") 15 v.Login() 16 assert.Equal(t, "foo", v.client.Token()) 17 } 18 19 func TestTokenLogin(t *testing.T) { 20 server, v := MockServer(404, "Not Found") 21 defer server.Close() 22 os.Setenv("VAULT_TOKEN", "foo") 23 defer os.Unsetenv("VAULT_TOKEN") 24 25 token, err := v.TokenLogin() 26 assert.NoError(t, err) 27 assert.Equal(t, "foo", token) 28 }