thepaperpilot/forgejo
1
0
Fork 0
Code Issues Pull requests Releases Activity Actions

[GITEA] Fix session generation for database (squash) timeutil.Mock

Browse source 
because of e743570f65 * Refactor timeutil package (#28623)
This commit is contained in:
Earl Warren 2024-01-01 16:15:15 +01:00
parent 3bb73e0813
commit acc6b51be2
No known key found for this signature in database
GPG key ID: 0579CB2928A78A00
1 changed files with 7 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
models/auth/session_test.go
View file

@ -17,7 +17,7 @@ import (
func TestAuthSession(t *testing.T) { func TestAuthSession(t *testing.T) {
assert.NoError(t, unittest.PrepareTestDatabase()) assert.NoError(t, unittest.PrepareTestDatabase())
defer timeutil.Unset() defer timeutil.MockUnset()
key := "I-Like-Free-Software" key := "I-Like-Free-Software"
@ -31,7 +31,7 @@ func TestAuthSession(t *testing.T) {
assert.NoError(t, err) assert.NoError(t, err)
now := time.Date(2021, 1, 1, 0, 0, 0, 0, time.UTC) now := time.Date(2021, 1, 1, 0, 0, 0, 0, time.UTC)
timeutil.Set(now) timeutil.MockSet(now)
// New session is created. // New session is created.
sess, err := auth.ReadSession(db.DefaultContext, key) sess, err := auth.ReadSession(db.DefaultContext, key)
@ -54,13 +54,13 @@ func TestAuthSession(t *testing.T) {
t.Run("Update session", func(t *testing.T) { t.Run("Update session", func(t *testing.T) {
data := []byte{0xba, 0xdd, 0xc0, 0xde} data := []byte{0xba, 0xdd, 0xc0, 0xde}
now := time.Date(2022, 1, 1, 0, 0, 0, 0, time.UTC) now := time.Date(2022, 1, 1, 0, 0, 0, 0, time.UTC)
timeutil.Set(now) timeutil.MockSet(now)
// Update session. // Update session.
err := auth.UpdateSession(db.DefaultContext, key, data) err := auth.UpdateSession(db.DefaultContext, key, data)
assert.NoError(t, err) assert.NoError(t, err)
timeutil.Set(time.Date(2021, 1, 1, 0, 0, 0, 0, time.UTC)) timeutil.MockSet(time.Date(2021, 1, 1, 0, 0, 0, 0, time.UTC))
// Read updated session. // Read updated session.
// Ensure data is updated and expiry is set from the update session call. // Ensure data is updated and expiry is set from the update session call.
@ -70,7 +70,7 @@ func TestAuthSession(t *testing.T) {
assert.EqualValues(t, data, sess.Data) assert.EqualValues(t, data, sess.Data)
assert.EqualValues(t, now.Unix(), sess.Expiry) assert.EqualValues(t, now.Unix(), sess.Expiry)
timeutil.Set(now) timeutil.MockSet(now)
}) })
t.Run("Delete session", func(t *testing.T) { t.Run("Delete session", func(t *testing.T) {
@ -97,13 +97,13 @@ func TestAuthSession(t *testing.T) {
}) })
t.Run("Cleanup sessions", func(t *testing.T) { t.Run("Cleanup sessions", func(t *testing.T) {
timeutil.Set(time.Date(2023, 1, 1, 0, 0, 0, 0, time.UTC)) timeutil.MockSet(time.Date(2023, 1, 1, 0, 0, 0, 0, time.UTC))
_, err := auth.ReadSession(db.DefaultContext, "sess-1") _, err := auth.ReadSession(db.DefaultContext, "sess-1")
assert.NoError(t, err) assert.NoError(t, err)
// One minute later. // One minute later.
timeutil.Set(time.Date(2023, 1, 1, 0, 1, 0, 0, time.UTC)) timeutil.MockSet(time.Date(2023, 1, 1, 0, 1, 0, 0, time.UTC))
_, err = auth.ReadSession(db.DefaultContext, "sess-2") _, err = auth.ReadSession(db.DefaultContext, "sess-2")
assert.NoError(t, err) assert.NoError(t, err)