Skip to content

Commit fa6271f

Browse files
committed
Update usage_test.go
1 parent 8302f27 commit fa6271f

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

premium/usage_test.go

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,6 @@ import (
1515
"github.com/aws/aws-sdk-go-v2/aws"
1616
"github.com/aws/aws-sdk-go-v2/service/marketplacemetering"
1717
"github.com/aws/aws-sdk-go-v2/service/marketplacemetering/types"
18-
"github.com/aws/smithy-go"
1918
cqapi "github.com/cloudquery/cloudquery-api-go"
2019
"github.com/cloudquery/cloudquery-api-go/auth"
2120
"github.com/cloudquery/cloudquery-api-go/config"
@@ -370,10 +369,12 @@ func usageMarketplaceDryRunHelper(t *testing.T, m *mocks.MockAWSMarketplaceClien
370369
UsageQuantity: aws.Int32(int32(0)),
371370
DryRun: aws.Bool(true)},
372371
}
373-
errTest := smithy.GenericAPIError{Code: "DryRunOperation", Message: "No errors detected in dry run"}
374-
out := marketplacemetering.MeterUsageOutput{}
375372

376-
return m.EXPECT().MeterUsage(gomock.Any(), inTest).Return(&out, &errTest)
373+
out := marketplacemetering.MeterUsageOutput{
374+
MeteringRecordId: aws.String("DryRunOperation"),
375+
}
376+
377+
return m.EXPECT().MeterUsage(gomock.Any(), inTest).Return(&out, nil)
377378
}
378379

379380
func TestUsageService_AWSMarketplaceDone(t *testing.T) {

0 commit comments

Comments
 (0)