Adding ec2meta function

Signed-off-by: Dave Henderson <dhenderson@gmail.com>

1 files changed, 26 insertions, 0 deletions

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
+}