-
Notifications
You must be signed in to change notification settings - Fork 0
/
users_test.go
153 lines (131 loc) · 6.77 KB
/
users_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
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
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
package headscale
import (
"context"
"errors"
"net/http"
"net/url"
"testing"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/mock"
)
func TestUserResource_List(t *testing.T) {
t.Run("success", func(t *testing.T) {
client := NewMockClient()
userResource := &UserResource{Client: client}
client.(*MockClient).On("buildURL", "user").Return(&url.URL{Path: "http://example.com/api/v1/user"})
client.(*MockClient).On("buildRequest", mock.Anything, http.MethodGet, &url.URL{Path: "http://example.com/api/v1/user"}, requestOptions{}).Return(&http.Request{}, nil)
client.(*MockClient).On("do", mock.Anything, mock.Anything).Return(nil).Run(func(args mock.Arguments) {
resp := args.Get(1).(*UsersResponse)
resp.Users = []User{{ID: "1", Name: "test-user"}}
})
users, err := userResource.List(context.Background())
assert.NoError(t, err)
assert.Len(t, users.Users, 1)
assert.Equal(t, "test-user", users.Users[0].Name)
})
t.Run("error", func(t *testing.T) {
client := NewMockClient()
userResource := &UserResource{Client: client}
client.(*MockClient).On("buildURL", "user").Return(&url.URL{Path: "http://example.com/api/v1/user"})
client.(*MockClient).On("buildRequest", mock.Anything, http.MethodGet, &url.URL{Path: "http://example.com/api/v1/user"}, requestOptions{}).Return(&http.Request{}, nil)
client.(*MockClient).On("do", mock.Anything, mock.Anything).Return(errors.New("request failed"))
users, err := userResource.List(context.Background())
assert.Error(t, err)
assert.Empty(t, users.Users)
})
}
func TestUserResource_Get(t *testing.T) {
t.Run("success", func(t *testing.T) {
client := NewMockClient()
userResource := &UserResource{Client: client}
client.(*MockClient).On("buildURL", "user", "1").Return(&url.URL{Path: "http://example.com/api/v1/user/1"})
client.(*MockClient).On("buildRequest", mock.Anything, http.MethodGet, &url.URL{Path: "http://example.com/api/v1/user/1"}, requestOptions{}).Return(&http.Request{}, nil)
client.(*MockClient).On("do", mock.Anything, mock.Anything).Return(nil).Run(func(args mock.Arguments) {
resp := args.Get(1).(*UserResponse)
resp.User = User{ID: "1", Name: "test-user"}
})
user, err := userResource.Get(context.Background(), "1")
assert.NoError(t, err)
assert.Equal(t, "test-user", user.User.Name)
})
t.Run("error", func(t *testing.T) {
client := NewMockClient()
userResource := &UserResource{Client: client}
client.(*MockClient).On("buildURL", "user", "1").Return(&url.URL{Path: "http://example.com/api/v1/user/1"})
client.(*MockClient).On("buildRequest", mock.Anything, http.MethodGet, &url.URL{Path: "http://example.com/api/v1/user/1"}, requestOptions{}).Return(&http.Request{}, nil)
client.(*MockClient).On("do", mock.Anything, mock.Anything).Return(errors.New("request failed"))
user, err := userResource.Get(context.Background(), "1")
assert.Error(t, err)
assert.Empty(t, user.User)
})
}
func TestUserResource_Create(t *testing.T) {
t.Run("success", func(t *testing.T) {
client := NewMockClient()
userResource := &UserResource{Client: client}
client.(*MockClient).On("buildURL", "user").Return(&url.URL{Path: "http://example.com/api/v1/user"})
client.(*MockClient).On("buildRequest", mock.Anything, http.MethodPost, &url.URL{Path: "http://example.com/api/v1/user"}, requestOptions{
body: CreateUserRequest{Name: "new-user"},
}).Return(&http.Request{}, nil)
client.(*MockClient).On("do", mock.Anything, mock.Anything).Return(nil).Run(func(args mock.Arguments) {
resp := args.Get(1).(*User)
resp.ID = "1"
resp.Name = "new-user"
})
user, err := userResource.Create(context.Background(), "new-user")
assert.NoError(t, err)
assert.Equal(t, "new-user", user.Name)
})
t.Run("error", func(t *testing.T) {
client := NewMockClient()
userResource := &UserResource{Client: client}
client.(*MockClient).On("buildURL", "user").Return(&url.URL{Path: "http://example.com/api/v1/user"})
client.(*MockClient).On("buildRequest", mock.Anything, http.MethodPost, &url.URL{Path: "http://example.com/api/v1/user"}, requestOptions{
body: CreateUserRequest{Name: "new-user"},
}).Return(&http.Request{}, nil)
client.(*MockClient).On("do", mock.Anything, mock.Anything).Return(errors.New("request failed"))
user, err := userResource.Create(context.Background(), "new-user")
assert.Error(t, err)
assert.Empty(t, user.ID)
})
}
func TestUserResource_Delete(t *testing.T) {
t.Run("success", func(t *testing.T) {
client := NewMockClient()
userResource := &UserResource{Client: client}
client.(*MockClient).On("buildURL", "user", "1").Return(&url.URL{Path: "http://example.com/api/v1/user/1"})
client.(*MockClient).On("buildRequest", mock.Anything, http.MethodDelete, &url.URL{Path: "http://example.com/api/v1/user/1"}, requestOptions{}).Return(&http.Request{}, nil)
client.(*MockClient).On("do", mock.Anything, mock.Anything).Return(nil)
err := userResource.Delete(context.Background(), "1")
assert.NoError(t, err)
})
t.Run("error", func(t *testing.T) {
client := NewMockClient()
userResource := &UserResource{Client: client}
client.(*MockClient).On("buildURL", "user", "1").Return(&url.URL{Path: "http://example.com/api/v1/user/1"})
client.(*MockClient).On("buildRequest", mock.Anything, http.MethodDelete, &url.URL{Path: "http://example.com/api/v1/user/1"}, requestOptions{}).Return(&http.Request{}, nil)
client.(*MockClient).On("do", mock.Anything, mock.Anything).Return(errors.New("request failed"))
err := userResource.Delete(context.Background(), "1")
assert.Error(t, err)
})
}
func TestUserResource_Rename(t *testing.T) {
t.Run("success", func(t *testing.T) {
client := NewMockClient()
userResource := &UserResource{Client: client}
client.(*MockClient).On("buildURL", "user", "1", "rename", "new-name").Return(&url.URL{Path: "http://example.com/api/v1/user/1/rename/new-name"})
client.(*MockClient).On("buildRequest", mock.Anything, http.MethodPost, &url.URL{Path: "http://example.com/api/v1/user/1/rename/new-name"}, requestOptions{}).Return(&http.Request{}, nil)
client.(*MockClient).On("do", mock.Anything, mock.Anything).Return(nil)
err := userResource.Rename(context.Background(), "1", "new-name")
assert.NoError(t, err)
})
t.Run("error", func(t *testing.T) {
client := NewMockClient()
userResource := &UserResource{Client: client}
client.(*MockClient).On("buildURL", "user", "1", "rename", "new-name").Return(&url.URL{Path: "http://example.com/api/v1/user/1/rename/new-name"})
client.(*MockClient).On("buildRequest", mock.Anything, http.MethodPost, &url.URL{Path: "http://example.com/api/v1/user/1/rename/new-name"}, requestOptions{}).Return(&http.Request{}, nil)
client.(*MockClient).On("do", mock.Anything, mock.Anything).Return(errors.New("request failed"))
err := userResource.Rename(context.Background(), "1", "new-name")
assert.Error(t, err)
})
}