forked from TrueCloudLab/lego
111 lines
2.6 KiB
Go
111 lines
2.6 KiB
Go
package sakuracloud
|
|
|
|
import (
|
|
"os"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/xenolf/lego/acme"
|
|
)
|
|
|
|
var (
|
|
sakuracloudLiveTest bool
|
|
sakuracloudAccessToken string
|
|
sakuracloudAccessSecret string
|
|
sakuracloudDomain string
|
|
)
|
|
|
|
func init() {
|
|
sakuracloudAccessToken = os.Getenv("SAKURACLOUD_ACCESS_TOKEN")
|
|
sakuracloudAccessSecret = os.Getenv("SAKURACLOUD_ACCESS_TOKEN_SECRET")
|
|
sakuracloudDomain = os.Getenv("SAKURACLOUD_DOMAIN")
|
|
|
|
if len(sakuracloudAccessToken) > 0 && len(sakuracloudAccessSecret) > 0 && len(sakuracloudDomain) > 0 {
|
|
sakuracloudLiveTest = true
|
|
}
|
|
}
|
|
|
|
func restoreEnv() {
|
|
os.Setenv("SAKURACLOUD_ACCESS_TOKEN", sakuracloudAccessToken)
|
|
os.Setenv("SAKURACLOUD_ACCESS_TOKEN_SECRET", sakuracloudAccessSecret)
|
|
}
|
|
|
|
//
|
|
// NewDNSProvider
|
|
//
|
|
|
|
func TestNewDNSProviderValid(t *testing.T) {
|
|
defer restoreEnv()
|
|
|
|
os.Setenv("SAKURACLOUD_ACCESS_TOKEN", "123")
|
|
os.Setenv("SAKURACLOUD_ACCESS_TOKEN_SECRET", "456")
|
|
provider, err := NewDNSProvider()
|
|
|
|
assert.NotNil(t, provider)
|
|
assert.Equal(t, acme.UserAgent, provider.client.UserAgent)
|
|
assert.NoError(t, err)
|
|
}
|
|
|
|
func TestNewDNSProviderInvalidWithMissingAccessToken(t *testing.T) {
|
|
defer restoreEnv()
|
|
|
|
os.Setenv("SAKURACLOUD_ACCESS_TOKEN", "")
|
|
provider, err := NewDNSProvider()
|
|
|
|
assert.Nil(t, provider)
|
|
assert.EqualError(t, err, "SakuraCloud: some credentials information are missing: SAKURACLOUD_ACCESS_TOKEN,SAKURACLOUD_ACCESS_TOKEN_SECRET")
|
|
}
|
|
|
|
//
|
|
// NewDNSProviderCredentials
|
|
//
|
|
|
|
func TestNewDNSProviderCredentialsValid(t *testing.T) {
|
|
provider, err := NewDNSProviderCredentials("123", "456")
|
|
|
|
assert.NotNil(t, provider)
|
|
assert.Equal(t, acme.UserAgent, provider.client.UserAgent)
|
|
assert.NoError(t, err)
|
|
}
|
|
|
|
func TestNewDNSProviderCredentialsInvalidWithMissingAccessToken(t *testing.T) {
|
|
provider, err := NewDNSProviderCredentials("", "")
|
|
|
|
assert.Nil(t, provider)
|
|
assert.EqualError(t, err, "SakuraCloud AccessToken is missing")
|
|
}
|
|
|
|
//
|
|
// Present
|
|
//
|
|
|
|
func TestLiveSakuraCloudPresent(t *testing.T) {
|
|
if !sakuracloudLiveTest {
|
|
t.Skip("skipping live test")
|
|
}
|
|
|
|
provider, err := NewDNSProviderCredentials(sakuracloudAccessToken, sakuracloudAccessSecret)
|
|
assert.NoError(t, err)
|
|
|
|
err = provider.Present(sakuracloudDomain, "", "123d==")
|
|
assert.NoError(t, err)
|
|
}
|
|
|
|
//
|
|
// Cleanup
|
|
//
|
|
|
|
func TestLiveSakuraCloudCleanUp(t *testing.T) {
|
|
if !sakuracloudLiveTest {
|
|
t.Skip("skipping live test")
|
|
}
|
|
|
|
time.Sleep(time.Second * 1)
|
|
|
|
provider, err := NewDNSProviderCredentials(sakuracloudAccessToken, sakuracloudAccessSecret)
|
|
assert.NoError(t, err)
|
|
|
|
err = provider.CleanUp(sakuracloudDomain, "", "123d==")
|
|
assert.NoError(t, err)
|
|
}
|