93 lines
2.1 KiB
Go
93 lines
2.1 KiB
Go
package middlewares
|
|
|
|
import (
|
|
"testing"
|
|
|
|
awssdk "github.com/aws/aws-sdk-go/aws"
|
|
|
|
"github.com/cloudskiff/driftctl/pkg/resource/aws"
|
|
|
|
"github.com/cloudskiff/driftctl/pkg/resource"
|
|
)
|
|
|
|
func TestDefaultRoute53RecordShouldBeIgnored(t *testing.T) {
|
|
middleware := NewRoute53DefaultZoneRecordSanitizer()
|
|
remoteResources := []resource.Resource{
|
|
&aws.AwsRoute53Zone{},
|
|
&aws.AwsRoute53Record{
|
|
Type: awssdk.String("NS"),
|
|
Id: "123_foobar_NS",
|
|
},
|
|
&aws.AwsRoute53Record{
|
|
Type: awssdk.String("SOA"),
|
|
Id: "123_foobar_SOA",
|
|
},
|
|
&aws.AwsRoute53Record{
|
|
Type: awssdk.String("A"),
|
|
Id: "123_foobar_A",
|
|
},
|
|
}
|
|
stateResources := []resource.Resource{
|
|
&aws.AwsRoute53Record{
|
|
Type: awssdk.String("NS"),
|
|
Id: "456_barfoo_NS",
|
|
},
|
|
}
|
|
err := middleware.Execute(&remoteResources, &stateResources)
|
|
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
if len(remoteResources) != 2 {
|
|
t.Error("Default records were not ignored")
|
|
}
|
|
remainingResource := remoteResources[1].(*aws.AwsRoute53Record)
|
|
if *remainingResource.Type != "A" {
|
|
t.Error("Default record is invalid")
|
|
}
|
|
}
|
|
|
|
func TestDefaultRoute53RecordShouldNotBeIgnoredWhenManaged(t *testing.T) {
|
|
middleware := NewRoute53DefaultZoneRecordSanitizer()
|
|
remoteResources := []resource.Resource{
|
|
&aws.AwsRoute53Zone{},
|
|
&aws.AwsRoute53Record{
|
|
Type: awssdk.String("NS"),
|
|
Id: "123_foobar_NS",
|
|
},
|
|
&aws.AwsRoute53Record{
|
|
Type: awssdk.String("SOA"),
|
|
Id: "123_foobar_SOA",
|
|
},
|
|
&aws.AwsRoute53Record{
|
|
Type: awssdk.String("A"),
|
|
Id: "123_foobar_A",
|
|
},
|
|
}
|
|
stateResources := []resource.Resource{
|
|
&aws.AwsRoute53Record{
|
|
Type: awssdk.String("NS"),
|
|
Id: "123_foobar_NS",
|
|
},
|
|
}
|
|
err := middleware.Execute(&remoteResources, &stateResources)
|
|
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
if len(remoteResources) != 3 {
|
|
t.Error("Default records were not ignored")
|
|
}
|
|
managedDefaultRecord := remoteResources[1].(*aws.AwsRoute53Record)
|
|
if *managedDefaultRecord.Type != "NS" {
|
|
t.Error("Default record is ignored but should not be")
|
|
}
|
|
|
|
ignoredRecord := remoteResources[2].(*aws.AwsRoute53Record)
|
|
if *ignoredRecord.Type != "A" {
|
|
t.Error("Non default record was ignored")
|
|
}
|
|
}
|