driftctl/pkg/middlewares/default_vpc_test.go

64 lines
1.2 KiB
Go

package middlewares
import (
"testing"
"github.com/snyk/driftctl/pkg/resource/aws"
"github.com/snyk/driftctl/enumeration/resource"
)
func TestAwsDefaultVPCShouldBeIgnored(t *testing.T) {
middleware := NewAwsDefaultVPC()
remoteResources := []*resource.Resource{
{
Id: "foobar",
Type: aws.AwsDefaultVpcResourceType,
Attrs: &resource.Attributes{
"id": "foobar",
},
},
}
stateResources := []*resource.Resource{}
err := middleware.Execute(&remoteResources, &stateResources)
if err != nil {
t.Error(err)
}
if len(remoteResources) != 0 {
t.Error("Default VPC was not ignored")
}
}
func TestAwsDefaultVPCShouldNotBeIgnoredWhenManaged(t *testing.T) {
middleware := NewAwsDefaultVPC()
remoteResources := []*resource.Resource{
{
Id: "foobar",
Type: aws.AwsDefaultVpcResourceType,
Attrs: &resource.Attributes{
"id": "foobar",
},
},
}
stateResources := []*resource.Resource{
{
Id: "foobar",
Type: aws.AwsDefaultVpcResourceType,
Attrs: &resource.Attributes{
"id": "foobar",
},
},
}
err := middleware.Execute(&remoteResources, &stateResources)
if err != nil {
t.Error(err)
}
if len(remoteResources) != 1 {
t.Error("Default VPC was ignored")
}
}