summaryrefslogtreecommitdiff
path: root/client/client_test.go
diff options
context:
space:
mode:
Diffstat (limited to 'client/client_test.go')
-rw-r--r--client/client_test.go17
1 files changed, 17 insertions, 0 deletions
diff --git a/client/client_test.go b/client/client_test.go
index b4a1b2a..9ccc149 100644
--- a/client/client_test.go
+++ b/client/client_test.go
@@ -64,3 +64,20 @@ func TestEncode(t *testing.T) {
t.Errorf("expected 'response' but got '%s'", resp)
}
}
+
+func TestUserAgent(t *testing.T) {
+ var ua string
+ var hf http.HandlerFunc = func(w http.ResponseWriter, r *http.Request) {
+ ua = r.Header.Get("user-agent")
+ }
+ server := httptest.NewServer(hf)
+ defer server.Close()
+
+ c := CaesarClient{Endpoint: server.URL}
+ c.EncodeMessage(strings.NewReader("secret message"))
+
+ expected := "caesar-client/1.0"
+ if ua != expected {
+ t.Errorf("expected user agent header %q but got %q", expected, ua)
+ }
+}