mirrors/mostlymatter
1
0
Форкнуть 0
Код Задачи Запросы на слияние Действия Пакеты Проекты Релизы Вики Активность

Migrate tests from "store/storetest/session_store.go" to use t… (#12805)

Просмотр исходного кода
Этот коммит содержится в:
pqzx
2019-10-17 23:36:17 +11:00
коммит произвёл Ben Schumacher
родитель c7e2689926
Коммит 1ad0cb646f
1 изменённых файлов: 49 добавлений и 98 удалений
Скачать Patch файл Скачать Diff файл Показать все файлы Свернуть все файлы

147
store/storetest/session_store.go
Просмотреть файл

@@ -34,9 +34,8 @@ func testSessionStoreSave(t *testing.T, ss store.Store) {
s1 := &model.Session{} s1 := &model.Session{}
s1.UserId = model.NewId() s1.UserId = model.NewId()
if _, err := ss.Session().Save(s1); err != nil { _, err := ss.Session().Save(s1)
t.Fatal(err) require.Nil(t, err)
}
} }
func testSessionGet(t *testing.T, ss store.Store) { func testSessionGet(t *testing.T, ss store.Store) {
@@ -59,21 +58,13 @@ func testSessionGet(t *testing.T, ss store.Store) {
s3, err = ss.Session().Save(s3) s3, err = ss.Session().Save(s3)
require.Nil(t, err) require.Nil(t, err)
if session, err := ss.Session().Get(s1.Id); err != nil { session, err := ss.Session().Get(s1.Id)
t.Fatal(err) require.Nil(t, err)
} else { require.Equal(t, session.Id, s1.Id, "should match")
if session.Id != s1.Id {
t.Fatal("should match")
}
}
if session, err := ss.Session().GetSessions(s1.UserId); err != nil { data, err := ss.Session().GetSessions(s1.UserId)
t.Fatal(err) require.Nil(t, err)
} else { require.Len(t, data, 3, "should match len")
if len(session) != 3 {
t.Fatal("should match len")
}
}
} }
func testSessionGetWithDeviceId(t *testing.T, ss store.Store) { func testSessionGetWithDeviceId(t *testing.T, ss store.Store) {
@@ -100,13 +91,9 @@ func testSessionGetWithDeviceId(t *testing.T, ss store.Store) {
s3, err = ss.Session().Save(s3) s3, err = ss.Session().Save(s3)
require.Nil(t, err) require.Nil(t, err)
if data, err := ss.Session().GetSessionsWithActiveDeviceIds(s1.UserId); err != nil { data, err := ss.Session().GetSessionsWithActiveDeviceIds(s1.UserId)
t.Fatal(err) require.Nil(t, err)
} else { require.Len(t, data, 1, "should match len")
if len(data) != 1 {
t.Fatal("should match len")
}
}
} }
func testSessionRemove(t *testing.T, ss store.Store) { func testSessionRemove(t *testing.T, ss store.Store) {
@@ -116,19 +103,15 @@ func testSessionRemove(t *testing.T, ss store.Store) {
s1, err := ss.Session().Save(s1) s1, err := ss.Session().Save(s1)
require.Nil(t, err) require.Nil(t, err)
if session, err := ss.Session().Get(s1.Id); err != nil { session, err := ss.Session().Get(s1.Id)
t.Fatal(err) require.Nil(t, err)
} else { require.Equal(t, session.Id, s1.Id, "should match")
if session.Id != s1.Id {
t.Fatal("should match")
}
}
removeErr := ss.Session().Remove(s1.Id) removeErr := ss.Session().Remove(s1.Id)
require.Nil(t, removeErr) require.Nil(t, removeErr)
if _, err := ss.Session().Get(s1.Id); err == nil {
t.Fatal("should have been removed") _, err = ss.Session().Get(s1.Id)
} require.NotNil(t, err, "should have been removed")
} }
func testSessionRemoveAll(t *testing.T, ss store.Store) { func testSessionRemoveAll(t *testing.T, ss store.Store) {
@@ -138,20 +121,15 @@ func testSessionRemoveAll(t *testing.T, ss store.Store) {
s1, err := ss.Session().Save(s1) s1, err := ss.Session().Save(s1)
require.Nil(t, err) require.Nil(t, err)
if session, err := ss.Session().Get(s1.Id); err != nil { session, err := ss.Session().Get(s1.Id)
t.Fatal(err) require.Nil(t, err)
} else { require.Equal(t, session.Id, s1.Id, "should match")
if session.Id != s1.Id {
t.Fatal("should match")
}
}
removeErr := ss.Session().RemoveAllSessions() removeErr := ss.Session().RemoveAllSessions()
require.Nil(t, removeErr) require.Nil(t, removeErr)
if _, err := ss.Session().Get(s1.Id); err == nil { _, err = ss.Session().Get(s1.Id)
t.Fatal("should have been removed") require.NotNil(t, err, "should have been removed")
}
} }
func testSessionRemoveByUser(t *testing.T, ss store.Store) { func testSessionRemoveByUser(t *testing.T, ss store.Store) {
@@ -161,20 +139,15 @@ func testSessionRemoveByUser(t *testing.T, ss store.Store) {
s1, err := ss.Session().Save(s1) s1, err := ss.Session().Save(s1)
require.Nil(t, err) require.Nil(t, err)
if session, err := ss.Session().Get(s1.Id); err != nil { session, err := ss.Session().Get(s1.Id)
t.Fatal(err) require.Nil(t, err)
} else { require.Equal(t, session.Id, s1.Id, "should match")
if session.Id != s1.Id {
t.Fatal("should match")
}
}
deleteErr := ss.Session().PermanentDeleteSessionsByUser(s1.UserId) deleteErr := ss.Session().PermanentDeleteSessionsByUser(s1.UserId)
require.Nil(t, deleteErr) require.Nil(t, deleteErr)
if _, err := ss.Session().Get(s1.Id); err == nil { _, err = ss.Session().Get(s1.Id)
t.Fatal("should have been removed") require.NotNil(t, err, "should have been removed")
}
} }
func testSessionRemoveToken(t *testing.T, ss store.Store) { func testSessionRemoveToken(t *testing.T, ss store.Store) {
@@ -184,28 +157,19 @@ func testSessionRemoveToken(t *testing.T, ss store.Store) {
s1, err := ss.Session().Save(s1) s1, err := ss.Session().Save(s1)
require.Nil(t, err) require.Nil(t, err)
if session, err := ss.Session().Get(s1.Id); err != nil { session, err := ss.Session().Get(s1.Id)
t.Fatal(err) require.Nil(t, err)
} else { require.Equal(t, session.Id, s1.Id, "should match")
if session.Id != s1.Id {
t.Fatal("should match")
}
}
removeErr := ss.Session().Remove(s1.Token) removeErr := ss.Session().Remove(s1.Token)
require.Nil(t, removeErr) require.Nil(t, removeErr)
if _, err := ss.Session().Get(s1.Id); err == nil { _, err = ss.Session().Get(s1.Id)
t.Fatal("should have been removed") require.NotNil(t, err, "should have been removed")
}
if session, err := ss.Session().GetSessions(s1.UserId); err != nil { data, err := ss.Session().GetSessions(s1.UserId)
t.Fatal(err) require.Nil(t, err)
} else { require.Len(t, data, 0, "should match len")
if len(session) != 0 {
t.Fatal("should match len")
}
}
} }
func testSessionUpdateDeviceId(t *testing.T, ss store.Store) { func testSessionUpdateDeviceId(t *testing.T, ss store.Store) {
@@ -215,9 +179,8 @@ func testSessionUpdateDeviceId(t *testing.T, ss store.Store) {
s1, err := ss.Session().Save(s1) s1, err := ss.Session().Save(s1)
require.Nil(t, err) require.Nil(t, err)
if _, err = ss.Session().UpdateDeviceId(s1.Id, model.PUSH_NOTIFY_APPLE+":1234567890", s1.ExpiresAt); err != nil { _, err = ss.Session().UpdateDeviceId(s1.Id, model.PUSH_NOTIFY_APPLE+":1234567890", s1.ExpiresAt)
t.Fatal(err) require.Nil(t, err)
}
s2 := &model.Session{} s2 := &model.Session{}
s2.UserId = model.NewId() s2.UserId = model.NewId()
@@ -225,9 +188,8 @@ func testSessionUpdateDeviceId(t *testing.T, ss store.Store) {
s2, err = ss.Session().Save(s2) s2, err = ss.Session().Save(s2)
require.Nil(t, err) require.Nil(t, err)
if _, err := ss.Session().UpdateDeviceId(s2.Id, model.PUSH_NOTIFY_APPLE+":1234567890", s1.ExpiresAt); err != nil { _, err = ss.Session().UpdateDeviceId(s2.Id, model.PUSH_NOTIFY_APPLE+":1234567890", s1.ExpiresAt)
t.Fatal(err) require.Nil(t, err)
}
} }
func testSessionUpdateDeviceId2(t *testing.T, ss store.Store) { func testSessionUpdateDeviceId2(t *testing.T, ss store.Store) {
@@ -237,9 +199,8 @@ func testSessionUpdateDeviceId2(t *testing.T, ss store.Store) {
s1, err := ss.Session().Save(s1) s1, err := ss.Session().Save(s1)
require.Nil(t, err) require.Nil(t, err)
if _, err = ss.Session().UpdateDeviceId(s1.Id, model.PUSH_NOTIFY_APPLE_REACT_NATIVE+":1234567890", s1.ExpiresAt); err != nil { _, err = ss.Session().UpdateDeviceId(s1.Id, model.PUSH_NOTIFY_APPLE_REACT_NATIVE+":1234567890", s1.ExpiresAt)
t.Fatal(err) require.Nil(t, err)
}
s2 := &model.Session{} s2 := &model.Session{}
s2.UserId = model.NewId() s2.UserId = model.NewId()
@@ -247,9 +208,8 @@ func testSessionUpdateDeviceId2(t *testing.T, ss store.Store) {
s2, err = ss.Session().Save(s2) s2, err = ss.Session().Save(s2)
require.Nil(t, err) require.Nil(t, err)
if _, err := ss.Session().UpdateDeviceId(s2.Id, model.PUSH_NOTIFY_APPLE_REACT_NATIVE+":1234567890", s1.ExpiresAt); err != nil { _, err = ss.Session().UpdateDeviceId(s2.Id, model.PUSH_NOTIFY_APPLE_REACT_NATIVE+":1234567890", s1.ExpiresAt)
t.Fatal(err) require.Nil(t, err)
}
} }
func testSessionStoreUpdateLastActivityAt(t *testing.T, ss store.Store) { func testSessionStoreUpdateLastActivityAt(t *testing.T, ss store.Store) {
@@ -262,14 +222,9 @@ func testSessionStoreUpdateLastActivityAt(t *testing.T, ss store.Store) {
err = ss.Session().UpdateLastActivityAt(s1.Id, 1234567890) err = ss.Session().UpdateLastActivityAt(s1.Id, 1234567890)
require.Nil(t, err) require.Nil(t, err)
if session, err := ss.Session().Get(s1.Id); err != nil { session, err := ss.Session().Get(s1.Id)
t.Fatal(err) require.Nil(t, err)
} else { require.EqualValues(t, session.LastActivityAt, 1234567890, "LastActivityAt not updated correctly")
if session.LastActivityAt != 1234567890 {
t.Fatal("LastActivityAt not updated correctly")
}
}
} }
func testSessionCount(t *testing.T, ss store.Store) { func testSessionCount(t *testing.T, ss store.Store) {
@@ -280,13 +235,9 @@ func testSessionCount(t *testing.T, ss store.Store) {
s1, err := ss.Session().Save(s1) s1, err := ss.Session().Save(s1)
require.Nil(t, err) require.Nil(t, err)
if count, err := ss.Session().AnalyticsSessionCount(); err != nil { count, err := ss.Session().AnalyticsSessionCount()
t.Fatal(err) require.Nil(t, err)
} else { require.NotZero(t, count, "should have at least 1 session")
if count == 0 {
t.Fatal("should have at least 1 session")
}
}
} }
func testSessionCleanup(t *testing.T, ss store.Store) { func testSessionCleanup(t *testing.T, ss store.Store) {