Function TestParseDecl

parse/parse_test.go:46–224

(t *testing.T)

}

func TestParseDecl(t *testing.T) {
	inclConstraint := ast.NewInclusionConstraint([]ast.Atom{
		ast.NewAtom("bar", ast.Variable{"X"}),
		ast.NewAtom("baz", ast.Variable{"X"}),
		ast.NewAtom("bak", ast.Variable{"X"}, ast.Variable{"Y"}),
	})
	tests := []struct {
		name string
		str  string
		want []ast.Decl
	}{
		{
			name: "one decl",
			str:  "Decl foo(X,Y).",
			want: []ast.Decl{makeDecl(t, ast.NewAtom("foo", ast.Variable{"X"}, ast.Variable{"Y"}), emptyDoc, nil, nil)},
		},
		{
			name: "one decl one bound",
			str:  "Decl foo(X,Y) bound [/string, fn:List(/string)].",
			want: []ast.Decl{makeDecl(t, ast.NewAtom("foo", ast.Variable{"X"}, ast.Variable{"Y"}),
				nil,
				[]ast.BoundDecl{
					ast.NewBoundDecl(ast.StringBound, ast.ApplyFn{symbols.ListType, []ast.BaseTerm{ast.StringBound}}),
				},
				nil)},
		},
		{
			name: "one decl one bound fancy syntax",
			str:  "Decl foo(X,Y) bound [/string, .List</string>].",
			want: []ast.Decl{makeDecl(t, ast.NewAtom("foo", ast.Variable{"X"}, ast.Variable{"Y"}),
				nil,
				[]ast.BoundDecl{
					ast.NewBoundDecl(ast.StringBound, ast.ApplyFn{Function: ast.FunctionSym{"fn:List", -1}, Args: []ast.BaseTerm{ast.StringBound}}),
				},
				nil)},
		},
		{
			name: "one decl two bounds w/ whitespace",
			str:  "Decl foo(X,Y) bound[   /string, /string ]bound[/number, /number].",
			want: []ast.Decl{makeDecl(t, ast.NewAtom("foo", ast.Variable{"X"}, ast.Variable{"Y"}),
				emptyDoc,
				[]ast.BoundDecl{
					ast.NewBoundDecl(ast.StringBound, ast.StringBound),
					ast.NewBoundDecl(ast.NumberBound, ast.NumberBound),
				},
				nil)},
		},
		{
			name: "one decl one inclusion",
			str:  "Decl foo(X,Y) inclusion[bar(X), baz(X), bak(X, Y)].",
			want: []ast.Decl{makeDecl(t, ast.NewAtom("foo", ast.Variable{"X"}, ast.Variable{"Y"}), emptyDoc, nil, &inclConstraint)},
		},
		{
			name: "one decl one bound one inclusion",
			str:  "Decl foo(X,Y) bound[/string, .Pair</string, /number>] inclusion[bar(X), baz(X), bak(X, Y)].",
			want: []ast.Decl{
				makeDecl(t, ast.NewAtom("foo", ast.Variable{"X"}, ast.Variable{"Y"}),
					emptyDoc,