From 188b08d217f61e86205ff321dc713de8a6600934 Mon Sep 17 00:00:00 2001 From: Dave Henderson Date: Fri, 25 Mar 2016 22:02:13 -0400 Subject: Adding ec2meta function Signed-off-by: Dave Henderson --- aws/testutils.go | 26 ++++++++++++++++++++++++++ 1 file changed, 26 insertions(+) create mode 100644 aws/testutils.go (limited to 'aws/testutils.go') diff --git a/aws/testutils.go b/aws/testutils.go new file mode 100644 index 00000000..a45acdc6 --- /dev/null +++ b/aws/testutils.go @@ -0,0 +1,26 @@ +package aws + +import ( + "fmt" + "net/http" + "net/http/httptest" + "net/url" +) + +// MockServer - +func MockServer(code int, body string) (*httptest.Server, *Ec2Meta) { + server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { + w.WriteHeader(code) + 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} + + client := &Ec2Meta{server.URL + "/", httpClient} + return server, client +} -- cgit v1.2.3