/
mockDynamoDb.go
40 lines (32 loc) · 1.26 KB
/
mockDynamoDb.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
package mocks
import (
"github.com/aws/aws-sdk-go/aws"
"github.com/aws/aws-sdk-go/aws/request"
"github.com/aws/aws-sdk-go/service/dynamodb"
)
type MockDynamoDB struct {
DeleteItemOutput *dynamodb.DeleteItemOutput
GetItemOutput *dynamodb.GetItemOutput
PutItemOutput *dynamodb.PutItemOutput
}
type MockDynamoDbInputs struct {
DeleteItemWithContext *dynamodb.DeleteItemInput
GetItemWithContext *dynamodb.GetItemInput
PutItemWithContext *dynamodb.PutItemInput
}
var MockDynamoDbCalls = MockDynamoDbInputs{}
func (m *MockDynamoDB) DeleteItemWithContext(ctx aws.Context, input *dynamodb.DeleteItemInput, opts ...request.Option) (*dynamodb.DeleteItemOutput, error) {
MockDynamoDbCalls.DeleteItemWithContext = input
return m.DeleteItemOutput, nil
}
func (m *MockDynamoDB) GetItemWithContext(ctx aws.Context, input *dynamodb.GetItemInput, opts ...request.Option) (*dynamodb.GetItemOutput, error) {
MockDynamoDbCalls.GetItemWithContext = input
return m.GetItemOutput, nil
}
func (m *MockDynamoDB) PutItemWithContext(ctx aws.Context, input *dynamodb.PutItemInput, opts ...request.Option) (*dynamodb.PutItemOutput, error) {
MockDynamoDbCalls.PutItemWithContext = input
return m.PutItemOutput, nil
}
func ResetMockDynamoDbCalls() {
MockDynamoDbCalls = MockDynamoDbInputs{}
}