Function TestTopDown

engine/topdown_test.go:26–82

(t *testing.T)

)

func TestTopDown(t *testing.T) {
	store := factstore.NewSimpleInMemoryStore()
	clauses := []ast.Clause{
		clause(`
missing_required(RequiredList, EnabledList, Witness) :- 
		:list:member(Witness, RequiredList),
		!:list:member(Witness, EnabledList).`),
	}

	missingRequiredDecl, err := ast.NewDecl(atom("missing_required(EnabledList, RequiredList, Witness)"), []ast.Atom{
		atom("mode('+', '+', '-')"),
		atom("deferred()"),
	}, nil, nil)
	if err != nil {
		t.Fatal(err)
	}
	decls := []ast.Decl{missingRequiredDecl}
	if err := analyzeAndEvalProgramWithDecls(t, clauses, decls, store); err != nil {
		t.Errorf("Program evaluation failed %v program %v", err, clauses)
		return
	}
	query, err := functional.EvalAtom(atom("missing_required(['foo'], ['bar'], Witness)"), nil)
	if err != nil {
		t.Fatal(err)
	}
	want, err := functional.EvalAtom(atom("missing_required(['foo'], ['bar'],'foo')"), nil)
	if err != nil {
		t.Fatal(err)
	}
	context := QueryContext{PredToRules: map[ast.PredicateSym][]ast.Clause{
		ast.PredicateSym{"missing_required", 3}: clauses,
	},
		PredToDecl: map[ast.PredicateSym]*ast.Decl{
			ast.PredicateSym{"missing_required", 3}: &missingRequiredDecl,
		},
		Store: store,
	}

	uf := unionfind.New()
	if err != nil {
		t.Fatal(err)
	}
	found := false
	err = context.EvalQuery(query, []ast.ArgMode{ast.ArgModeInput, ast.ArgModeInput, ast.ArgModeOutput}, uf, func(got ast.Atom) error {
		found = true
		if !got.Equals(want) {
			t.Errorf("got: %v want: %v", got, want)
		}
		return nil
	})
	if err != nil {
		t.Fatal(err)
	}
	if !found {
		t.Errorf("got nothing want: %v", want)
	}
}