96 lines
2.5 KiB
Go
96 lines
2.5 KiB
Go
package aws_test
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/sirupsen/logrus"
|
|
|
|
"github.com/cloudskiff/driftctl/pkg/analyser"
|
|
awsresources "github.com/cloudskiff/driftctl/pkg/resource/aws"
|
|
"github.com/cloudskiff/driftctl/test"
|
|
|
|
"github.com/r3labs/diff/v2"
|
|
|
|
"github.com/aws/aws-sdk-go/aws"
|
|
|
|
"github.com/aws/aws-sdk-go/service/sqs"
|
|
"github.com/cloudskiff/driftctl/test/acceptance"
|
|
"github.com/cloudskiff/driftctl/test/acceptance/awsutils"
|
|
)
|
|
|
|
func TestAcc_AwsSqsQueue(t *testing.T) {
|
|
var mutatedQueue string
|
|
acceptance.Run(t, acceptance.AccTestCase{
|
|
TerraformVersion: "0.14.9",
|
|
Paths: []string{"./testdata/acc/aws_sqs_queue"},
|
|
Args: []string{"scan", "--filter", "Type=='aws_sqs_queue'"},
|
|
Checks: []acceptance.AccCheck{
|
|
{
|
|
Env: map[string]string{
|
|
"AWS_REGION": "us-east-1",
|
|
},
|
|
PreExec: func() {
|
|
err := acceptance.RetryFor(60*time.Second, func(doneCh chan struct{}) error {
|
|
return sqs.New(awsutils.Session()).ListQueuesPages(&sqs.ListQueuesInput{},
|
|
func(resp *sqs.ListQueuesOutput, lastPage bool) bool {
|
|
logrus.Debugf("Retrieved %d SQS queues", len(resp.QueueUrls))
|
|
if len(resp.QueueUrls) == 2 {
|
|
doneCh <- struct{}{}
|
|
}
|
|
return !lastPage
|
|
},
|
|
)
|
|
})
|
|
if err != nil {
|
|
t.Fatal("Timeout while fetching SQS queues")
|
|
}
|
|
},
|
|
Check: func(result *test.ScanResult, stdout string, err error) {
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
result.AssertInfrastructureIsInSync()
|
|
result.Equal(2, result.Summary().TotalManaged)
|
|
mutatedQueue = result.Managed()[0].TerraformId()
|
|
},
|
|
},
|
|
{
|
|
Env: map[string]string{
|
|
"AWS_REGION": "us-east-1",
|
|
},
|
|
PreExec: func() {
|
|
client := sqs.New(awsutils.Session())
|
|
attributes := make(map[string]*string)
|
|
attributes["DelaySeconds"] = aws.String("200")
|
|
_, err := client.SetQueueAttributes(&sqs.SetQueueAttributesInput{
|
|
Attributes: attributes,
|
|
QueueUrl: aws.String(mutatedQueue),
|
|
})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
},
|
|
Check: func(result *test.ScanResult, stdout string, err error) {
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
result.AssertDriftCountTotal(1)
|
|
result.AssertResourceHasDrift(
|
|
mutatedQueue,
|
|
awsresources.AwsSqsQueueResourceType,
|
|
analyser.Change{
|
|
Change: diff.Change{
|
|
Type: diff.UPDATE,
|
|
Path: []string{"DelaySeconds"},
|
|
From: float64(0),
|
|
To: float64(200),
|
|
},
|
|
},
|
|
)
|
|
},
|
|
},
|
|
},
|
|
})
|
|
}
|