-
Notifications
You must be signed in to change notification settings - Fork 0
/
narcissus_test.go
105 lines (91 loc) · 2.73 KB
/
narcissus_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
package narcissus
import (
"fmt"
"testing"
"github.com/stretchr/testify/assert"
"github.com/aws/aws-sdk-go/service/ssm"
"github.com/aws/aws-sdk-go/service/ssm/ssmiface"
)
type mockedGetParameter struct {
ssmiface.SSMAPI
MockData map[string]string
}
func (m mockedGetParameter) GetParameter(input *ssm.GetParameterInput) (*ssm.GetParameterOutput, error) {
mockValue := m.MockData[*input.Name]
parameterOutput := ssm.GetParameterOutput{
Parameter: &ssm.Parameter{
Value: &mockValue,
},
}
return ¶meterOutput, nil
}
var wrapper = Wrapper{
Client: mockedGetParameter{
MockData: map[string]string{
"/path/to/parameters/Name/FirstName": "Jane",
"/path/to/parameters/Name/LastName": "Doe",
"/path/to/parameters/Contact/Email": "jane.doe@email.com",
"/path/to/parameters/Contact/Number": "(123)456-7890",
"/path/to/parameters/FavoriteNumber": "26",
"/path/to/parameters/FavoriteInconvenientNumber": "26.24",
},
},
}
func TestUpdateBySSM(t *testing.T) {
type Name struct {
FirstName string `ssm:"Name/FirstName"`
LastName string `ssm:"Name/LastName"`
}
type Contact struct {
Email string `ssm:"Contact/Email"`
Number string `ssm:"Contact/Number"`
}
type Person struct {
Name Name
Contact Contact
FavoriteNumber int `ssm:"FavoriteNumber"`
FavoriteInconvenientNumber float64 `ssm:"FavoriteInconvenientNumber"`
}
person := Person{}
ssmPath := "/path/to/parameters/"
err := wrapper.UpdateBySSM(&person, &ssmPath)
if err != nil {
t.Log(err)
}
expectedPerson := Person{
Name: Name{
FirstName: "Jane",
LastName: "Doe",
},
Contact: Contact{
Email: "jane.doe@email.com",
Number: "(123)456-7890",
},
FavoriteNumber: 26,
FavoriteInconvenientNumber: 26.24,
}
assert.Equal(t, person, expectedPerson, "Updated incorrectly by SSM")
}
func ExampleUpdateBySSM() {
type Name struct {
FirstName string `ssm:"Name/FirstName"`
LastName string `ssm:"Name/LastName"`
}
type Contact struct {
Email string `ssm:"Contact/Email"`
Number string `ssm:"Contact/Number"`
}
type Person struct {
Name Name
Contact Contact
FavoriteNumber int `ssm:"FavoriteNumber"`
FavoriteInconvenientNumber float64 `ssm:"FavoriteInconvenientNumber"`
}
person := Person{}
ssmPath := "/path/to/parameters/"
// You can get this wrapper like so: wrapper := narcissus.Wrapper{Client: client}
_ = wrapper.UpdateBySSM(&person, &ssmPath)
// If you don't want to reuse a client, simply use narcissus.UpdateBySSM(&person, &ssmPath)
fmt.Println(person)
// Output: {{Jane Doe} {jane.doe@email.com (123)456-7890} 26 26.24}
}