Function TestTemporalStore_GetFactsAt

factstore/temporal_test.go:149–210

(t *testing.T)

}

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

	// Alice employed from 2020-2023
	aliceEmployed := ast.NewAtom("employed", name("/alice"))
	store.Add(aliceEmployed, ast.TimeInterval(
		ast.Date(2020, 1, 1),
		ast.Date(2023, 12, 31),
	))

	// Bob employed from 2022 onwards
	bobEmployed := ast.NewAtom("employed", name("/bob"))
	store.Add(bobEmployed, ast.TimeInterval(
		ast.Date(2022, 1, 1),
		ast.Date(2099, 12, 31),
	))

	tests := []struct {
		name      string
		queryTime time.Time
		wantCount int
	}{
		{
			name:      "before both",
			queryTime: ast.Date(2019, 6, 15),
			wantCount: 0,
		},
		{
			name:      "alice only",
			queryTime: ast.Date(2021, 6, 15),
			wantCount: 1,
		},
		{
			name:      "both employed",
			queryTime: ast.Date(2022, 6, 15),
			wantCount: 2,
		},
		{
			name:      "bob only (after alice left)",
			queryTime: ast.Date(2024, 6, 15),
			wantCount: 1,
		},
	}

	query := ast.NewQuery(ast.PredicateSym{Symbol: "employed", Arity: 1})

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			count := 0
			err := store.GetFactsAt(query, tt.queryTime, func(tf TemporalFact) error {
				count++
				return nil
			})
			if err != nil {
				t.Fatalf("GetFactsAt error: %v", err)
			}
			if count != tt.wantCount {
				t.Errorf("GetFactsAt count = %d, want %d", count, tt.wantCount)