package session

import (
	"testing"

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

func TestSession(t *testing.T) {
	session := NewSession("session1")

	// 测试ID
	assert.Equal(t, "session1", session.ID(), "Expected session ID to match")

	// 测试Set和Get
	err := session.Set("key1", "value1")
	require.NoError(t, err, "Expected no error when setting value")
	
	val, err := session.Get("key1")
	require.NoError(t, err, "Expected no error when getting value")
	assert.Equal(t, "value1", val, "Expected value to match")

	// 测试Delete
	err = session.Delete("key1")
	require.NoError(t, err, "Expected no error when deleting value")
	
	val, err = session.Get("key1")
	require.NoError(t, err)
	assert.Nil(t, val, "Expected value to be nil after deletion")
}

func TestSessionIsNew(t *testing.T) {
	session := NewSession("session1")

	// 测试IsNew（新创建的session应该是新的）
	assert.True(t, session.IsNew(), "Expected session to be new")
}

func TestSessionConcurrent(t *testing.T) {
	session := NewSession("session1")

	// 并发Set和Get
	done := make(chan bool, 100)
	for i := 0; i < 100; i++ {
		go func(id int) {
			key := string(rune(id))
			session.Set(key, id)
			val, _ := session.Get(key)
			assert.Equal(t, id, val)
			done <- true
		}(i)
	}

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

func TestSessionClear(t *testing.T) {
	session := NewSession("session1")
	
	// 设置一些值
	session.Set("key1", "value1")
	session.Set("key2", "value2")
	
	// 测试Clear
	err := session.Clear()
	require.NoError(t, err, "Expected no error when clearing")
	
	// 验证值已被清空
	val, _ := session.Get("key1")
	assert.Nil(t, val, "Expected key1 to be nil after clear")
	
	val, _ = session.Get("key2")
	assert.Nil(t, val, "Expected key2 to be nil after clear")
}