forked from mweagle/Sparta
-
Notifications
You must be signed in to change notification settings - Fork 0
/
discovery_test.go
112 lines (103 loc) · 4.05 KB
/
discovery_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
package sparta
import (
"encoding/json"
"testing"
)
const discoveryDataTags = `
{
"SESMessageStoreBucketa622fdfda5789d596c08c79124f12b978b3da772": {
"DomainName": "spartaapplication-sesmessagestorebucketa622fdfda5-1b8t1fol64if3.s3.amazonaws.com",
"Ref": "spartaapplication-sesmessagestorebucketa622fdfda5-1b8t1fol64if3",
"Tags": [
{
"Key": "sparta:logicalBucketName",
"Value": "Special"
}
],
"Type": "AWS::S3::Bucket",
"WebsiteURL": "http://spartaapplication-sesmessagestorebucketa622fdfda5-1b8t1fol64if3.s3-website-us-west-2.amazonaws.com"
},
"aws:cloudformation:stack-id": "arn:aws:cloudformation:us-west-2:123412341234:stack/SpartaApplication/c25fab50-c904-11e5-acca-503f20f2ade6",
"aws:cloudformation:stack-name": "SpartaApplication",
"golangFunc": "main.echoSESEvent",
"sparta:cloudformation:region": "us-west-2"
}
`
var discoveryDataNoTags = `
{
"S3DynamicBucket5de4436284814c262da3b904c1f3fc73b23cea00": {
"DomainName": "spartaapplication-s3dynamicbucket5de4436284814c26-ll4cejoliisg.s3.amazonaws.com",
"Ref": "spartaapplication-s3dynamicbucket5de4436284814c26-ll4cejoliisg",
"Type": "AWS::S3::Bucket",
"WebsiteURL": "http://spartaapplication-s3dynamicbucket5de4436284814c26-ll4cejoliisg.s3-website-us-west-2.amazonaws.com"
},
"aws:cloudformation:stack-id": "arn:aws:cloudformation:us-west-2:123412341234:stack/SpartaApplication/c25fab50-c904-11e5-acca-503f20f2ade6",
"aws:cloudformation:stack-name": "SpartaApplication",
"golangFunc": "main.echoS3DynamicBucketEvent",
"sparta:cloudformation:region": "us-west-2"
}
`
func TestDiscoveryInitialized(t *testing.T) {
// Ensure that sparta.Discover() can only be called from a lambda function
logger, _ := NewLogger("warning")
initializeDiscovery("DiscoveryTest", testLambdaData(), logger)
configuration, err := Discover()
t.Logf("Configuration: %#v", configuration)
t.Logf("Error: %#v", err)
if err == nil {
t.Errorf("sparta.Discover() failed to reject invalid call site")
}
t.Logf("Properly rejected invalid callsite: %s", err.Error())
}
func TestDiscoveryNotInitialized(t *testing.T) {
// Ensure that sparta.Discover() can only be called from a lambda function
configuration, err := Discover()
t.Logf("Configuration: %#v", configuration)
t.Logf("Error: %#v", err)
if err == nil {
t.Errorf("sparta.Discover() failed to error when not initialized")
}
t.Logf("Properly rejected invalid callsite: %s", err.Error())
}
func TestDiscoveryUnmarshalTags(t *testing.T) {
// Ensure that sparta.Discover() can only be called from a lambda function
var info DiscoveryInfo
err := json.Unmarshal([]byte(discoveryDataTags), &info)
if nil != err {
t.Errorf("Failed to unmarshal discovery data with tags")
}
if 1 != len(info.Resources) {
t.Errorf("Failed to unmarshal single resource")
}
if 1 != len(info.Resources["SESMessageStoreBucketa622fdfda5789d596c08c79124f12b978b3da772"].Tags) {
t.Errorf("Invalid Tags unmarshaled resource with single tag")
}
if info.Resources["SESMessageStoreBucketa622fdfda5789d596c08c79124f12b978b3da772"].Tags["sparta:logicalBucketName"] != "Special" {
t.Errorf("Failed to unmarshal `sparta:logicalBucketName` tag")
}
t.Logf("Discovery Info: %#v", info)
}
func TestDiscoveryUnmarshalNoTags(t *testing.T) {
// Ensure that sparta.Discover() can only be called from a lambda function
var info DiscoveryInfo
err := json.Unmarshal([]byte(discoveryDataNoTags), &info)
if nil != err {
t.Errorf("Failed to unmarshal discovery data without tags")
}
if 1 != len(info.Resources) {
t.Errorf("Failed to unmarshal single resource")
}
if 0 != len(info.Resources["S3DynamicBucket5de4436284814c262da3b904c1f3fc73b23cea00"].Tags) {
t.Errorf("Invalid Tags unmarshaled for tagless resource")
}
t.Logf("Discovery Info: %#v", info)
}
func TestDiscoveryEmptyMetadata(t *testing.T) {
// Ensure that sparta.Discover() can only be called from a lambda function
var info DiscoveryInfo
err := json.Unmarshal([]byte("{}"), &info)
if nil != err {
t.Errorf("Failed to unmarshal empty discovery data")
}
t.Logf("Discovery Info: %#v", info)
}