/
interface.go
80 lines (73 loc) · 4.17 KB
/
interface.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
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
// Code generated by private/model/cli/gen-api/main.go. DO NOT EDIT.
// Package sagemakerfeaturestoreruntimeiface provides an interface to enable mocking the Amazon SageMaker Feature Store Runtime service client
// for testing your code.
//
// It is important to note that this interface will have breaking changes
// when the service model is updated and adds new API operations, paginators,
// and waiters.
package sagemakerfeaturestoreruntimeiface
import (
"github.com/aws/aws-sdk-go/aws"
"github.com/aws/aws-sdk-go/aws/request"
"github.com/aws/aws-sdk-go/service/sagemakerfeaturestoreruntime"
)
// SageMakerFeatureStoreRuntimeAPI provides an interface to enable mocking the
// sagemakerfeaturestoreruntime.SageMakerFeatureStoreRuntime service client's API operation,
// paginators, and waiters. This make unit testing your code that calls out
// to the SDK's service client's calls easier.
//
// The best way to use this interface is so the SDK's service client's calls
// can be stubbed out for unit testing your code with the SDK without needing
// to inject custom request handlers into the SDK's request pipeline.
//
// // myFunc uses an SDK service client to make a request to
// // Amazon SageMaker Feature Store Runtime.
// func myFunc(svc sagemakerfeaturestoreruntimeiface.SageMakerFeatureStoreRuntimeAPI) bool {
// // Make svc.BatchGetRecord request
// }
//
// func main() {
// sess := session.New()
// svc := sagemakerfeaturestoreruntime.New(sess)
//
// myFunc(svc)
// }
//
// In your _test.go file:
//
// // Define a mock struct to be used in your unit tests of myFunc.
// type mockSageMakerFeatureStoreRuntimeClient struct {
// sagemakerfeaturestoreruntimeiface.SageMakerFeatureStoreRuntimeAPI
// }
// func (m *mockSageMakerFeatureStoreRuntimeClient) BatchGetRecord(input *sagemakerfeaturestoreruntime.BatchGetRecordInput) (*sagemakerfeaturestoreruntime.BatchGetRecordOutput, error) {
// // mock response/functionality
// }
//
// func TestMyFunc(t *testing.T) {
// // Setup Test
// mockSvc := &mockSageMakerFeatureStoreRuntimeClient{}
//
// myfunc(mockSvc)
//
// // Verify myFunc's functionality
// }
//
// It is important to note that this interface will have breaking changes
// when the service model is updated and adds new API operations, paginators,
// and waiters. Its suggested to use the pattern above for testing, or using
// tooling to generate mocks to satisfy the interfaces.
type SageMakerFeatureStoreRuntimeAPI interface {
BatchGetRecord(*sagemakerfeaturestoreruntime.BatchGetRecordInput) (*sagemakerfeaturestoreruntime.BatchGetRecordOutput, error)
BatchGetRecordWithContext(aws.Context, *sagemakerfeaturestoreruntime.BatchGetRecordInput, ...request.Option) (*sagemakerfeaturestoreruntime.BatchGetRecordOutput, error)
BatchGetRecordRequest(*sagemakerfeaturestoreruntime.BatchGetRecordInput) (*request.Request, *sagemakerfeaturestoreruntime.BatchGetRecordOutput)
DeleteRecord(*sagemakerfeaturestoreruntime.DeleteRecordInput) (*sagemakerfeaturestoreruntime.DeleteRecordOutput, error)
DeleteRecordWithContext(aws.Context, *sagemakerfeaturestoreruntime.DeleteRecordInput, ...request.Option) (*sagemakerfeaturestoreruntime.DeleteRecordOutput, error)
DeleteRecordRequest(*sagemakerfeaturestoreruntime.DeleteRecordInput) (*request.Request, *sagemakerfeaturestoreruntime.DeleteRecordOutput)
GetRecord(*sagemakerfeaturestoreruntime.GetRecordInput) (*sagemakerfeaturestoreruntime.GetRecordOutput, error)
GetRecordWithContext(aws.Context, *sagemakerfeaturestoreruntime.GetRecordInput, ...request.Option) (*sagemakerfeaturestoreruntime.GetRecordOutput, error)
GetRecordRequest(*sagemakerfeaturestoreruntime.GetRecordInput) (*request.Request, *sagemakerfeaturestoreruntime.GetRecordOutput)
PutRecord(*sagemakerfeaturestoreruntime.PutRecordInput) (*sagemakerfeaturestoreruntime.PutRecordOutput, error)
PutRecordWithContext(aws.Context, *sagemakerfeaturestoreruntime.PutRecordInput, ...request.Option) (*sagemakerfeaturestoreruntime.PutRecordOutput, error)
PutRecordRequest(*sagemakerfeaturestoreruntime.PutRecordInput) (*request.Request, *sagemakerfeaturestoreruntime.PutRecordOutput)
}
var _ SageMakerFeatureStoreRuntimeAPI = (*sagemakerfeaturestoreruntime.SageMakerFeatureStoreRuntime)(nil)