1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
|
package ibd
import (
"context"
"testing"
"time"
"github.com/ansg191/ibd-trader-backend/internal/database"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
)
func TestClient_getCookie(t *testing.T) {
t.Parallel()
t.Run("no cookies", func(t *testing.T) {
t.Parallel()
client := NewClient(
nil,
new(emptyCookieSourceStub),
)
_, _, err := client.getCookie(context.Background(), nil)
assert.ErrorIs(t, err, ErrNoAvailableCookies)
})
t.Run("no cookies by subject", func(t *testing.T) {
t.Parallel()
client := NewClient(
nil,
new(emptyCookieSourceStub),
)
subject := "test"
_, _, err := client.getCookie(context.Background(), &subject)
assert.ErrorIs(t, err, ErrNoAvailableCookies)
})
t.Run("get any cookie", func(t *testing.T) {
t.Parallel()
client := NewClient(
nil,
new(emptyCookieSourceStub),
)
id, cookie, err := client.getCookie(context.Background(), nil)
require.NoError(t, err)
assert.Equal(t, uint(42), id)
assert.Equal(t, cookieName, cookie.Name)
assert.Equal(t, "test-token", cookie.Value)
assert.Equal(t, "/", cookie.Path)
assert.Equal(t, time.Unix(0, 0), cookie.Expires)
assert.Equal(t, "investors.com", cookie.Domain)
})
t.Run("get cookie by subject", func(t *testing.T) {
t.Parallel()
client := NewClient(
nil,
new(emptyCookieSourceStub),
)
subject := "test"
id, cookie, err := client.getCookie(context.Background(), &subject)
require.NoError(t, err)
assert.Equal(t, uint(42), id)
assert.Equal(t, cookieName, cookie.Name)
assert.Equal(t, "test-token", cookie.Value)
assert.Equal(t, "/", cookie.Path)
assert.Equal(t, time.Unix(0, 0), cookie.Expires)
assert.Equal(t, "investors.com", cookie.Domain)
})
}
type emptyCookieSourceStub struct{}
func (c *emptyCookieSourceStub) GetAnyCookie(_ context.Context) (*database.IBDCookie, error) {
return nil, nil
}
func (c *emptyCookieSourceStub) GetCookies(_ context.Context, _ string, _ bool) ([]database.IBDCookie, error) {
return nil, nil
}
func (c *emptyCookieSourceStub) ReportCookieFailure(_ context.Context, _ uint) error {
return nil
}
var testCookie = database.IBDCookie{
ID: 42,
Token: "test-token",
Expiry: time.Unix(0, 0),
}
type cookieSourceStub struct{}
func (c *cookieSourceStub) GetAnyCookie(_ context.Context) (*database.IBDCookie, error) {
return &testCookie, nil
}
func (c *cookieSourceStub) GetCookies(_ context.Context, _ string, _ bool) ([]database.IBDCookie, error) {
return []database.IBDCookie{testCookie}, nil
}
func (c *cookieSourceStub) ReportCookieFailure(_ context.Context, _ uint) error {
return nil
}
|
