|
1 | 1 | package model |
2 | 2 |
|
3 | 3 | import ( |
4 | | - "context" |
5 | 4 | "encoding/json" |
6 | 5 | "errors" |
7 | 6 | "testing" |
@@ -39,9 +38,9 @@ func TestListModelsEmpty(t *testing.T) { |
39 | 38 | ContinuationToken: openfga.PtrString(""), |
40 | 39 | } |
41 | 40 | mockRequest.EXPECT().Options(options).Return(mockExecute) |
42 | | - mockFgaClient.EXPECT().ReadAuthorizationModels(context.Background()).Return(mockRequest) |
| 41 | + mockFgaClient.EXPECT().ReadAuthorizationModels(t.Context()).Return(mockRequest) |
43 | 42 |
|
44 | | - output, err := listModels(mockFgaClient, 5) |
| 43 | + output, err := listModels(t.Context(), mockFgaClient, 5) |
45 | 44 | if err != nil { |
46 | 45 | t.Error(err) |
47 | 46 | } |
@@ -74,9 +73,9 @@ func TestListModelsFail(t *testing.T) { |
74 | 73 | ContinuationToken: openfga.PtrString(""), |
75 | 74 | } |
76 | 75 | mockRequest.EXPECT().Options(options).Return(mockExecute) |
77 | | - mockFgaClient.EXPECT().ReadAuthorizationModels(context.Background()).Return(mockRequest) |
| 76 | + mockFgaClient.EXPECT().ReadAuthorizationModels(t.Context()).Return(mockRequest) |
78 | 77 |
|
79 | | - _, err := listModels(mockFgaClient, 5) |
| 78 | + _, err := listModels(t.Context(), mockFgaClient, 5) |
80 | 79 | if err == nil { |
81 | 80 | t.Error("Expect error but there is none") |
82 | 81 | } |
@@ -112,9 +111,9 @@ func TestListModelsSinglePage(t *testing.T) { |
112 | 111 | ContinuationToken: openfga.PtrString(""), |
113 | 112 | } |
114 | 113 | mockRequest.EXPECT().Options(options).Return(mockExecute) |
115 | | - mockFgaClient.EXPECT().ReadAuthorizationModels(context.Background()).Return(mockRequest) |
| 114 | + mockFgaClient.EXPECT().ReadAuthorizationModels(t.Context()).Return(mockRequest) |
116 | 115 |
|
117 | | - output, err := listModels(mockFgaClient, 5) |
| 116 | + output, err := listModels(t.Context(), mockFgaClient, 5) |
118 | 117 | if err != nil { |
119 | 118 | t.Error(err) |
120 | 119 | } |
@@ -191,11 +190,11 @@ func TestListModelsMultiPage(t *testing.T) { |
191 | 190 | mockRequest2.EXPECT().Options(options2).Return(mockExecute2), |
192 | 191 | ) |
193 | 192 | gomock.InOrder( |
194 | | - mockFgaClient.EXPECT().ReadAuthorizationModels(context.Background()).Return(mockRequest1), |
195 | | - mockFgaClient.EXPECT().ReadAuthorizationModels(context.Background()).Return(mockRequest2), |
| 193 | + mockFgaClient.EXPECT().ReadAuthorizationModels(t.Context()).Return(mockRequest1), |
| 194 | + mockFgaClient.EXPECT().ReadAuthorizationModels(t.Context()).Return(mockRequest2), |
196 | 195 | ) |
197 | 196 |
|
198 | | - output, err := listModels(mockFgaClient, 2) |
| 197 | + output, err := listModels(t.Context(), mockFgaClient, 2) |
199 | 198 | if err != nil { |
200 | 199 | t.Error(err) |
201 | 200 | } |
@@ -244,9 +243,9 @@ func TestListModelsMultiPageMaxPage(t *testing.T) { |
244 | 243 | mockExecute1.EXPECT().Execute().Return(&response1, nil) |
245 | 244 |
|
246 | 245 | mockRequest1.EXPECT().Options(options1).Return(mockExecute1) |
247 | | - mockFgaClient.EXPECT().ReadAuthorizationModels(context.Background()).Return(mockRequest1) |
| 246 | + mockFgaClient.EXPECT().ReadAuthorizationModels(t.Context()).Return(mockRequest1) |
248 | 247 |
|
249 | | - output, err := listModels(mockFgaClient, 0) |
| 248 | + output, err := listModels(t.Context(), mockFgaClient, 0) |
250 | 249 | if err != nil { |
251 | 250 | t.Error(err) |
252 | 251 | } |
|
0 commit comments