1package session 2 3import ( 4 "strings" 5 "testing" 6) 7 8func TestSession_WithDefaultUserAgent(t *testing.T) { 9 s := New() 10 expected := getDefaultUserAgent() 11 actual := s.userAgent 12 if actual != expected { 13 t.Errorf("UserAgent expected %s, actual %s", expected, actual) 14 } 15} 16 17func TestAddToUserAgent(t *testing.T) { 18 s := New() 19 ua := "product/v1 ( myproduct )" 20 s.AppendUserAgent(ua) 21 if !strings.HasSuffix(s.userAgent, ua) { 22 t.Errorf("UserAgent expected to end with %s, but ends with %s", ua, s.userAgent) 23 } 24 if !strings.HasPrefix(s.userAgent, getDefaultUserAgent()) { 25 t.Errorf("UserAgent expected to start with %s, but starts with %s", getDefaultUserAgent(), s.userAgent) 26 } 27} 28 29func TestResetUserAgent(t *testing.T) { 30 s := New() 31 ua := "product/v1 ( myproduct )" 32 s.AppendUserAgent(ua) 33 s.ResetUserAgent() 34 if s.userAgent != getDefaultUserAgent() { 35 t.Errorf("UserAgent expected to reset to %s, but found to be %s", getDefaultUserAgent(), s.userAgent) 36 } 37} 38