summaryrefslogtreecommitdiffstats
path: root/vendor/github.com/goamz/goamz/cloudwatch/cloudwatch_test.go
blob: a4271f1eae4d6727464d8e3e2958bb8a099e6b1e (plain)
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
122
123
124
125
126
127
128
129
130
131
132
package cloudwatch_test

import (
	"testing"

	"github.com/goamz/goamz/aws"
	"github.com/goamz/goamz/cloudwatch"
	"github.com/goamz/goamz/testutil"
	. "gopkg.in/check.v1"
)

func Test(t *testing.T) {
	TestingT(t)
}

type S struct {
	cw *cloudwatch.CloudWatch
}

var _ = Suite(&S{})

var testServer = testutil.NewHTTPServer()

func (s *S) SetUpSuite(c *C) {
	testServer.Start()
	auth := aws.Auth{AccessKey: "abc", SecretKey: "123"}
	s.cw, _ = cloudwatch.NewCloudWatch(auth, aws.ServiceInfo{testServer.URL, aws.V2Signature})
}

func (s *S) TearDownTest(c *C) {
	testServer.Flush()
}

func getTestAlarm() *cloudwatch.MetricAlarm {
	alarm := new(cloudwatch.MetricAlarm)

	alarm.AlarmName = "TestAlarm"
	alarm.MetricName = "TestMetric"
	alarm.Namespace = "TestNamespace"
	alarm.ComparisonOperator = "LessThanThreshold"
	alarm.Threshold = 1
	alarm.EvaluationPeriods = 5
	alarm.Period = 60
	alarm.Statistic = "Sum"

	return alarm
}

func (s *S) TestPutAlarm(c *C) {
	testServer.Response(200, nil, "<RequestId>123</RequestId>")

	alarm := getTestAlarm()

	_, err := s.cw.PutMetricAlarm(alarm)
	c.Assert(err, IsNil)

	req := testServer.WaitRequest()
	c.Assert(req.Method, Equals, "POST")
	c.Assert(req.URL.Path, Equals, "/")
	c.Assert(req.Form["Action"], DeepEquals, []string{"PutMetricAlarm"})
	c.Assert(req.Form["AlarmName"], DeepEquals, []string{"TestAlarm"})
	c.Assert(req.Form["ComparisonOperator"], DeepEquals, []string{"LessThanThreshold"})
	c.Assert(req.Form["EvaluationPeriods"], DeepEquals, []string{"5"})
	c.Assert(req.Form["Threshold"], DeepEquals, []string{"1.0000000000E+00"})
	c.Assert(req.Form["Period"], DeepEquals, []string{"60"})
	c.Assert(req.Form["Statistic"], DeepEquals, []string{"Sum"})
}

func (s *S) TestPutAlarmWithAction(c *C) {
	testServer.Response(200, nil, "<RequestId>123</RequestId>")

	alarm := getTestAlarm()

	alarm.AlarmActions = []cloudwatch.AlarmAction{
		cloudwatch.AlarmAction{
			ARN: "123",
		},
	}

	alarm.OkActions = []cloudwatch.AlarmAction{
		cloudwatch.AlarmAction{
			ARN: "456",
		},
	}

	alarm.InsufficientDataActions = []cloudwatch.AlarmAction{
		cloudwatch.AlarmAction{
			ARN: "789",
		},
	}

	_, err := s.cw.PutMetricAlarm(alarm)
	c.Assert(err, IsNil)

	req := testServer.WaitRequest()
	c.Assert(req.Method, Equals, "POST")
	c.Assert(req.URL.Path, Equals, "/")
	c.Assert(req.Form["Action"], DeepEquals, []string{"PutMetricAlarm"})
	c.Assert(req.Form["AlarmActions.member.1"], DeepEquals, []string{"123"})
	c.Assert(req.Form["OKActions.member.1"], DeepEquals, []string{"456"})
	c.Assert(req.Form["InsufficientDataActions.member.1"], DeepEquals, []string{"789"})
	c.Assert(req.Form["AlarmName"], DeepEquals, []string{"TestAlarm"})
	c.Assert(req.Form["ComparisonOperator"], DeepEquals, []string{"LessThanThreshold"})
	c.Assert(req.Form["EvaluationPeriods"], DeepEquals, []string{"5"})
	c.Assert(req.Form["Threshold"], DeepEquals, []string{"1.0000000000E+00"})
	c.Assert(req.Form["Period"], DeepEquals, []string{"60"})
	c.Assert(req.Form["Statistic"], DeepEquals, []string{"Sum"})
}

func (s *S) TestPutAlarmInvalidComapirsonOperator(c *C) {
	testServer.Response(200, nil, "<RequestId>123</RequestId>")

	alarm := getTestAlarm()

	alarm.ComparisonOperator = "LessThan"

	_, err := s.cw.PutMetricAlarm(alarm)
	c.Assert(err, NotNil)
	c.Assert(err.Error(), Equals, "ComparisonOperator is not valid")
}

func (s *S) TestPutAlarmInvalidStatistic(c *C) {
	testServer.Response(200, nil, "<RequestId>123</RequestId>")

	alarm := getTestAlarm()

	alarm.Statistic = "Count"

	_, err := s.cw.PutMetricAlarm(alarm)
	c.Assert(err, NotNil)
	c.Assert(err.Error(), Equals, "Invalid statistic value supplied")
}