-
Notifications
You must be signed in to change notification settings - Fork 1
/
secretsmanager_test.go
121 lines (115 loc) · 2.81 KB
/
secretsmanager_test.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
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
package secretsmanager
import (
"testing"
"github.com/aws/aws-sdk-go/aws"
"github.com/aws/aws-sdk-go/service/secretsmanager"
"github.com/aws/aws-sdk-go/service/secretsmanager/secretsmanageriface"
)
type mockedSecret struct {
secretsmanageriface.SecretsManagerAPI
Resp secretsmanager.GetSecretValueOutput
}
// GetSecret return mocked secret value
func (m mockedSecret) GetSecretValue(in *secretsmanager.GetSecretValueInput) (*secretsmanager.GetSecretValueOutput, error) {
return &m.Resp, nil
}
func TestGetSecret(t *testing.T) {
testCases := []struct {
arg *SecretSpec
mock secretsmanager.GetSecretValueOutput
want string
ok bool
}{
{
arg: &SecretSpec{Name: "test/secret"},
mock: secretsmanager.GetSecretValueOutput{
Name: aws.String("test/secret"),
SecretString: aws.String(`{"key": "test"}`),
},
want: "test",
ok: true,
},
{
arg: &SecretSpec{
Name: "test/secret",
Key: "key",
},
mock: secretsmanager.GetSecretValueOutput{
Name: aws.String("test/secret"),
SecretString: aws.String(`{"key": "test"}`),
},
want: "test",
ok: true,
},
{
arg: &SecretSpec{
Name: "test/secret",
Key: "second_key",
},
mock: secretsmanager.GetSecretValueOutput{
Name: aws.String("test/secret"),
SecretString: aws.String(`{"first_key": "first_val", "second_key": "second_val"}`),
},
want: "second_val",
ok: true,
},
{
arg: &SecretSpec{
Name: "test/secret",
},
mock: secretsmanager.GetSecretValueOutput{
Name: aws.String("test/secret"),
SecretString: aws.String(`{"first_key": "first_val", "second_key": "second_val"}`),
},
want: "first_val",
ok: true,
},
{
arg: &SecretSpec{
Name: "test/secret",
Key: "nonexistent",
},
mock: secretsmanager.GetSecretValueOutput{
Name: aws.String("test/secret"),
SecretString: aws.String(`{"key": "test"}`),
},
ok: false,
},
{
arg: &SecretSpec{
Name: "test/secret",
Key: "nonexistent",
},
mock: secretsmanager.GetSecretValueOutput{
Name: aws.String("test/secret"),
SecretString: aws.String(`{"first_key": "first_val", "second_key": "second_val"}`),
},
ok: false,
},
{
arg: &SecretSpec{
Name: "test/secret",
Key: "nonexistent",
},
mock: secretsmanager.GetSecretValueOutput{},
ok: false,
},
}
for _, test := range testCases {
c := &Client{
api: mockedSecret{Resp: test.mock},
}
got, err := c.GetSecret(test.arg)
if test.ok {
if got != test.want {
t.Fatalf("want %v, got %v, error %v, using arg %v", test.want, got, err, test.arg)
}
}
if !test.ok {
if err == nil {
t.Fatalf("error expected but got %q, using arg %v", err, test.arg)
}
}
t.Logf("arg (%v), want %v, got %v, err %v", test.arg, test.want, got, err)
}
}