forked from TrueCloudLab/lego
148 lines
3.2 KiB
Go
148 lines
3.2 KiB
Go
package sakuracloud
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"sync"
|
|
"testing"
|
|
|
|
client "github.com/sacloud/api-client-go"
|
|
"github.com/sacloud/iaas-api-go"
|
|
"github.com/sacloud/iaas-api-go/helper/api"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func setupTest(t *testing.T) {
|
|
t.Helper()
|
|
|
|
t.Setenv("SAKURACLOUD_FAKE_MODE", "1")
|
|
|
|
createDummyZone(t, fakeCaller())
|
|
}
|
|
|
|
func fakeCaller() iaas.APICaller {
|
|
return api.NewCallerWithOptions(&api.CallerOptions{
|
|
Options: &client.Options{
|
|
AccessToken: "dummy",
|
|
AccessTokenSecret: "dummy",
|
|
},
|
|
FakeMode: true,
|
|
})
|
|
}
|
|
|
|
func createDummyZone(t *testing.T, caller iaas.APICaller) {
|
|
t.Helper()
|
|
|
|
ctx := context.Background()
|
|
|
|
dnsOp := iaas.NewDNSOp(caller)
|
|
|
|
// cleanup
|
|
zones, err := dnsOp.Find(ctx, &iaas.FindCondition{})
|
|
require.NoError(t, err)
|
|
|
|
for _, zone := range zones.DNS {
|
|
if zone.Name == "example.com" {
|
|
err = dnsOp.Delete(ctx, zone.ID)
|
|
require.NoError(t, err)
|
|
break
|
|
}
|
|
}
|
|
|
|
// create dummy zone
|
|
_, err = iaas.NewDNSOp(caller).Create(context.Background(), &iaas.DNSCreateRequest{Name: "example.com"})
|
|
require.NoError(t, err)
|
|
}
|
|
|
|
func TestDNSProvider_addAndCleanupRecords(t *testing.T) {
|
|
setupTest(t)
|
|
|
|
config := NewDefaultConfig()
|
|
config.Token = "token1"
|
|
config.Secret = "secret1"
|
|
|
|
p, err := NewDNSProviderConfig(config)
|
|
require.NoError(t, err)
|
|
|
|
t.Run("addTXTRecord", func(t *testing.T) {
|
|
err = p.addTXTRecord("test.example.com", "example.com", "dummyValue", 10)
|
|
require.NoError(t, err)
|
|
|
|
updZone, e := p.getHostedZone("example.com")
|
|
require.NoError(t, e)
|
|
require.NotNil(t, updZone)
|
|
|
|
require.Len(t, updZone.Records, 1)
|
|
})
|
|
|
|
t.Run("cleanupTXTRecord", func(t *testing.T) {
|
|
err = p.cleanupTXTRecord("test.example.com", "example.com", "dummyValue")
|
|
require.NoError(t, err)
|
|
|
|
updZone, e := p.getHostedZone("example.com")
|
|
require.NoError(t, e)
|
|
require.NotNil(t, updZone)
|
|
|
|
require.Len(t, updZone.Records, 0)
|
|
})
|
|
}
|
|
|
|
func TestDNSProvider_concurrentAddAndCleanupRecords(t *testing.T) {
|
|
setupTest(t)
|
|
|
|
dummyRecordCount := 10
|
|
|
|
var providers []*DNSProvider
|
|
for i := 0; i < dummyRecordCount; i++ {
|
|
config := NewDefaultConfig()
|
|
config.Token = "token3"
|
|
config.Secret = "secret3"
|
|
|
|
p, err := NewDNSProviderConfig(config)
|
|
require.NoError(t, err)
|
|
|
|
providers = append(providers, p)
|
|
}
|
|
|
|
var wg sync.WaitGroup
|
|
|
|
t.Run("addTXTRecord", func(t *testing.T) {
|
|
wg.Add(len(providers))
|
|
|
|
for i, p := range providers {
|
|
go func(j int, client *DNSProvider) {
|
|
err := client.addTXTRecord(fmt.Sprintf("test%d.example.com", j), "example.com", "dummyValue", 10)
|
|
require.NoError(t, err)
|
|
wg.Done()
|
|
}(i, p)
|
|
}
|
|
|
|
wg.Wait()
|
|
|
|
updZone, err := providers[0].getHostedZone("example.com")
|
|
require.NoError(t, err)
|
|
require.NotNil(t, updZone)
|
|
|
|
require.Len(t, updZone.Records, dummyRecordCount)
|
|
})
|
|
|
|
t.Run("cleanupTXTRecord", func(t *testing.T) {
|
|
wg.Add(len(providers))
|
|
|
|
for i, p := range providers {
|
|
go func(i int, client *DNSProvider) {
|
|
err := client.cleanupTXTRecord(fmt.Sprintf("test%d.example.com", i), "example.com", "dummyValue")
|
|
require.NoError(t, err)
|
|
wg.Done()
|
|
}(i, p)
|
|
}
|
|
|
|
wg.Wait()
|
|
|
|
updZone, err := providers[0].getHostedZone("example.com")
|
|
require.NoError(t, err)
|
|
require.NotNil(t, updZone)
|
|
|
|
require.Len(t, updZone.Records, 0)
|
|
})
|
|
}
|