Function TestNewLogicMemoryMiddleware

pkg/middleware/logic_memory_test.go:16–54

(t *testing.T)

)

func TestNewLogicMemoryMiddleware(t *testing.T) {
	store := logic.NewInMemoryStore()
	manager, err := logic.NewManager(&logic.ManagerConfig{Store: store})
	require.NoError(t, err)

	t.Run("valid config", func(t *testing.T) {
		mw, err := NewLogicMemoryMiddleware(&LogicMemoryMiddlewareConfig{
			Manager:         manager,
			EnableCapture:   true,
			EnableInjection: true,
		})
		require.NoError(t, err)
		require.NotNil(t, mw)
		assert.Equal(t, "logic_memory", mw.Name())
	})

	t.Run("nil config", func(t *testing.T) {
		_, err := NewLogicMemoryMiddleware(nil)
		assert.Error(t, err)
		assert.Contains(t, err.Error(), "config is required")
	})

	t.Run("nil manager", func(t *testing.T) {
		_, err := NewLogicMemoryMiddleware(&LogicMemoryMiddlewareConfig{})
		assert.Error(t, err)
		assert.Contains(t, err.Error(), "manager is required")
	})

	t.Run("default values", func(t *testing.T) {
		mw, err := NewLogicMemoryMiddleware(&LogicMemoryMiddlewareConfig{
			Manager: manager,
		})
		require.NoError(t, err)
		assert.Equal(t, 5, mw.config.MaxMemories)
		assert.Equal(t, 0.6, mw.config.MinConfidence)
		assert.Equal(t, 7, mw.config.Priority)
		assert.Equal(t, "system_prompt_end", mw.config.InjectionPoint)
	})
}

func TestLogicMemoryMiddleware_WrapModelCall(t *testing.T) {
	ctx := context.Background()