forked from TrueCloudLab/lego
70 lines
1.4 KiB
Go
70 lines
1.4 KiB
Go
package lightsail
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/aws/aws-sdk-go-v2/aws"
|
|
awsconfig "github.com/aws/aws-sdk-go-v2/config"
|
|
"github.com/aws/aws-sdk-go-v2/service/lightsail"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestLiveTTL(t *testing.T) {
|
|
if !envTest.IsLiveTest() {
|
|
t.Skip("skipping live test")
|
|
}
|
|
|
|
envTest.RestoreEnv()
|
|
|
|
provider, err := NewDNSProvider()
|
|
require.NoError(t, err)
|
|
|
|
domain := envTest.GetDomain()
|
|
|
|
err = provider.Present(domain, "foo", "bar")
|
|
require.NoError(t, err)
|
|
|
|
// we need a separate Lightsail client here as the one in the DNS provider is unexported.
|
|
fqdn := "_acme-challenge." + domain
|
|
|
|
ctx := context.Background()
|
|
|
|
cfg, err := awsconfig.LoadDefaultConfig(ctx)
|
|
require.NoError(t, err)
|
|
|
|
svc := lightsail.NewFromConfig(cfg)
|
|
require.NoError(t, err)
|
|
|
|
defer func() {
|
|
errC := provider.CleanUp(domain, "foo", "bar")
|
|
if errC != nil {
|
|
t.Log(errC)
|
|
}
|
|
}()
|
|
|
|
params := &lightsail.GetDomainInput{
|
|
DomainName: aws.String(domain),
|
|
}
|
|
|
|
resp, err := svc.GetDomain(ctx, params)
|
|
require.NoError(t, err)
|
|
|
|
entries := resp.Domain.DomainEntries
|
|
for _, entry := range entries {
|
|
if deref(entry.Type) == "TXT" && deref(entry.Name) == fqdn {
|
|
return
|
|
}
|
|
}
|
|
|
|
t.Fatalf("Could not find a TXT record for _acme-challenge.%s", domain)
|
|
}
|
|
|
|
func deref[T string | int | int32 | int64 | bool](v *T) T {
|
|
if v == nil {
|
|
var zero T
|
|
return zero
|
|
}
|
|
|
|
return *v
|
|
}
|