diff options
Diffstat (limited to 'store/storetest')
-rw-r--r-- | store/storetest/mocks/UserAccessTokenStore.go | 16 | ||||
-rw-r--r-- | store/storetest/user_access_token_store.go | 6 |
2 files changed, 22 insertions, 0 deletions
diff --git a/store/storetest/mocks/UserAccessTokenStore.go b/store/storetest/mocks/UserAccessTokenStore.go index 87541fd76..60e08076c 100644 --- a/store/storetest/mocks/UserAccessTokenStore.go +++ b/store/storetest/mocks/UserAccessTokenStore.go @@ -61,6 +61,22 @@ func (_m *UserAccessTokenStore) Get(tokenId string) store.StoreChannel { return r0 } +// GetAll provides a mock function with given fields: +func (_m *UserAccessTokenStore) GetAll(page int, perPage int) store.StoreChannel { + ret := _m.Called(page, perPage) + + var r0 store.StoreChannel + if rf, ok := ret.Get(0).(func(int, int) store.StoreChannel); ok { + r0 = rf(page, perPage) + } else { + if ret.Get(0) != nil { + r0 = ret.Get(0).(store.StoreChannel) + } + } + + return r0 +} + // GetByToken provides a mock function with given fields: tokenString func (_m *UserAccessTokenStore) GetByToken(tokenString string) store.StoreChannel { ret := _m.Called(tokenString) diff --git a/store/storetest/user_access_token_store.go b/store/storetest/user_access_token_store.go index 661c969da..c32023d30 100644 --- a/store/storetest/user_access_token_store.go +++ b/store/storetest/user_access_token_store.go @@ -54,6 +54,12 @@ func testUserAccessTokenSaveGetDelete(t *testing.T, ss store.Store) { t.Fatal("received incorrect number of tokens after save") } + if result := <-ss.UserAccessToken().GetAll(0, 100); result.Err != nil { + t.Fatal(result.Err) + } else if received := result.Data.([]*model.UserAccessToken); len(received) != 1 { + t.Fatal("received incorrect number of tokens after save") + } + if result := <-ss.UserAccessToken().Delete(uat.Id); result.Err != nil { t.Fatal(result.Err) } |