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