95b2547c54
- Deleted session-related functions (UserID, GuestID, UserRoles, UserOrGuestID) from the agent package to streamline the codebase. - Removed RAG-related code and references, simplifying the agent's architecture. - Updated API and load functions to reflect these changes, ensuring consistency across the agent module. - Enhanced test coverage by cleaning up deprecated test cases related to removed functionalities.
146 lines
3.4 KiB
Go
146 lines
3.4 KiB
Go
package assistant
|
|
|
|
// func TestCache_Basic(t *testing.T) {
|
|
// cache := NewCache(2)
|
|
|
|
// // Test empty cache
|
|
// if cache.Len() != 0 {
|
|
// t.Errorf("Expected empty cache, got length %d", cache.Len())
|
|
// }
|
|
|
|
// // Test adding items
|
|
// assistant1 := &Assistant{ID: "1", Name: "Test1"}
|
|
// assistant2 := &Assistant{ID: "2", Name: "Test2"}
|
|
|
|
// cache.Put(assistant1)
|
|
// cache.Put(assistant2)
|
|
|
|
// if cache.Len() != 2 {
|
|
// t.Errorf("Expected cache length 2, got %d", cache.Len())
|
|
// }
|
|
|
|
// // Test getting items
|
|
// if a, exists := cache.Get("1"); !exists || a.ID != "1" {
|
|
// t.Error("Failed to get assistant1")
|
|
// }
|
|
|
|
// if a, exists := cache.Get("2"); !exists || a.ID != "2" {
|
|
// t.Error("Failed to get assistant2")
|
|
// }
|
|
// }
|
|
|
|
// func TestCache_LRU(t *testing.T) {
|
|
// cache := NewCache(2)
|
|
|
|
// assistant1 := &Assistant{ID: "1", Name: "Test1"}
|
|
// assistant2 := &Assistant{ID: "2", Name: "Test2"}
|
|
// assistant3 := &Assistant{ID: "3", Name: "Test3"}
|
|
|
|
// // Add first two items
|
|
// cache.Put(assistant1)
|
|
// cache.Put(assistant2)
|
|
|
|
// // Access assistant1 to make it most recently used
|
|
// cache.Get("1")
|
|
|
|
// // Add third item, should evict assistant2
|
|
// cache.Put(assistant3)
|
|
|
|
// // Check assistant2 was evicted
|
|
// if _, exists := cache.Get("2"); exists {
|
|
// t.Error("Assistant2 should have been evicted")
|
|
// }
|
|
|
|
// // Check assistant1 and assistant3 are still present
|
|
// if _, exists := cache.Get("1"); !exists {
|
|
// t.Error("Assistant1 should still be in cache")
|
|
// }
|
|
// if _, exists := cache.Get("3"); !exists {
|
|
// t.Error("Assistant3 should be in cache")
|
|
// }
|
|
// }
|
|
|
|
// func TestCache_Remove(t *testing.T) {
|
|
// cache := NewCache(2)
|
|
|
|
// assistant1 := &Assistant{ID: "1", Name: "Test1"}
|
|
// cache.Put(assistant1)
|
|
|
|
// // Test remove existing item
|
|
// cache.Remove("1")
|
|
// if cache.Len() != 0 {
|
|
// t.Error("Cache should be empty after removing item")
|
|
// }
|
|
|
|
// // Test remove non-existing item
|
|
// cache.Remove("nonexistent")
|
|
// if cache.Len() != 0 {
|
|
// t.Error("Cache length should not change when removing non-existent item")
|
|
// }
|
|
// }
|
|
|
|
// func TestCache_Clear(t *testing.T) {
|
|
// cache := NewCache(2)
|
|
|
|
// assistant1 := &Assistant{ID: "1", Name: "Test1"}
|
|
// assistant2 := &Assistant{ID: "2", Name: "Test2"}
|
|
|
|
// cache.Put(assistant1)
|
|
// cache.Put(assistant2)
|
|
|
|
// cache.Clear()
|
|
// if cache.Len() != 0 {
|
|
// t.Error("Cache should be empty after clear")
|
|
// }
|
|
// }
|
|
|
|
// func TestCache_Concurrent(t *testing.T) {
|
|
// cache := NewCache(100)
|
|
// var wg sync.WaitGroup
|
|
// workers := 10
|
|
// iterations := 100
|
|
|
|
// // Concurrent writes
|
|
// for i := 0; i < workers; i++ {
|
|
// wg.Add(1)
|
|
// go func(workerID int) {
|
|
// defer wg.Done()
|
|
// for j := 0; j < iterations; j++ {
|
|
// assistant := &Assistant{
|
|
// ID: string(rune('A' + workerID)),
|
|
// Name: "Test",
|
|
// }
|
|
// cache.Put(assistant)
|
|
// }
|
|
// }(i)
|
|
// }
|
|
|
|
// // Concurrent reads
|
|
// for i := 0; i < workers; i++ {
|
|
// wg.Add(1)
|
|
// go func(workerID int) {
|
|
// defer wg.Done()
|
|
// for j := 0; j < iterations; j++ {
|
|
// cache.Get(string(rune('A' + workerID)))
|
|
// }
|
|
// }(i)
|
|
// }
|
|
|
|
// wg.Wait()
|
|
// }
|
|
|
|
// func TestCache_NilInput(t *testing.T) {
|
|
// cache := NewCache(2)
|
|
|
|
// // Test putting nil assistant
|
|
// cache.Put(nil)
|
|
// if cache.Len() != 0 {
|
|
// t.Error("Cache should not store nil assistant")
|
|
// }
|
|
|
|
// // Test putting assistant with empty ID
|
|
// cache.Put(&Assistant{ID: "", Name: "Test"})
|
|
// if cache.Len() != 0 {
|
|
// t.Error("Cache should not store assistant with empty ID")
|
|
// }
|
|
// }
|