219 lines
6.1 KiB
Go
219 lines
6.1 KiB
Go
package entity
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
|
|
"github.com/photoprism/photoprism/pkg/rnd"
|
|
)
|
|
|
|
func TestNewSession(t *testing.T) {
|
|
t.Run("NoSessionData", func(t *testing.T) {
|
|
m := NewSession(UnixDay, UnixHour*6)
|
|
|
|
assert.True(t, rnd.IsSessionID(m.ID))
|
|
assert.False(t, m.CreatedAt.IsZero())
|
|
assert.False(t, m.UpdatedAt.IsZero())
|
|
assert.False(t, m.ExpiresAt().IsZero())
|
|
assert.NotEmpty(t, m.ID)
|
|
assert.NotNil(t, m.Data())
|
|
assert.Equal(t, 0, len(m.Data().Tokens))
|
|
})
|
|
t.Run("EmptySessionData", func(t *testing.T) {
|
|
m := NewSession(UnixDay, UnixHour*6)
|
|
m.SetData(NewSessionData())
|
|
|
|
assert.True(t, rnd.IsSessionID(m.ID))
|
|
assert.False(t, m.CreatedAt.IsZero())
|
|
assert.False(t, m.UpdatedAt.IsZero())
|
|
assert.False(t, m.ExpiresAt().IsZero())
|
|
assert.NotEmpty(t, m.ID)
|
|
assert.NotNil(t, m.Data())
|
|
assert.Equal(t, 0, len(m.Data().Tokens))
|
|
})
|
|
t.Run("WithSessionData", func(t *testing.T) {
|
|
data := NewSessionData()
|
|
data.Tokens = []string{"foo", "bar"}
|
|
m := NewSession(UnixDay, UnixHour*6)
|
|
m.SetData(data)
|
|
|
|
assert.True(t, rnd.IsSessionID(m.ID))
|
|
assert.False(t, m.CreatedAt.IsZero())
|
|
assert.False(t, m.UpdatedAt.IsZero())
|
|
assert.False(t, m.ExpiresAt().IsZero())
|
|
assert.NotEmpty(t, m.ID)
|
|
assert.NotNil(t, m.Data())
|
|
assert.Len(t, m.Data().Tokens, 2)
|
|
assert.Equal(t, "foo", m.Data().Tokens[0])
|
|
assert.Equal(t, "bar", m.Data().Tokens[1])
|
|
})
|
|
}
|
|
|
|
func TestSession_SetData(t *testing.T) {
|
|
t.Run("Nil", func(t *testing.T) {
|
|
m := NewSession(UnixDay, UnixHour*6)
|
|
|
|
assert.NotNil(t, m)
|
|
|
|
sess := m.SetData(nil)
|
|
|
|
assert.NotNil(t, sess)
|
|
assert.NotEmpty(t, sess.ID)
|
|
assert.Equal(t, sess.ID, m.ID)
|
|
})
|
|
}
|
|
|
|
func TestSession_Expires(t *testing.T) {
|
|
t.Run("Set expiry date", func(t *testing.T) {
|
|
m := NewSession(UnixDay, UnixHour)
|
|
initialExpiryDate := m.SessExpires
|
|
m.Expires(time.Date(2035, 01, 15, 12, 30, 0, 0, time.UTC))
|
|
finalExpiryDate := m.SessExpires
|
|
assert.Greater(t, finalExpiryDate, initialExpiryDate)
|
|
|
|
})
|
|
t.Run("Try to set zero date", func(t *testing.T) {
|
|
m := NewSession(UnixDay, UnixHour)
|
|
initialExpiryDate := m.SessExpires
|
|
m.Expires(time.Date(1, 1, 1, 0, 0, 0, 0, time.UTC))
|
|
finalExpiryDate := m.SessExpires
|
|
assert.Equal(t, finalExpiryDate, initialExpiryDate)
|
|
})
|
|
}
|
|
|
|
func TestDeleteExpiredSessions(t *testing.T) {
|
|
assert.Equal(t, 0, DeleteExpiredSessions())
|
|
m := NewSession(UnixDay, UnixHour)
|
|
m.Expires(time.Date(2000, 01, 15, 12, 30, 0, 0, time.UTC))
|
|
m.Save()
|
|
assert.Equal(t, 1, DeleteExpiredSessions())
|
|
}
|
|
|
|
func TestSessionStatusUnauthorized(t *testing.T) {
|
|
m := SessionStatusUnauthorized()
|
|
assert.Equal(t, 401, m.Status)
|
|
assert.IsType(t, &Session{}, m)
|
|
}
|
|
|
|
func TestSessionStatusForbidden(t *testing.T) {
|
|
m := SessionStatusForbidden()
|
|
assert.Equal(t, 403, m.Status)
|
|
assert.IsType(t, &Session{}, m)
|
|
}
|
|
|
|
func TestFindSessionByRefID(t *testing.T) {
|
|
t.Run("Nil", func(t *testing.T) {
|
|
assert.Nil(t, FindSessionByRefID(""))
|
|
})
|
|
t.Run("alice", func(t *testing.T) {
|
|
m := FindSessionByRefID("sessxkkcabcd")
|
|
assert.Equal(t, "alice", m.UserName)
|
|
assert.IsType(t, &Session{}, m)
|
|
})
|
|
}
|
|
|
|
func TestSession_RegenerateID(t *testing.T) {
|
|
m := NewSession(UnixDay, UnixHour)
|
|
initialID := m.ID
|
|
m.RegenerateID()
|
|
finalID := m.ID
|
|
assert.NotEqual(t, initialID, finalID)
|
|
}
|
|
|
|
func TestSession_TimedOut(t *testing.T) {
|
|
t.Run("NewSession", func(t *testing.T) {
|
|
m := NewSession(UnixDay, UnixHour)
|
|
assert.False(t, m.TimeoutAt().IsZero())
|
|
assert.Equal(t, m.ExpiresAt(), m.TimeoutAt())
|
|
assert.False(t, m.TimedOut())
|
|
})
|
|
t.Run("NoExpiration", func(t *testing.T) {
|
|
m := NewSession(0, UnixHour)
|
|
t.Logf("Timeout: %s, Expiration: %s", m.TimeoutAt().String(), m.ExpiresAt())
|
|
assert.True(t, m.TimeoutAt().IsZero())
|
|
assert.Equal(t, m.ExpiresAt(), m.TimeoutAt())
|
|
assert.False(t, m.TimedOut())
|
|
assert.True(t, m.ExpiresAt().IsZero())
|
|
})
|
|
t.Run("NoTimeout", func(t *testing.T) {
|
|
m := NewSession(UnixDay, 0)
|
|
t.Logf("Timeout: %s, Expiration: %s", m.TimeoutAt().String(), m.ExpiresAt())
|
|
assert.False(t, m.TimeoutAt().IsZero())
|
|
assert.Equal(t, m.ExpiresAt(), m.TimeoutAt())
|
|
assert.False(t, m.TimedOut())
|
|
assert.False(t, m.ExpiresAt().IsZero())
|
|
})
|
|
t.Run("TimedOut", func(t *testing.T) {
|
|
m := NewSession(UnixDay, UnixHour)
|
|
utc := UnixTime()
|
|
|
|
m.LastActive = utc - (UnixHour + 1)
|
|
|
|
assert.False(t, m.TimeoutAt().IsZero())
|
|
assert.True(t, m.TimedOut())
|
|
})
|
|
t.Run("NotTimedOut", func(t *testing.T) {
|
|
m := NewSession(UnixDay, UnixHour)
|
|
utc := UnixTime()
|
|
|
|
m.LastActive = utc - (UnixHour - 10)
|
|
|
|
assert.False(t, m.TimeoutAt().IsZero())
|
|
assert.False(t, m.TimedOut())
|
|
})
|
|
}
|
|
|
|
func TestSession_Expired(t *testing.T) {
|
|
t.Run("NewSession", func(t *testing.T) {
|
|
m := NewSession(UnixDay, UnixHour)
|
|
t.Logf("Timeout: %s, Expiration: %s", m.TimeoutAt().String(), m.ExpiresAt())
|
|
assert.False(t, m.ExpiresAt().IsZero())
|
|
assert.False(t, m.Expired())
|
|
assert.False(t, m.TimeoutAt().IsZero())
|
|
assert.False(t, m.TimedOut())
|
|
})
|
|
t.Run("NoExpiration", func(t *testing.T) {
|
|
m := NewSession(0, 0)
|
|
t.Logf("Timeout: %s, Expiration: %s", m.TimeoutAt().String(), m.ExpiresAt())
|
|
assert.True(t, m.ExpiresAt().IsZero())
|
|
assert.False(t, m.Expired())
|
|
assert.True(t, m.TimeoutAt().IsZero())
|
|
assert.False(t, m.TimedOut())
|
|
})
|
|
t.Run("NoExpiration", func(t *testing.T) {
|
|
m := NewSession(0, 0)
|
|
t.Logf("Timeout: %s, Expiration: %s", m.TimeoutAt().String(), m.ExpiresAt())
|
|
assert.True(t, m.ExpiresAt().IsZero())
|
|
assert.False(t, m.Expired())
|
|
assert.True(t, m.TimeoutAt().IsZero())
|
|
assert.False(t, m.TimedOut())
|
|
})
|
|
t.Run("Expired", func(t *testing.T) {
|
|
m := NewSession(UnixDay, UnixHour)
|
|
t.Logf("Timeout: %s, Expiration: %s", m.TimeoutAt().String(), m.ExpiresAt())
|
|
utc := UnixTime()
|
|
|
|
m.SessExpires = utc - 10
|
|
|
|
assert.False(t, m.ExpiresAt().IsZero())
|
|
assert.True(t, m.Expired())
|
|
assert.False(t, m.TimeoutAt().IsZero())
|
|
assert.True(t, m.TimedOut())
|
|
assert.Equal(t, m.ExpiresAt(), m.TimeoutAt())
|
|
})
|
|
t.Run("NotExpired", func(t *testing.T) {
|
|
m := NewSession(UnixDay, UnixHour)
|
|
utc := UnixTime()
|
|
|
|
m.SessExpires = utc + 10
|
|
|
|
assert.False(t, m.ExpiresAt().IsZero())
|
|
assert.False(t, m.Expired())
|
|
assert.False(t, m.TimeoutAt().IsZero())
|
|
assert.False(t, m.TimedOut())
|
|
assert.Equal(t, m.ExpiresAt(), m.TimeoutAt())
|
|
})
|
|
}
|