diff options
Diffstat (limited to 'client/client_test.go')
-rw-r--r-- | client/client_test.go | 17 |
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) + } +} |