/
testutils.go
70 lines (60 loc) · 1.54 KB
/
testutils.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
package aws
import (
"fmt"
"net/http"
"net/http/httptest"
"net/url"
"github.com/aws/aws-sdk-go/service/ec2"
)
// MockServer -
func MockServer(code int, body string) (*httptest.Server, *Ec2Meta) {
server, httpClient := MockHTTPServer(code, body)
client := &Ec2Meta{server.URL + "/", httpClient, false, make(map[string]string), ClientOptions{}}
return server, client
}
// MockHTTPServer -
func MockHTTPServer(code int, body string) (*httptest.Server, *http.Client) {
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
w.WriteHeader(code)
// nolint: errcheck
fmt.Fprintln(w, body)
}))
tr := &http.Transport{
Proxy: func(req *http.Request) (*url.URL, error) {
return url.Parse(server.URL)
},
}
httpClient := &http.Client{Transport: tr}
return server, httpClient
}
// NewDummyEc2Info -
func NewDummyEc2Info(metaClient *Ec2Meta) *Ec2Info {
i := &Ec2Info{
metaClient: metaClient,
describer: func() (InstanceDescriber, error) { return DummyInstanceDescriber{}, nil },
}
return i
}
// NewDummyEc2Meta -
func NewDummyEc2Meta() *Ec2Meta {
return &Ec2Meta{nonAWS: true}
}
// DummyInstanceDescriber - test doubles
type DummyInstanceDescriber struct {
tags []*ec2.Tag
}
// DescribeInstances -
func (d DummyInstanceDescriber) DescribeInstances(*ec2.DescribeInstancesInput) (*ec2.DescribeInstancesOutput, error) {
output := &ec2.DescribeInstancesOutput{
Reservations: []*ec2.Reservation{
{
Instances: []*ec2.Instance{
{
Tags: d.tags,
},
},
},
},
}
return output, nil
}