lego/providers/dns/vultr/vultr_test.go

139 lines
2.5 KiB
Go
Raw Normal View History

2016-03-20 23:23:37 +00:00
package vultr
import (
"os"
"testing"
"time"
"github.com/stretchr/testify/require"
2016-03-20 23:23:37 +00:00
)
var (
liveTest bool
envTestAPIKey string
envTestDomain string
2016-03-20 23:23:37 +00:00
)
func init() {
envTestAPIKey = os.Getenv("VULTR_API_KEY")
envTestDomain = os.Getenv("VULTR_TEST_DOMAIN")
liveTest = len(envTestAPIKey) > 0 && len(envTestDomain) > 0
2016-03-20 23:23:37 +00:00
}
func restoreEnv() {
os.Setenv("VULTR_API_KEY", envTestAPIKey)
2016-03-20 23:23:37 +00:00
}
func TestNewDNSProvider(t *testing.T) {
testCases := []struct {
desc string
envVars map[string]string
expected string
}{
{
desc: "success",
envVars: map[string]string{
"VULTR_API_KEY": "123",
},
},
{
desc: "missing api key",
envVars: map[string]string{
"VULTR_API_KEY": "",
},
expected: "vultr: some credentials information are missing: VULTR_API_KEY",
},
}
for _, test := range testCases {
t.Run(test.desc, func(t *testing.T) {
defer restoreEnv()
for key, value := range test.envVars {
if len(value) == 0 {
os.Unsetenv(key)
} else {
os.Setenv(key, value)
}
}
p, err := NewDNSProvider()
if len(test.expected) == 0 {
require.NoError(t, err)
require.NotNil(t, p)
require.NotNil(t, p.config)
require.NotNil(t, p.client)
} else {
require.EqualError(t, err, test.expected)
}
})
}
2016-03-20 23:23:37 +00:00
}
func TestNewDNSProviderConfig(t *testing.T) {
testCases := []struct {
desc string
apiKey string
expected string
}{
{
desc: "success",
apiKey: "123",
},
{
desc: "missing credentials",
expected: "vultr: credentials missing",
},
}
for _, test := range testCases {
t.Run(test.desc, func(t *testing.T) {
defer restoreEnv()
os.Unsetenv("VULTR_API_KEY")
config := NewDefaultConfig()
config.APIKey = test.apiKey
p, err := NewDNSProviderConfig(config)
if len(test.expected) == 0 {
require.NoError(t, err)
require.NotNil(t, p)
require.NotNil(t, p.config)
require.NotNil(t, p.client)
} else {
require.EqualError(t, err, test.expected)
}
})
}
2016-03-20 23:23:37 +00:00
}
func TestLivePresent(t *testing.T) {
if !liveTest {
t.Skip("skipping live test")
}
restoreEnv()
2016-03-20 23:23:37 +00:00
provider, err := NewDNSProvider()
require.NoError(t, err)
2016-03-20 23:23:37 +00:00
err = provider.Present(envTestDomain, "", "123d==")
require.NoError(t, err)
2016-03-20 23:23:37 +00:00
}
func TestLiveCleanUp(t *testing.T) {
if !liveTest {
t.Skip("skipping live test")
}
restoreEnv()
2016-03-20 23:23:37 +00:00
provider, err := NewDNSProvider()
require.NoError(t, err)
2016-03-20 23:23:37 +00:00
time.Sleep(1 * time.Second)
err = provider.CleanUp(envTestDomain, "", "123d==")
require.NoError(t, err)
2016-03-20 23:23:37 +00:00
}