// Copyright (C) 2022 Storj Labs, Inc. // See LICENSE for copying information. package satellitedb_test import ( "testing" "time" "github.com/stretchr/testify/require" "storj.io/common/testcontext" "storj.io/common/testrand" "storj.io/storj/satellite" "storj.io/storj/satellite/satellitedb/satellitedbtest" ) func TestWebappSessionsCreate(t *testing.T) { satellitedbtest.Run(t, func(ctx *testcontext.Context, t *testing.T, db satellite.DB) { sessions := db.Console().WebappSessions() id := testrand.UUID() userID := testrand.UUID() address := "127.0.0.1" userAgent := "test_user_agent" expiresAt := time.Now().Add(24 * time.Hour).Truncate(time.Second) session, err := sessions.Create(ctx, id, userID, address, userAgent, expiresAt) require.NoError(t, err) require.Equal(t, id, session.ID) require.Equal(t, userID, session.UserID) require.Equal(t, address, session.Address) require.Equal(t, userAgent, session.UserAgent) require.Equal(t, expiresAt, session.ExpiresAt) _, err = sessions.Create(ctx, id, userID, address, userAgent, expiresAt) require.Error(t, err) }) } func TestWebappSessionsGetBySessionID(t *testing.T) { satellitedbtest.Run(t, func(ctx *testcontext.Context, t *testing.T, db satellite.DB) { sessions := db.Console().WebappSessions() id := testrand.UUID() userID := testrand.UUID() address := "127.0.0.1" userAgent := "test_user_agent" expiresAt := time.Now().Add(24 * time.Hour).Truncate(time.Second) session, err := sessions.Create(ctx, id, userID, address, userAgent, expiresAt) require.NoError(t, err) require.Equal(t, id, session.ID) require.Equal(t, userID, session.UserID) require.Equal(t, address, session.Address) require.Equal(t, userAgent, session.UserAgent) require.Equal(t, expiresAt, session.ExpiresAt) session, err = sessions.GetBySessionID(ctx, session.ID) require.NoError(t, err) require.Equal(t, id, session.ID) require.Equal(t, userID, session.UserID) require.Equal(t, address, session.Address) require.Equal(t, userAgent, session.UserAgent) require.Equal(t, expiresAt, session.ExpiresAt) }) } func TestWebappSessionsGetAllByUserID(t *testing.T) { satellitedbtest.Run(t, func(ctx *testcontext.Context, t *testing.T, db satellite.DB) { sessions := db.Console().WebappSessions() id1 := testrand.UUID() userID := testrand.UUID() address := "127.0.0.1" userAgent := "test_user_agent" expiresAt := time.Now().Add(24 * time.Hour).Truncate(time.Second) session1, err := sessions.Create(ctx, id1, userID, address, userAgent, expiresAt) require.NoError(t, err) require.Equal(t, id1, session1.ID) require.Equal(t, userID, session1.UserID) require.Equal(t, address, session1.Address) require.Equal(t, userAgent, session1.UserAgent) require.Equal(t, expiresAt, session1.ExpiresAt) id2 := testrand.UUID() session2, err := sessions.Create(ctx, id2, userID, address, userAgent, expiresAt) require.NoError(t, err) require.Equal(t, id2, session2.ID) require.Equal(t, userID, session2.UserID) require.Equal(t, address, session2.Address) require.Equal(t, userAgent, session2.UserAgent) require.Equal(t, expiresAt, session2.ExpiresAt) allSessions, err := sessions.GetAllByUserID(ctx, userID) require.NoError(t, err) require.Len(t, allSessions, 2) var foundSession1, foundSession2 bool if id1 == allSessions[0].ID || id1 == allSessions[1].ID { foundSession1 = true } if id2 == allSessions[0].ID || id2 == allSessions[1].ID { foundSession2 = true } require.True(t, foundSession1) require.True(t, foundSession2) require.Equal(t, userID, allSessions[0].UserID) require.Equal(t, address, allSessions[0].Address) require.Equal(t, userAgent, allSessions[0].UserAgent) require.Equal(t, expiresAt, allSessions[0].ExpiresAt) require.Equal(t, userID, allSessions[1].UserID) require.Equal(t, address, allSessions[1].Address) require.Equal(t, userAgent, allSessions[1].UserAgent) require.Equal(t, expiresAt, allSessions[1].ExpiresAt) }) } func TestWebappSessionsDeleteBySessionID(t *testing.T) { satellitedbtest.Run(t, func(ctx *testcontext.Context, t *testing.T, db satellite.DB) { sessions := db.Console().WebappSessions() id1 := testrand.UUID() userID := testrand.UUID() address := "127.0.0.1" userAgent := "test_user_agent" expiresAt := time.Now().Add(24 * time.Hour).Truncate(time.Second) session1, err := sessions.Create(ctx, id1, userID, address, userAgent, expiresAt) require.NoError(t, err) id2 := testrand.UUID() session2, err := sessions.Create(ctx, id2, userID, address, userAgent, expiresAt) require.NoError(t, err) require.NoError(t, sessions.DeleteBySessionID(ctx, session1.ID)) _, err = sessions.GetBySessionID(ctx, session1.ID) require.Error(t, err) _, err = sessions.GetBySessionID(ctx, session2.ID) require.NoError(t, err) }) } func TestWebappSessionsDeleteAllByUserID(t *testing.T) { satellitedbtest.Run(t, func(ctx *testcontext.Context, t *testing.T, db satellite.DB) { sessions := db.Console().WebappSessions() id1 := testrand.UUID() userID := testrand.UUID() address := "127.0.0.1" userAgent := "test_user_agent" expiresAt := time.Now().Add(24 * time.Hour).Truncate(time.Second) session1, err := sessions.Create(ctx, id1, userID, address, userAgent, expiresAt) require.NoError(t, err) require.Equal(t, id1, session1.ID) require.Equal(t, userID, session1.UserID) require.Equal(t, address, session1.Address) require.Equal(t, userAgent, session1.UserAgent) require.Equal(t, expiresAt, session1.ExpiresAt) id2 := testrand.UUID() session2, err := sessions.Create(ctx, id2, userID, address, userAgent, expiresAt) require.NoError(t, err) require.Equal(t, id2, session2.ID) require.Equal(t, userID, session2.UserID) require.Equal(t, address, session2.Address) require.Equal(t, userAgent, session2.UserAgent) require.Equal(t, expiresAt, session2.ExpiresAt) deleted, err := sessions.DeleteAllByUserID(ctx, userID) require.NoError(t, err) require.Equal(t, deleted, int64(2)) allSessions, err := sessions.GetAllByUserID(ctx, userID) require.NoError(t, err) require.Len(t, allSessions, 0) }) }