-
-
Notifications
You must be signed in to change notification settings - Fork 0
/
websocket_auth_test.go
61 lines (56 loc) · 1.33 KB
/
websocket_auth_test.go
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
package kraken
import (
"context"
"net/http"
"net/http/httptest"
"net/url"
"reflect"
"testing"
)
func TestWebsocketsAuth_WebsocketsToken(t *testing.T) {
ctx := context.Background()
type fields struct {
apiMock *httptest.Server
}
type args struct {
ctx context.Context
}
tests := []struct {
name string
fields fields
args args
want *WebsocketsToken
wantErr bool
}{
{
name: "error creating request",
fields: fields{apiMock: createFakeServer(http.StatusOK, "")},
args: args{},
wantErr: true,
},
{
name: "get websockets token",
fields: fields{apiMock: createFakeServer(http.StatusOK, "ws_auth.json")},
args: args{ctx: ctx},
want: &WebsocketsToken{
Token: "1Dwc4lzSwNWOAwkMdqhssNNFhs1ed606d1WcF3XfEMw",
Expires: 900,
},
},
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
baseURL, _ := url.Parse(tt.fields.apiMock.URL + "/")
c := New(tt.fields.apiMock.Client()).WithAuth(Secrets{})
c.baseURL = baseURL
got, err := c.WebsocketsAuth.WebsocketsToken(tt.args.ctx)
if (err != nil) != tt.wantErr {
t.Errorf("WebsocketsAuth.WebsocketsToken() error = %v, wantErr %v", err, tt.wantErr)
return
}
if !reflect.DeepEqual(got, tt.want) {
t.Errorf("WebsocketsAuth.WebsocketsToken() = %v, want %v", got, tt.want)
}
})
}
}