diff options
Diffstat (limited to 'store/sqlstore/user_access_token_store_test.go')
-rw-r--r-- | store/sqlstore/user_access_token_store_test.go | 81 |
1 files changed, 4 insertions, 77 deletions
diff --git a/store/sqlstore/user_access_token_store_test.go b/store/sqlstore/user_access_token_store_test.go index 74ce53b64..4bca12ccc 100644 --- a/store/sqlstore/user_access_token_store_test.go +++ b/store/sqlstore/user_access_token_store_test.go @@ -1,4 +1,4 @@ -// Copyright (c) 2017-present Mattermost, Inc. All Rights Reserved. +// Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved. // See License.txt for license information. package sqlstore @@ -6,82 +6,9 @@ package sqlstore import ( "testing" - "github.com/mattermost/mattermost-server/model" - "github.com/mattermost/mattermost-server/store" + "github.com/mattermost/mattermost-server/store/storetest" ) -func TestUserAccessTokenSaveGetDelete(t *testing.T) { - ss := Setup() - - uat := &model.UserAccessToken{ - Token: model.NewId(), - UserId: model.NewId(), - 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().Get(uat.Id); result.Err != nil { - t.Fatal(result.Err) - } else if received := result.Data.(*model.UserAccessToken); received.Token != uat.Token { - t.Fatal("received incorrect token after save") - } - - if result := <-ss.UserAccessToken().GetByToken(uat.Token); result.Err != nil { - t.Fatal(result.Err) - } else if received := result.Data.(*model.UserAccessToken); received.Token != uat.Token { - t.Fatal("received incorrect token after save") - } - - if result := <-ss.UserAccessToken().GetByToken("notarealtoken"); result.Err == nil { - t.Fatal("should have failed on bad token") - } - - if result := <-ss.UserAccessToken().GetByUser(uat.UserId, 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) - } - - if err := (<-ss.Session().Get(s1.Token)).Err; err == nil { - t.Fatal("should error - session should be deleted") - } - - if err := (<-ss.UserAccessToken().GetByToken(s1.Token)).Err; err == nil { - t.Fatal("should error - access token should be deleted") - } - - s2 := model.Session{} - s2.UserId = uat.UserId - s2.Token = uat.Token - - store.Must(ss.Session().Save(&s2)) - - if result := <-ss.UserAccessToken().Save(uat); result.Err != nil { - t.Fatal(result.Err) - } - - if result := <-ss.UserAccessToken().DeleteAllForUser(uat.UserId); result.Err != nil { - t.Fatal(result.Err) - } - - if err := (<-ss.Session().Get(s2.Token)).Err; err == nil { - t.Fatal("should error - session should be deleted") - } - - if err := (<-ss.UserAccessToken().GetByToken(s2.Token)).Err; err == nil { - t.Fatal("should error - access token should be deleted") - } +func TestUserAccessTokenStore(t *testing.T) { + StoreTest(t, storetest.TestUserAccessTokenStore) } |