forked from TrueCloudLab/lego
82 lines
1.9 KiB
Go
82 lines
1.9 KiB
Go
package alidns
|
|
|
|
import (
|
|
"os"
|
|
"testing"
|
|
"time"
|
|
// "time"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
var (
|
|
alidnsLiveTest bool
|
|
alidnsAPIKey string
|
|
alidnsSecretKey string
|
|
alidnsDomain string
|
|
)
|
|
|
|
func init() {
|
|
alidnsAPIKey = os.Getenv("ALICLOUD_ACCESS_KEY")
|
|
alidnsSecretKey = os.Getenv("ALICLOUD_SECRET_KEY")
|
|
alidnsDomain = os.Getenv("ALIDNS_DOMAIN")
|
|
|
|
if len(alidnsAPIKey) > 0 && len(alidnsSecretKey) > 0 && len(alidnsDomain) > 0 {
|
|
alidnsLiveTest = true
|
|
}
|
|
}
|
|
|
|
func restoreEnv() {
|
|
os.Setenv("ALICLOUD_ACCESS_KEY", alidnsAPIKey)
|
|
os.Setenv("ALICLOUD_SECRET_KEY", alidnsSecretKey)
|
|
}
|
|
|
|
func TestNewDNSProviderValid(t *testing.T) {
|
|
defer restoreEnv()
|
|
os.Setenv("ALICLOUD_ACCESS_KEY", "")
|
|
os.Setenv("ALICLOUD_SECRET_KEY", "")
|
|
|
|
_, err := NewDNSProviderCredentials("123", "123", "")
|
|
assert.NoError(t, err)
|
|
}
|
|
|
|
func TestNewDNSProviderValidEnv(t *testing.T) {
|
|
defer restoreEnv()
|
|
os.Setenv("ALICLOUD_ACCESS_KEY", "123")
|
|
os.Setenv("ALICLOUD_SECRET_KEY", "123")
|
|
|
|
_, err := NewDNSProvider()
|
|
assert.NoError(t, err)
|
|
}
|
|
|
|
func TestNewDNSProviderMissingCredErr(t *testing.T) {
|
|
defer restoreEnv()
|
|
os.Setenv("ALICLOUD_ACCESS_KEY", "")
|
|
os.Setenv("ALICLOUD_SECRET_KEY", "")
|
|
|
|
_, err := NewDNSProvider()
|
|
assert.EqualError(t, err, "AliDNS: some credentials information are missing: ALICLOUD_ACCESS_KEY,ALICLOUD_SECRET_KEY")
|
|
}
|
|
|
|
func TestCloudXNSPresent(t *testing.T) {
|
|
if !alidnsLiveTest {
|
|
t.Skip("skipping live test")
|
|
}
|
|
|
|
provider, err := NewDNSProviderCredentials(alidnsAPIKey, alidnsSecretKey, "")
|
|
assert.NoError(t, err)
|
|
|
|
err = provider.Present(alidnsDomain, "", "123d==")
|
|
assert.NoError(t, err)
|
|
}
|
|
|
|
func TestLivednspodCleanUp(t *testing.T) {
|
|
if !alidnsLiveTest {
|
|
t.Skip("skipping live test")
|
|
}
|
|
|
|
time.Sleep(time.Second * 1)
|
|
provider, err := NewDNSProviderCredentials(alidnsAPIKey, alidnsSecretKey, "")
|
|
assert.NoError(t, err)
|
|
err = provider.CleanUp(alidnsDomain, "", "123d==")
|
|
assert.NoError(t, err)
|
|
}
|