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
81
82
83
84
85
86
|
// Code generated by smithy-go-codegen DO NOT EDIT.
//go:build snapshot
package marketplaceentitlementservice
import (
"context"
"errors"
"fmt"
"github.com/aws/smithy-go/middleware"
"io"
"io/fs"
"os"
"testing"
)
const ssprefix = "snapshot"
type snapshotOK struct{}
func (snapshotOK) Error() string { return "error: success" }
func createp(path string) (*os.File, error) {
if err := os.Mkdir(ssprefix, 0700); err != nil && !errors.Is(err, fs.ErrExist) {
return nil, err
}
return os.Create(path)
}
func sspath(op string) string {
return fmt.Sprintf("%s/api_op_%s.go.snap", ssprefix, op)
}
func updateSnapshot(stack *middleware.Stack, operation string) error {
f, err := createp(sspath(operation))
if err != nil {
return err
}
defer f.Close()
if _, err := f.Write([]byte(stack.String())); err != nil {
return err
}
return snapshotOK{}
}
func testSnapshot(stack *middleware.Stack, operation string) error {
f, err := os.Open(sspath(operation))
if errors.Is(err, fs.ErrNotExist) {
return snapshotOK{}
}
if err != nil {
return err
}
defer f.Close()
expected, err := io.ReadAll(f)
if err != nil {
return err
}
if actual := stack.String(); actual != string(expected) {
return fmt.Errorf("%s != %s", expected, actual)
}
return snapshotOK{}
}
func TestCheckSnapshot_GetEntitlements(t *testing.T) {
svc := New(Options{})
_, err := svc.GetEntitlements(context.Background(), nil, func(o *Options) {
o.APIOptions = append(o.APIOptions, func(stack *middleware.Stack) error {
return testSnapshot(stack, "GetEntitlements")
})
})
if _, ok := err.(snapshotOK); !ok && err != nil {
t.Fatal(err)
}
}
func TestUpdateSnapshot_GetEntitlements(t *testing.T) {
svc := New(Options{})
_, err := svc.GetEntitlements(context.Background(), nil, func(o *Options) {
o.APIOptions = append(o.APIOptions, func(stack *middleware.Stack) error {
return updateSnapshot(stack, "GetEntitlements")
})
})
if _, ok := err.(snapshotOK); !ok && err != nil {
t.Fatal(err)
}
}
|
