summaryrefslogtreecommitdiffstats
path: root/vendor/github.com/xenolf/lego/providers/dns/pdns/pdns_test.go
blob: 70e7670edf4d44c423d1e69c7f300e4aee2cfaad (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
package pdns

import (
	"net/url"
	"os"
	"testing"

	"github.com/stretchr/testify/assert"
)

var (
	pdnsLiveTest bool
	pdnsURL      *url.URL
	pdnsURLStr   string
	pdnsAPIKey   string
	pdnsDomain   string
)

func init() {
	pdnsURLStr = os.Getenv("PDNS_API_URL")
	pdnsURL, _ = url.Parse(pdnsURLStr)
	pdnsAPIKey = os.Getenv("PDNS_API_KEY")
	pdnsDomain = os.Getenv("PDNS_DOMAIN")
	if len(pdnsURLStr) > 0 && len(pdnsAPIKey) > 0 && len(pdnsDomain) > 0 {
		pdnsLiveTest = true
	}
}

func restorePdnsEnv() {
	os.Setenv("PDNS_API_URL", pdnsURLStr)
	os.Setenv("PDNS_API_KEY", pdnsAPIKey)
}

func TestNewDNSProviderValid(t *testing.T) {
	os.Setenv("PDNS_API_URL", "")
	os.Setenv("PDNS_API_KEY", "")
	tmpURL, _ := url.Parse("http://localhost:8081")
	_, err := NewDNSProviderCredentials(tmpURL, "123")
	assert.NoError(t, err)
	restorePdnsEnv()
}

func TestNewDNSProviderValidEnv(t *testing.T) {
	os.Setenv("PDNS_API_URL", "http://localhost:8081")
	os.Setenv("PDNS_API_KEY", "123")
	_, err := NewDNSProvider()
	assert.NoError(t, err)
	restorePdnsEnv()
}

func TestNewDNSProviderMissingHostErr(t *testing.T) {
	os.Setenv("PDNS_API_URL", "")
	os.Setenv("PDNS_API_KEY", "123")
	_, err := NewDNSProvider()
	assert.EqualError(t, err, "PDNS API URL missing")
	restorePdnsEnv()
}

func TestNewDNSProviderMissingKeyErr(t *testing.T) {
	os.Setenv("PDNS_API_URL", pdnsURLStr)
	os.Setenv("PDNS_API_KEY", "")
	_, err := NewDNSProvider()
	assert.EqualError(t, err, "PDNS API key missing")
	restorePdnsEnv()
}

func TestPdnsPresentAndCleanup(t *testing.T) {
	if !pdnsLiveTest {
		t.Skip("skipping live test")
	}

	provider, err := NewDNSProviderCredentials(pdnsURL, pdnsAPIKey)
	assert.NoError(t, err)

	err = provider.Present(pdnsDomain, "", "123d==")
	assert.NoError(t, err)

	err = provider.CleanUp(pdnsDomain, "", "123d==")
	assert.NoError(t, err)
}