package storage

import (
	"testing"
	"time"

	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
)

func TestMemoryStorage(t *testing.T) {
	storage := NewMemoryStorage(0) // 0表示不过期

	// 测试Create
	sess, err := storage.Create("session1")
	require.NoError(t, err, "Expected no error when creating session")
	assert.Equal(t, "session1", sess.ID(), "Expected session ID to match")

	// 测试Get
	retrieved, err := storage.Get("session1")
	require.NoError(t, err, "Expected no error when getting session")
	require.NotNil(t, retrieved, "Expected session to be non-nil")
	assert.Equal(t, "session1", retrieved.ID(), "Expected session ID to match")

	// 测试Save
	err = storage.Save(sess)
	require.NoError(t, err, "Expected no error when saving session")

	// 测试Delete
	err = storage.Delete("session1")
	require.NoError(t, err, "Expected no error when deleting session")

	// 验证删除（Get应该返回nil, nil）
	retrieved, err = storage.Get("session1")
	require.NoError(t, err) // Get可能返回nil, nil而不是错误
	assert.Nil(t, retrieved, "Expected session to be nil after deletion")
}

func TestMemoryStorageGetNotFound(t *testing.T) {
	storage := NewMemoryStorage(0)

	// 测试获取不存在的session（应该返回nil, nil）
	retrieved, err := storage.Get("nonexistent")
	require.NoError(t, err) // Get返回nil, nil而不是错误
	assert.Nil(t, retrieved, "Expected session to be nil")
}

func TestMemoryStorageConcurrent(t *testing.T) {
	storage := NewMemoryStorage(0)

	// 并发创建和获取session
	done := make(chan bool, 100)
	for i := 0; i < 100; i++ {
		go func(id int) {
			sessionID := string(rune(id))
			sess, _ := storage.Create(sessionID)
			if sess != nil {
				_ = sess.Set("key", id)
				retrieved, _ := storage.Get(sessionID)
				if retrieved != nil {
					val, _ := retrieved.Get("key")
					assert.Equal(t, id, val)
				}
			}
			done <- true
		}(i)
	}

	// 等待所有goroutine完成
	for i := 0; i < 100; i++ {
		<-done
	}
}

func TestMemoryStorageExpiration(t *testing.T) {
	storage := NewMemoryStorage(100 * time.Millisecond)

	// 创建session
	sess, err := storage.Create("session1")
	require.NoError(t, err)
	assert.NotNil(t, sess)

	// 立即获取应该成功
	retrieved, err := storage.Get("session1")
	require.NoError(t, err)
	assert.NotNil(t, retrieved)

	// 等待过期
	time.Sleep(150 * time.Millisecond)

	// 清理过期session
	err = storage.Cleanup()
	require.NoError(t, err)

	// 获取应该返回nil（已过期）
	retrieved, err = storage.Get("session1")
	require.NoError(t, err)
	assert.Nil(t, retrieved, "Expected session to be nil after expiration")
}