Function TestRecursiveNextPath

graph/iterator/recursive_test.go:90–120

(t *testing.T)

}

func TestRecursiveNextPath(t *testing.T) {
	ctx := context.TODO()
	qs := rec_test_qs
	start := qs.NodesAllIterator()
	start = Tag(start, "person")
	it := singleHop(qs, "follows")(start)
	and := NewAnd()
	and.AddSubIterator(it)
	fixed := NewFixed()
	fixed.Add(graph.PreFetched(quad.Raw("alice")))
	and.AddSubIterator(fixed)
	r := NewRecursive(and, singleHop(qs, "parent"), 0)

	expected := []string{"fred", "fred", "fred", "fred", "greg", "greg", "greg", "greg"}
	var got []string
	for r.Next(ctx) {
		res := make(map[string]graph.Ref)
		r.TagResults(res)
		got = append(got, quad.ToString(qs.NameOf(res["person"])))
		for r.NextPath(ctx) {
			res := make(map[string]graph.Ref)
			r.TagResults(res)
			got = append(got, quad.ToString(qs.NameOf(res["person"])))
		}
	}
	sort.Strings(expected)
	sort.Strings(got)
	if !reflect.DeepEqual(got, expected) {
		t.Errorf("Failed to check NextPath, got: %v, expected: %v", got, expected)
	}
}