Migrate tests from store/storetest/status_store.go to use test… (#12873)

Этот коммит содержится в:
Nikhil Ranjan
2019-11-01 12:52:18 +01:00
коммит произвёл Ben Schumacher
родитель 0b24c336b3
Коммит 3687a0b6c1

Просмотреть файл

@@ -23,9 +23,8 @@ func testStatusStore(t *testing.T, ss store.Store) {
status.LastActivityAt = 10
if _, err := ss.Status().Get(status.UserId); err != nil {
t.Fatal(err)
}
_, err := ss.Status().Get(status.UserId)
require.Nil(t, err)
status2 := &model.Status{UserId: model.NewId(), Status: model.STATUS_AWAY, Manual: false, LastActivityAt: 0, ActiveChannel: ""}
require.Nil(t, ss.Status().SaveOrUpdate(status2))
@@ -33,40 +32,28 @@ func testStatusStore(t *testing.T, ss store.Store) {
status3 := &model.Status{UserId: model.NewId(), Status: model.STATUS_OFFLINE, Manual: false, LastActivityAt: 0, ActiveChannel: ""}
require.Nil(t, ss.Status().SaveOrUpdate(status3))
if statuses, err := ss.Status().GetByIds([]string{status.UserId, "junk"}); err != nil {
t.Fatal(err)
} else {
if len(statuses) != 1 {
t.Fatal("should only have 1 status")
}
}
statuses, err := ss.Status().GetByIds([]string{status.UserId, "junk"})
require.Nil(t, err)
require.Len(t, statuses, 1, "should only have 1 status")
if err := ss.Status().ResetAll(); err != nil {
t.Fatal(err)
}
err = ss.Status().ResetAll()
require.Nil(t, err)
if statusParameter, err := ss.Status().Get(status.UserId); err != nil {
t.Fatal(err)
} else {
if statusParameter.Status != model.STATUS_OFFLINE {
t.Fatal("should be offline")
}
}
statusParameter, err := ss.Status().Get(status.UserId)
require.Nil(t, err)
require.Equal(t, statusParameter.Status, model.STATUS_OFFLINE, "should be offline")
if err := ss.Status().UpdateLastActivityAt(status.UserId, 10); err != nil {
t.Fatal(err)
}
err = ss.Status().UpdateLastActivityAt(status.UserId, 10)
require.Nil(t, err)
}
func testActiveUserCount(t *testing.T, ss store.Store) {
status := &model.Status{UserId: model.NewId(), Status: model.STATUS_ONLINE, Manual: false, LastActivityAt: model.GetMillis(), ActiveChannel: ""}
require.Nil(t, ss.Status().SaveOrUpdate(status))
if count, err := ss.Status().GetTotalActiveUsersCount(); err != nil {
t.Fatal(err)
} else {
require.True(t, count > 0, "expected count > 0, got %d", count)
}
count, err := ss.Status().GetTotalActiveUsersCount()
require.Nil(t, err)
require.True(t, count > 0, "expected count > 0, got %d", count)
}
type ByUserId []*model.Status