Function TestInMemoryStore_GetStats

pkg/memory/logic/store_test.go:252–295

(t *testing.T)

}

func TestInMemoryStore_GetStats(t *testing.T) {
	store := NewInMemoryStore()
	defer func() { _ = store.Close() }()

	ctx := context.Background()
	namespace := "user:123"

	require.NoError(t, store.Save(ctx, &LogicMemory{
		Namespace: namespace,
		Scope:     ScopeUser,
		Type:      "preference",
		Key:       "key1",
		Provenance: &memory.MemoryProvenance{
			Confidence: 0.8,
		},
	}))
	require.NoError(t, store.Save(ctx, &LogicMemory{
		Namespace: namespace,
		Scope:     ScopeSession,
		Type:      "preference",
		Key:       "key2",
		Provenance: &memory.MemoryProvenance{
			Confidence: 0.6,
		},
	}))
	require.NoError(t, store.Save(ctx, &LogicMemory{
		Namespace: namespace,
		Scope:     ScopeUser,
		Type:      "behavior",
		Key:       "key3",
		Provenance: &memory.MemoryProvenance{
			Confidence: 0.7,
		},
	}))

	stats, err := store.GetStats(ctx, namespace)
	require.NoError(t, err)
	assert.Equal(t, 3, stats.TotalCount)
	assert.Equal(t, 2, stats.CountByType["preference"])
	assert.Equal(t, 1, stats.CountByType["behavior"])
	assert.Equal(t, 2, stats.CountByScope[ScopeUser])
	assert.Equal(t, 1, stats.CountByScope[ScopeSession])
	assert.InDelta(t, 0.7, stats.AverageConfidence, 0.01)
}

func TestInMemoryStore_Prune(t *testing.T) {
	store := NewInMemoryStore()