Function TestInMemoryService_Concurrency

pkg/session/inmemory_test.go:543–586

(t *testing.T)

}

func TestInMemoryService_Concurrency(t *testing.T) {
	ctx := context.Background()
	service := NewInMemoryService()

	sess, _ := service.Create(ctx, &CreateRequest{
		AppName: "test-app",
		UserID:  "user-1",
		AgentID: "agent-1",
	})

	t.Run("并发追加事件", func(t *testing.T) {
		done := make(chan bool)
		numGoroutines := 10
		eventsPerGoroutine := 10

		for i := range numGoroutines {
			go func(id int) {
				for j := range eventsPerGoroutine {
					event := &Event{
						ID:           "evt-concurrent-" + string(rune('0'+id)) + "-" + string(rune('0'+j)),
						Timestamp:    time.Now(),
						InvocationID: "inv-1",
						AgentID:      "agent-1",
						Branch:       "root",
						Author:       "user",
					}
					if err := service.AppendEvent(ctx, sess.ID(), event); err != nil {
						t.Errorf("AppendEvent failed: %v", err)
					}
				}
				done <- true
			}(i)
		}

		// 等待所有 goroutine 完成
		for range numGoroutines {
			<-done
		}

		// 验证所有事件都已追加
		events, _ := service.GetEvents(ctx, sess.ID(), nil)
		assert.Len(t, events, numGoroutines*eventsPerGoroutine)
	})
}

func TestInMemoryService_StateScopes(t *testing.T) {
	t.Skip("Skipping: inmemory service implementation has issues with state scopes")