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 | 43 |
2 files changed, 59 insertions, 0 deletions
diff --git a/store/storetest/mocks/UserAccessTokenStore.go b/store/storetest/mocks/UserAccessTokenStore.go index 60e08076c..b989fa1cc 100644 --- a/store/storetest/mocks/UserAccessTokenStore.go +++ b/store/storetest/mocks/UserAccessTokenStore.go @@ -109,6 +109,22 @@ func (_m *UserAccessTokenStore) GetByUser(userId string, page int, perPage int) return r0 } +// Search provides a mock function with given fields: +func (_m *UserAccessTokenStore) Search(term string) store.StoreChannel { + ret := _m.Called(term) + + var r0 store.StoreChannel + if rf, ok := ret.Get(0).(func(string) store.StoreChannel); ok { + r0 = rf(term) + } else { + if ret.Get(0) != nil { + r0 = ret.Get(0).(store.StoreChannel) + } + } + + return r0 +} + // Save provides a mock function with given fields: token func (_m *UserAccessTokenStore) Save(token *model.UserAccessToken) store.StoreChannel { ret := _m.Called(token) diff --git a/store/storetest/user_access_token_store.go b/store/storetest/user_access_token_store.go index c32023d30..e8eb8ca60 100644 --- a/store/storetest/user_access_token_store.go +++ b/store/storetest/user_access_token_store.go @@ -13,6 +13,7 @@ import ( func TestUserAccessTokenStore(t *testing.T, ss store.Store) { t.Run("UserAccessTokenSaveGetDelete", func(t *testing.T) { testUserAccessTokenSaveGetDelete(t, ss) }) t.Run("UserAccessTokenDisableEnable", func(t *testing.T) { testUserAccessTokenDisableEnable(t, ss) }) + t.Run("UserAccessTokenSearch", func(t *testing.T) { testUserAccessTokenSearch(t, ss) }) } func testUserAccessTokenSaveGetDelete(t *testing.T, ss store.Store) { @@ -130,3 +131,45 @@ func testUserAccessTokenDisableEnable(t *testing.T, ss store.Store) { t.Fatal(err) } } + +func testUserAccessTokenSearch(t *testing.T, ss store.Store) { + u1 := model.User{} + u1.Email = model.NewId() + u1.Username = model.NewId() + + store.Must(ss.User().Save(&u1)) + + uat := &model.UserAccessToken{ + Token: model.NewId(), + UserId: u1.Id, + Description: "testtoken", + } + + s1 := model.Session{} + s1.UserId = uat.UserId + s1.Token = uat.Token + + store.Must(ss.Session().Save(&s1)) + + if result := <-ss.UserAccessToken().Save(uat); result.Err != nil { + t.Fatal(result.Err) + } + + if result := <-ss.UserAccessToken().Search(uat.Id); 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 search") + } + + if result := <-ss.UserAccessToken().Search(uat.UserId); 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 search") + } + + if result := <-ss.UserAccessToken().Search(u1.Username); 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 search") + } +} |