lego/providers/dns/dns_providers_test.go

package dns

import (
	"os"
	"testing"

	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
	"github.com/xenolf/lego/providers/dns/exoscale"
)

var (
	apiKey    string
	apiSecret string
)

func init() {
	apiSecret = os.Getenv("EXOSCALE_API_SECRET")
	apiKey = os.Getenv("EXOSCALE_API_KEY")
}

func restoreExoscaleEnv() {
	os.Setenv("EXOSCALE_API_KEY", apiKey)
	os.Setenv("EXOSCALE_API_SECRET", apiSecret)
}

func TestKnownDNSProviderSuccess(t *testing.T) {
	defer restoreExoscaleEnv()
	os.Setenv("EXOSCALE_API_KEY", "abc")
	os.Setenv("EXOSCALE_API_SECRET", "123")

	provider, err := NewDNSChallengeProviderByName("exoscale")
	require.NoError(t, err)
	assert.NotNil(t, provider)

	assert.IsType(t, &exoscale.DNSProvider{}, provider, "Not loaded correct DNS provider")
}

func TestKnownDNSProviderError(t *testing.T) {
	defer restoreExoscaleEnv()
	os.Setenv("EXOSCALE_API_KEY", "")
	os.Setenv("EXOSCALE_API_SECRET", "")

	_, err := NewDNSChallengeProviderByName("exoscale")
	assert.Error(t, err)
}

func TestUnknownDNSProvider(t *testing.T) {
	_, err := NewDNSChallengeProviderByName("foobar")
	assert.Error(t, err)
}