Skip to content
This repository has been archived by the owner on Aug 16, 2022. It is now read-only.

Commit

Permalink
Move test
Browse files Browse the repository at this point in the history
  • Loading branch information
spangenberg committed Nov 25, 2021
1 parent ed7b269 commit 5bd3373
Show file tree
Hide file tree
Showing 3 changed files with 15 additions and 39 deletions.
26 changes: 0 additions & 26 deletions client/mocks/builders_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -12,8 +12,6 @@ import (
cloudwatchTypes "github.com/aws/aws-sdk-go-v2/service/cloudwatch/types"
"github.com/aws/aws-sdk-go-v2/service/cloudwatchlogs"
cloudwatchlogsTypes "github.com/aws/aws-sdk-go-v2/service/cloudwatchlogs/types"
"github.com/aws/aws-sdk-go-v2/service/configservice"
configserviceTypes "github.com/aws/aws-sdk-go-v2/service/configservice/types"
"github.com/aws/aws-sdk-go-v2/service/directconnect"
directconnectTypes "github.com/aws/aws-sdk-go-v2/service/directconnect/types"
"github.com/aws/aws-sdk-go-v2/service/ec2"
Expand Down Expand Up @@ -226,30 +224,6 @@ func buildCloudwatchLogsFiltersMock(t *testing.T, ctrl *gomock.Controller) clien
}
}

func buildConfigConfigurationRecordersMock(t *testing.T, ctrl *gomock.Controller) client.Services {
m := mocks.NewMockConfigServiceClient(ctrl)
l := configserviceTypes.ConfigurationRecorder{}
if err := faker.FakeData(&l); err != nil {
t.Fatal(err)
}
sl := configserviceTypes.ConfigurationRecorderStatus{}
if err := faker.FakeData(&sl); err != nil {
t.Fatal(err)
}
sl.Name = l.Name
m.EXPECT().DescribeConfigurationRecorderStatus(gomock.Any(), gomock.Any(), gomock.Any()).Return(
&configservice.DescribeConfigurationRecorderStatusOutput{
ConfigurationRecordersStatus: []configserviceTypes.ConfigurationRecorderStatus{sl},
}, nil)
m.EXPECT().DescribeConfigurationRecorders(gomock.Any(), gomock.Any(), gomock.Any()).Return(
&configservice.DescribeConfigurationRecordersOutput{
ConfigurationRecorders: []configserviceTypes.ConfigurationRecorder{l},
}, nil)
return client.Services{
ConfigService: m,
}
}

func buildDirectconnectGatewaysMock(t *testing.T, ctrl *gomock.Controller) client.Services {
m := mocks.NewMockDirectconnectClient(ctrl)
l := directconnectTypes.DirectConnectGateway{}
Expand Down
5 changes: 0 additions & 5 deletions client/mocks/resources_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -65,11 +65,6 @@ func TestResources(t *testing.T) {
mockBuilder: buildCloudwatchLogsFiltersMock,
mainTable: resources.CloudwatchlogsFilters(),
},
{
resource: "config.configuration_recorders",
mockBuilder: buildConfigConfigurationRecordersMock,
mainTable: resources.ConfigConfigurationRecorders(),
},
{
resource: "directconnect.gateways",
mockBuilder: buildDirectconnectGatewaysMock,
Expand Down
23 changes: 15 additions & 8 deletions resources/config_configuration_recorders_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -5,25 +5,32 @@ import (

"github.com/aws/aws-sdk-go-v2/service/configservice"
"github.com/aws/aws-sdk-go-v2/service/configservice/types"
"github.com/cloudquery/cq-provider-aws/client"
"github.com/cloudquery/cq-provider-aws/client/mocks"
"github.com/cloudquery/faker/v3"
"github.com/golang/mock/gomock"

"github.com/cloudquery/cq-provider-aws/client"
"github.com/cloudquery/cq-provider-aws/client/mocks"
)

func buildConfigConfigurationRecorders(t *testing.T, ctrl *gomock.Controller) client.Services {
m := mocks.NewMockConfigServiceClient(ctrl)

cr := types.ConfigurationRecorder{}
err := faker.FakeData(&cr)
if err != nil {
l := types.ConfigurationRecorder{}
if err := faker.FakeData(&l); err != nil {
t.Fatal(err)
}
sl := types.ConfigurationRecorderStatus{}
if err := faker.FakeData(&sl); err != nil {
t.Fatal(err)
}
sl.Name = l.Name
m.EXPECT().DescribeConfigurationRecorderStatus(gomock.Any(), gomock.Any(), gomock.Any()).Return(
&configservice.DescribeConfigurationRecorderStatusOutput{
ConfigurationRecordersStatus: []types.ConfigurationRecorderStatus{sl},
}, nil)
m.EXPECT().DescribeConfigurationRecorders(gomock.Any(), gomock.Any(), gomock.Any()).Return(
&configservice.DescribeConfigurationRecordersOutput{
ConfigurationRecorders: []types.ConfigurationRecorder{cr},
ConfigurationRecorders: []types.ConfigurationRecorder{l},
}, nil)

return client.Services{
ConfigService: m,
}
Expand Down

0 comments on commit 5bd3373

Please sign in to comment.