From 6afa4fb08383d0968887465542b352667228fa30 Mon Sep 17 00:00:00 2001 From: Cory LaNou Date: Sat, 7 Mar 2015 08:03:43 -0700 Subject: [PATCH] update test to expect a default user agent --- client/influxdb_test.go | 11 +++++------ 1 file changed, 5 insertions(+), 6 deletions(-) diff --git a/client/influxdb_test.go b/client/influxdb_test.go index 9d870c382a..97daea6d91 100644 --- a/client/influxdb_test.go +++ b/client/influxdb_test.go @@ -6,6 +6,7 @@ import ( "net/http" "net/http/httptest" "net/url" + "strings" "testing" "time" @@ -134,8 +135,6 @@ func TestClient_UserAgent(t *testing.T) { t.Fatalf("unexpected error. expected %v, actual %v", nil, err) } - defaultUserAgent := receivedUserAgent - tests := []struct { name string userAgent string @@ -144,7 +143,7 @@ func TestClient_UserAgent(t *testing.T) { { name: "Empty user agent", userAgent: "", - expected: defaultUserAgent, + expected: "InfluxDBClient/", }, { name: "Custom user agent", @@ -167,7 +166,7 @@ func TestClient_UserAgent(t *testing.T) { if err != nil { t.Fatalf("unexpected error. expected %v, actual %v", nil, err) } - if receivedUserAgent != test.expected { + if !strings.HasPrefix(receivedUserAgent, test.expected) { t.Fatalf("Unexpected user agent. expected %v, actual %v", test.expected, receivedUserAgent) } @@ -177,7 +176,7 @@ func TestClient_UserAgent(t *testing.T) { if err != nil { t.Fatalf("unexpected error. expected %v, actual %v", nil, err) } - if receivedUserAgent != test.expected { + if !strings.HasPrefix(receivedUserAgent, test.expected) { t.Fatalf("Unexpected user agent. expected %v, actual %v", test.expected, receivedUserAgent) } @@ -186,7 +185,7 @@ func TestClient_UserAgent(t *testing.T) { if err != nil { t.Fatalf("unexpected error. expected %v, actual %v", nil, err) } - if receivedUserAgent != test.expected { + if !strings.HasPrefix(receivedUserAgent, test.expected) { t.Fatalf("Unexpected user agent. expected %v, actual %v", test.expected, receivedUserAgent) } }