forked from TrueCloudLab/lego
47219adc00
Different DNS providers were handling credentials in different ways. Some were reading credential environment variables in cli_handlers.go and then passing them into the NewDNSProvider function, while others were reading the environment variables within their NewDNSProvider functions. This change replaces each DNS challenge's NewDNSProvider function with two new functions: (1) a NewDNSProvider function that takes no parameters and uses the environment to read credentials, and (2) a NewDNSProviderCredentials that takes credentials as parameters.
80 lines
1.8 KiB
Go
80 lines
1.8 KiB
Go
package cloudflare
|
|
|
|
import (
|
|
"os"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
var (
|
|
cflareLiveTest bool
|
|
cflareEmail string
|
|
cflareAPIKey string
|
|
cflareDomain string
|
|
)
|
|
|
|
func init() {
|
|
cflareEmail = os.Getenv("CLOUDFLARE_EMAIL")
|
|
cflareAPIKey = os.Getenv("CLOUDFLARE_API_KEY")
|
|
cflareDomain = os.Getenv("CLOUDFLARE_DOMAIN")
|
|
if len(cflareEmail) > 0 && len(cflareAPIKey) > 0 && len(cflareDomain) > 0 {
|
|
cflareLiveTest = true
|
|
}
|
|
}
|
|
|
|
func restoreCloudFlareEnv() {
|
|
os.Setenv("CLOUDFLARE_EMAIL", cflareEmail)
|
|
os.Setenv("CLOUDFLARE_API_KEY", cflareAPIKey)
|
|
}
|
|
|
|
func TestNewDNSProviderValid(t *testing.T) {
|
|
os.Setenv("CLOUDFLARE_EMAIL", "")
|
|
os.Setenv("CLOUDFLARE_API_KEY", "")
|
|
_, err := NewDNSProviderCredentials("123", "123")
|
|
assert.NoError(t, err)
|
|
restoreCloudFlareEnv()
|
|
}
|
|
|
|
func TestNewDNSProviderValidEnv(t *testing.T) {
|
|
os.Setenv("CLOUDFLARE_EMAIL", "test@example.com")
|
|
os.Setenv("CLOUDFLARE_API_KEY", "123")
|
|
_, err := NewDNSProvider()
|
|
assert.NoError(t, err)
|
|
restoreCloudFlareEnv()
|
|
}
|
|
|
|
func TestNewDNSProviderMissingCredErr(t *testing.T) {
|
|
os.Setenv("CLOUDFLARE_EMAIL", "")
|
|
os.Setenv("CLOUDFLARE_API_KEY", "")
|
|
_, err := NewDNSProvider()
|
|
assert.EqualError(t, err, "CloudFlare credentials missing")
|
|
restoreCloudFlareEnv()
|
|
}
|
|
|
|
func TestCloudFlarePresent(t *testing.T) {
|
|
if !cflareLiveTest {
|
|
t.Skip("skipping live test")
|
|
}
|
|
|
|
provider, err := NewDNSProviderCredentials(cflareEmail, cflareAPIKey)
|
|
assert.NoError(t, err)
|
|
|
|
err = provider.Present(cflareDomain, "", "123d==")
|
|
assert.NoError(t, err)
|
|
}
|
|
|
|
func TestCloudFlareCleanUp(t *testing.T) {
|
|
if !cflareLiveTest {
|
|
t.Skip("skipping live test")
|
|
}
|
|
|
|
time.Sleep(time.Second * 2)
|
|
|
|
provider, err := NewDNSProviderCredentials(cflareEmail, cflareAPIKey)
|
|
assert.NoError(t, err)
|
|
|
|
err = provider.CleanUp(cflareDomain, "", "123d==")
|
|
assert.NoError(t, err)
|
|
}
|