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  }