Cloudfront implementation

client/client.go

@@ -66,6 +66,7 @@ var allRegions = []string{
 
 type Services struct {
 	Autoscaling      AutoscalingClient
+	Cloudfront       CloudfrontClient
 	Cloudtrail       CloudtrailClient
 	Cloudwatch       CloudwatchClient
 	CloudwatchLogs   CloudwatchLogsClient

client/mocks/builders_test.go

@@ -1,6 +1,8 @@
 package mocks_test
 
 import (
+	"github.com/aws/aws-sdk-go-v2/service/cloudfront"
+	cloudfrontTypes "github.com/aws/aws-sdk-go-v2/service/cloudfront/types"
 	"testing"
 
 	"github.com/aws/aws-sdk-go-v2/service/autoscaling"
@@ -88,6 +90,30 @@ func buildEcsClusterMock(t *testing.T, ctrl *gomock.Controller) client.Services
 	return services
 }
 
+func buildCloudfrontDistributionsMock(t *testing.T, ctrl *gomock.Controller) client.Services {
+	m := mocks.NewMockCloudfrontClient(ctrl)
+	services := client.Services{
+		Cloudfront: m,
+	}
+
+	ds := cloudfrontTypes.DistributionSummary{}
+	err := faker.FakeData(&ds)
+	if err != nil {
+		t.Fatal(err)
+	}
+	cloudfrontOutput := &cloudfront.ListDistributionsOutput{
+		DistributionList: &cloudfrontTypes.DistributionList{
+			Items: []cloudfrontTypes.DistributionSummary{ds},
+		},
+	}
+
+	m.EXPECT().ListDistributions(gomock.Any(), gomock.Any(), gomock.Any()).Return(
+		cloudfrontOutput,
+		nil,
+	)
+	return services
+}
+
 func buildCloudtrailTrailsMock(t *testing.T, ctrl *gomock.Controller) client.Services {
 	m := mocks.NewMockCloudtrailClient(ctrl)
 	services := client.Services{