diff options
author | Chris <ccbrown112@gmail.com> | 2017-10-06 11:08:59 -0700 |
---|---|---|
committer | GitHub <noreply@github.com> | 2017-10-06 11:08:59 -0700 |
commit | 70e5f00241473c27a3008959ce08832c75e76ba8 (patch) | |
tree | 3afd074964c9220cc4a87210286a13dfd38c61d0 /store/sqlstore/session_store_test.go | |
parent | d1958bdc49cd10277ca2e27bb2eea499c5994954 (diff) | |
download | chat-70e5f00241473c27a3008959ce08832c75e76ba8.tar.gz chat-70e5f00241473c27a3008959ce08832c75e76ba8.tar.bz2 chat-70e5f00241473c27a3008959ce08832c75e76ba8.zip |
store/storetest package (#7588)
* prerequisites
* storetest package
Diffstat (limited to 'store/sqlstore/session_store_test.go')
-rw-r--r-- | store/sqlstore/session_store_test.go | 250 |
1 files changed, 3 insertions, 247 deletions
diff --git a/store/sqlstore/session_store_test.go b/store/sqlstore/session_store_test.go index 1eed17b51..c84d4248c 100644 --- a/store/sqlstore/session_store_test.go +++ b/store/sqlstore/session_store_test.go @@ -6,253 +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 TestSessionStoreSave(t *testing.T) { - ss := Setup() - - s1 := model.Session{} - s1.UserId = model.NewId() - - if err := (<-ss.Session().Save(&s1)).Err; err != nil { - t.Fatal(err) - } -} - -func TestSessionGet(t *testing.T) { - ss := Setup() - - s1 := model.Session{} - s1.UserId = model.NewId() - store.Must(ss.Session().Save(&s1)) - - s2 := model.Session{} - s2.UserId = s1.UserId - store.Must(ss.Session().Save(&s2)) - - s3 := model.Session{} - s3.UserId = s1.UserId - s3.ExpiresAt = 1 - store.Must(ss.Session().Save(&s3)) - - if rs1 := (<-ss.Session().Get(s1.Id)); rs1.Err != nil { - t.Fatal(rs1.Err) - } else { - if rs1.Data.(*model.Session).Id != s1.Id { - t.Fatal("should match") - } - } - - if rs2 := (<-ss.Session().GetSessions(s1.UserId)); rs2.Err != nil { - t.Fatal(rs2.Err) - } else { - if len(rs2.Data.([]*model.Session)) != 2 { - t.Fatal("should match len") - } - } -} - -func TestSessionGetWithDeviceId(t *testing.T) { - ss := Setup() - - s1 := model.Session{} - s1.UserId = model.NewId() - s1.ExpiresAt = model.GetMillis() + 10000 - store.Must(ss.Session().Save(&s1)) - - s2 := model.Session{} - s2.UserId = s1.UserId - s2.DeviceId = model.NewId() - s2.ExpiresAt = model.GetMillis() + 10000 - store.Must(ss.Session().Save(&s2)) - - s3 := model.Session{} - s3.UserId = s1.UserId - s3.ExpiresAt = 1 - s3.DeviceId = model.NewId() - store.Must(ss.Session().Save(&s3)) - - if rs1 := (<-ss.Session().GetSessionsWithActiveDeviceIds(s1.UserId)); rs1.Err != nil { - t.Fatal(rs1.Err) - } else { - if len(rs1.Data.([]*model.Session)) != 1 { - t.Fatal("should match len") - } - } -} - -func TestSessionRemove(t *testing.T) { - ss := Setup() - - s1 := model.Session{} - s1.UserId = model.NewId() - store.Must(ss.Session().Save(&s1)) - - if rs1 := (<-ss.Session().Get(s1.Id)); rs1.Err != nil { - t.Fatal(rs1.Err) - } else { - if rs1.Data.(*model.Session).Id != s1.Id { - t.Fatal("should match") - } - } - - store.Must(ss.Session().Remove(s1.Id)) - - if rs2 := (<-ss.Session().Get(s1.Id)); rs2.Err == nil { - t.Fatal("should have been removed") - } -} - -func TestSessionRemoveAll(t *testing.T) { - ss := Setup() - - s1 := model.Session{} - s1.UserId = model.NewId() - store.Must(ss.Session().Save(&s1)) - - if rs1 := (<-ss.Session().Get(s1.Id)); rs1.Err != nil { - t.Fatal(rs1.Err) - } else { - if rs1.Data.(*model.Session).Id != s1.Id { - t.Fatal("should match") - } - } - - store.Must(ss.Session().RemoveAllSessions()) - - if rs2 := (<-ss.Session().Get(s1.Id)); rs2.Err == nil { - t.Fatal("should have been removed") - } -} - -func TestSessionRemoveByUser(t *testing.T) { - ss := Setup() - - s1 := model.Session{} - s1.UserId = model.NewId() - store.Must(ss.Session().Save(&s1)) - - if rs1 := (<-ss.Session().Get(s1.Id)); rs1.Err != nil { - t.Fatal(rs1.Err) - } else { - if rs1.Data.(*model.Session).Id != s1.Id { - t.Fatal("should match") - } - } - - store.Must(ss.Session().PermanentDeleteSessionsByUser(s1.UserId)) - - if rs2 := (<-ss.Session().Get(s1.Id)); rs2.Err == nil { - t.Fatal("should have been removed") - } -} - -func TestSessionRemoveToken(t *testing.T) { - ss := Setup() - - s1 := model.Session{} - s1.UserId = model.NewId() - store.Must(ss.Session().Save(&s1)) - - if rs1 := (<-ss.Session().Get(s1.Id)); rs1.Err != nil { - t.Fatal(rs1.Err) - } else { - if rs1.Data.(*model.Session).Id != s1.Id { - t.Fatal("should match") - } - } - - store.Must(ss.Session().Remove(s1.Token)) - - if rs2 := (<-ss.Session().Get(s1.Id)); rs2.Err == nil { - t.Fatal("should have been removed") - } - - if rs3 := (<-ss.Session().GetSessions(s1.UserId)); rs3.Err != nil { - t.Fatal(rs3.Err) - } else { - if len(rs3.Data.([]*model.Session)) != 0 { - t.Fatal("should match len") - } - } -} - -func TestSessionUpdateDeviceId(t *testing.T) { - ss := Setup() - - s1 := model.Session{} - s1.UserId = model.NewId() - store.Must(ss.Session().Save(&s1)) - - if rs1 := (<-ss.Session().UpdateDeviceId(s1.Id, model.PUSH_NOTIFY_APPLE+":1234567890", s1.ExpiresAt)); rs1.Err != nil { - t.Fatal(rs1.Err) - } - - s2 := model.Session{} - s2.UserId = model.NewId() - store.Must(ss.Session().Save(&s2)) - - if rs2 := (<-ss.Session().UpdateDeviceId(s2.Id, model.PUSH_NOTIFY_APPLE+":1234567890", s1.ExpiresAt)); rs2.Err != nil { - t.Fatal(rs2.Err) - } -} - -func TestSessionUpdateDeviceId2(t *testing.T) { - ss := Setup() - - s1 := model.Session{} - s1.UserId = model.NewId() - store.Must(ss.Session().Save(&s1)) - - if rs1 := (<-ss.Session().UpdateDeviceId(s1.Id, model.PUSH_NOTIFY_APPLE_REACT_NATIVE+":1234567890", s1.ExpiresAt)); rs1.Err != nil { - t.Fatal(rs1.Err) - } - - s2 := model.Session{} - s2.UserId = model.NewId() - store.Must(ss.Session().Save(&s2)) - - if rs2 := (<-ss.Session().UpdateDeviceId(s2.Id, model.PUSH_NOTIFY_APPLE_REACT_NATIVE+":1234567890", s1.ExpiresAt)); rs2.Err != nil { - t.Fatal(rs2.Err) - } -} - -func TestSessionStoreUpdateLastActivityAt(t *testing.T) { - ss := Setup() - - s1 := model.Session{} - s1.UserId = model.NewId() - store.Must(ss.Session().Save(&s1)) - - if err := (<-ss.Session().UpdateLastActivityAt(s1.Id, 1234567890)).Err; err != nil { - t.Fatal(err) - } - - if r1 := <-ss.Session().Get(s1.Id); r1.Err != nil { - t.Fatal(r1.Err) - } else { - if r1.Data.(*model.Session).LastActivityAt != 1234567890 { - t.Fatal("LastActivityAt not updated correctly") - } - } - -} - -func TestSessionCount(t *testing.T) { - ss := Setup() - - s1 := model.Session{} - s1.UserId = model.NewId() - s1.ExpiresAt = model.GetMillis() + 100000 - store.Must(ss.Session().Save(&s1)) - - if r1 := <-ss.Session().AnalyticsSessionCount(); r1.Err != nil { - t.Fatal(r1.Err) - } else { - if r1.Data.(int64) == 0 { - t.Fatal("should have at least 1 session") - } - } +func TestSessionStore(t *testing.T) { + StoreTest(t, storetest.TestSessionStore) } |