aboutsummaryrefslogtreecommitdiffstats
path: root/client/go/util/http_test.go
diff options
context:
space:
mode:
Diffstat (limited to 'client/go/util/http_test.go')
-rw-r--r--client/go/util/http_test.go46
1 files changed, 46 insertions, 0 deletions
diff --git a/client/go/util/http_test.go b/client/go/util/http_test.go
new file mode 100644
index 00000000000..03b155488d9
--- /dev/null
+++ b/client/go/util/http_test.go
@@ -0,0 +1,46 @@
+// Copyright Verizon Media. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
+// Basic testing of our HTTP client wrapper
+// Author: bratseth
+
+package util
+
+import (
+ "bytes"
+ "github.com/stretchr/testify/assert"
+ "io/ioutil"
+ "net/http"
+ "testing"
+ "time"
+)
+
+type mockHttpClient struct {}
+
+func (c mockHttpClient) Do(request *http.Request, timeout time.Duration) (response *http.Response, error error) {
+ var status int
+ var body string
+ if request.URL.String() == "http://host/okpath" {
+ status = 200
+ body = "OK body"
+ } else {
+ status = 500
+ body = "Unexpected url body"
+ }
+
+ return &http.Response{
+ StatusCode: status,
+ Header: make(http.Header),
+ Body: ioutil.NopCloser(bytes.NewBufferString(body)),
+ },
+ nil
+}
+
+func TestHttpRequest(t *testing.T) {
+ ActiveHttpClient = mockHttpClient{}
+
+ response := HttpGet("http://host", "/okpath", "description")
+ assert.Equal(t, 200, response.StatusCode)
+
+ response = HttpGet("http://host", "/otherpath", "description")
+ assert.Equal(t, 500, response.StatusCode)
+}
+