1
0
mirror of https://github.com/go-acme/lego.git synced 2024-12-23 09:15:11 +02:00
lego/providers/dns/route53/route53_test.go

178 lines
4.3 KiB
Go
Raw Normal View History

2019-03-11 18:56:48 +02:00
package route53
import (
"net/http/httptest"
"os"
"testing"
"time"
"github.com/aws/aws-sdk-go/aws"
"github.com/aws/aws-sdk-go/aws/credentials"
"github.com/aws/aws-sdk-go/aws/session"
"github.com/aws/aws-sdk-go/service/route53"
"github.com/go-acme/lego/v3/platform/tester"
"github.com/stretchr/testify/assert"
2018-10-09 19:03:07 +02:00
"github.com/stretchr/testify/require"
)
var envTest = tester.NewEnvTest(
"AWS_ACCESS_KEY_ID",
"AWS_SECRET_ACCESS_KEY",
"AWS_REGION",
"AWS_HOSTED_ZONE_ID",
"AWS_MAX_RETRIES",
"AWS_TTL",
"AWS_PROPAGATION_TIMEOUT",
"AWS_POLLING_INTERVAL").
WithDomain("R53_DOMAIN").
WithLiveTestRequirements("AWS_ACCESS_KEY_ID", "AWS_SECRET_ACCESS_KEY", "AWS_REGION", "R53_DOMAIN")
func makeTestProvider(ts *httptest.Server) *DNSProvider {
config := &aws.Config{
Credentials: credentials.NewStaticCredentials("abc", "123", " "),
Endpoint: aws.String(ts.URL),
Region: aws.String("mock-region"),
MaxRetries: aws.Int(1),
}
2018-10-09 19:03:07 +02:00
sess, err := session.NewSession(config)
if err != nil {
panic(err)
}
client := route53.New(sess)
cfg := NewDefaultConfig()
return &DNSProvider{client: client, config: cfg}
}
2018-10-09 19:03:07 +02:00
func Test_loadCredentials_FromEnv(t *testing.T) {
defer envTest.RestoreEnv()
envTest.ClearEnv()
os.Setenv("AWS_ACCESS_KEY_ID", "123")
2018-10-09 19:03:07 +02:00
os.Setenv("AWS_SECRET_ACCESS_KEY", "456")
os.Setenv("AWS_REGION", "us-east-1")
config := &aws.Config{
CredentialsChainVerboseErrors: aws.Bool(true),
}
2018-10-09 19:03:07 +02:00
sess, err := session.NewSession(config)
require.NoError(t, err)
value, err := sess.Config.Credentials.Get()
require.NoError(t, err, "Expected credentials to be set from environment")
2018-10-09 19:03:07 +02:00
expected := credentials.Value{
AccessKeyID: "123",
SecretAccessKey: "456",
SessionToken: "",
ProviderName: "EnvConfigCredentials",
}
assert.Equal(t, expected, value)
}
2018-10-09 19:03:07 +02:00
func Test_loadRegion_FromEnv(t *testing.T) {
defer envTest.RestoreEnv()
envTest.ClearEnv()
2018-10-09 19:03:07 +02:00
os.Setenv("AWS_REGION", route53.CloudWatchRegionUsEast1)
2018-10-09 19:03:07 +02:00
sess, err := session.NewSession(aws.NewConfig())
require.NoError(t, err)
region := aws.StringValue(sess.Config.Region)
assert.Equal(t, route53.CloudWatchRegionUsEast1, region, "Region")
}
2018-10-09 19:03:07 +02:00
func Test_getHostedZoneID_FromEnv(t *testing.T) {
defer envTest.RestoreEnv()
envTest.ClearEnv()
2018-10-09 19:03:07 +02:00
expectedZoneID := "zoneID"
os.Setenv("AWS_HOSTED_ZONE_ID", expectedZoneID)
provider, err := NewDNSProvider()
require.NoError(t, err)
2018-10-09 19:03:07 +02:00
hostedZoneID, err := provider.getHostedZoneID("whatever")
require.NoError(t, err, "HostedZoneID")
2018-10-09 19:03:07 +02:00
assert.Equal(t, expectedZoneID, hostedZoneID)
}
2018-10-09 19:03:07 +02:00
func TestNewDefaultConfig(t *testing.T) {
defer envTest.RestoreEnv()
2018-10-09 19:03:07 +02:00
testCases := []struct {
desc string
envVars map[string]string
expected *Config
}{
{
desc: "default configuration",
expected: &Config{
MaxRetries: 5,
TTL: 10,
PropagationTimeout: 2 * time.Minute,
PollingInterval: 4 * time.Second,
},
},
{
desc: "",
envVars: map[string]string{
"AWS_MAX_RETRIES": "10",
"AWS_TTL": "99",
"AWS_PROPAGATION_TIMEOUT": "60",
"AWS_POLLING_INTERVAL": "60",
"AWS_HOSTED_ZONE_ID": "abc123",
},
expected: &Config{
MaxRetries: 10,
TTL: 99,
PropagationTimeout: 60 * time.Second,
PollingInterval: 60 * time.Second,
HostedZoneID: "abc123",
},
},
}
for _, test := range testCases {
t.Run(test.desc, func(t *testing.T) {
envTest.ClearEnv()
2018-10-09 19:03:07 +02:00
for key, value := range test.envVars {
os.Setenv(key, value)
}
config := NewDefaultConfig()
assert.Equal(t, test.expected, config)
})
}
}
func TestDNSProvider_Present(t *testing.T) {
mockResponses := MockResponseMap{
2018-10-09 19:03:07 +02:00
"/2013-04-01/hostedzonesbyname": {StatusCode: 200, Body: ListHostedZonesByNameResponse},
"/2013-04-01/hostedzone/ABCDEFG/rrset/": {StatusCode: 200, Body: ChangeResourceRecordSetsResponse},
"/2013-04-01/change/123456": {StatusCode: 200, Body: GetChangeResponse},
"/2013-04-01/hostedzone/ABCDEFG/rrset?name=_acme-challenge.example.com.&type=TXT": {
StatusCode: 200,
Body: "",
},
}
ts := newMockServer(t, mockResponses)
defer ts.Close()
defer envTest.RestoreEnv()
envTest.ClearEnv()
provider := makeTestProvider(ts)
domain := "example.com"
keyAuth := "123456d=="
err := provider.Present(domain, "", keyAuth)
require.NoError(t, err, "Expected Present to return no error")
}