Function TestTemporalStore_WithIntervalTree

factstore/interval_tree_test.go:260–304

(t *testing.T)

}

func TestTemporalStore_WithIntervalTree(t *testing.T) {
	store := NewTemporalStore()

	atom := ast.Atom{
		Predicate: ast.PredicateSym{Symbol: "test", Arity: 1},
		Args:      []ast.BaseTerm{ast.String("alice")},
	}

	interval := makeTestInterval(100, 200)

	// Add fact
	added, err := store.Add(atom, interval)
	if err != nil {
		t.Fatalf("Add failed: %v", err)
	}
	if !added {
		t.Error("Expected Add to return true")
	}

	// Check count
	if store.EstimateFactCount() != 1 {
		t.Errorf("Expected count 1, got %d", store.EstimateFactCount())
	}

	// Query at valid time
	query := ast.NewQuery(atom.Predicate)
	foundAt := false
	store.GetFactsAt(query, time.Unix(0, 150), func(tf TemporalFact) error {
		foundAt = true
		return nil
	})
	if !foundAt {
		t.Error("Expected to find fact at valid time")
	}

	// Query at invalid time
	foundAt = false
	store.GetFactsAt(query, time.Unix(0, 50), func(tf TemporalFact) error {
		foundAt = true
		return nil
	})
	if foundAt {
		t.Error("Did not expect to find fact at invalid time")
	}
}

func TestTemporalStore_IntervalTreeLimit(t *testing.T) {
	store := NewTemporalStore(WithMaxIntervalsPerAtom(3))