mirror of
https://github.com/mattermost/mattermost.git
synced 2025-02-25 18:55:24 -06:00
7fa4913f90
* Add column to UserAccessTokens table * PLT-7794 Add user access token enable/disable endpoints * replaced eliminated global variable * updates to user_access_token_store and upgrade.go * style fix and cleanup
127 lines
3.5 KiB
Go
127 lines
3.5 KiB
Go
// Copyright (c) 2017-present Mattermost, Inc. All Rights Reserved.
|
|
// See License.txt for license information.
|
|
|
|
package storetest
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/mattermost/mattermost-server/model"
|
|
"github.com/mattermost/mattermost-server/store"
|
|
)
|
|
|
|
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) })
|
|
}
|
|
|
|
func testUserAccessTokenSaveGetDelete(t *testing.T, ss store.Store) {
|
|
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 testUserAccessTokenDisableEnable(t *testing.T, ss store.Store) {
|
|
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 err := (<-ss.UserAccessToken().UpdateTokenDisable(uat.Id)).Err; err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if err := (<-ss.Session().Get(s1.Token)).Err; err == nil {
|
|
t.Fatal("should error - session should be deleted")
|
|
}
|
|
|
|
s2 := model.Session{}
|
|
s2.UserId = uat.UserId
|
|
s2.Token = uat.Token
|
|
|
|
store.Must(ss.Session().Save(&s2))
|
|
|
|
if err := (<-ss.UserAccessToken().UpdateTokenEnable(uat.Id)).Err; err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|