Function TestParseUnitDecls

parse/parse_test.go:306–365

(t *testing.T)

}

func TestParseUnitDecls(t *testing.T) {
	tests := []struct {
		name string
		str  string
		want []ast.Decl
	}{
		{
			name: "Regular use",
			str:  "Use foo.bar!",
			want: []ast.Decl{
				makeDecl(t, ast.NewAtom("Package"), []ast.Atom{ast.NewAtom("name", ast.String(""))}, nil, nil),
				makeDecl(t, ast.NewAtom("Use"), []ast.Atom{ast.NewAtom("name", ast.String("foo.bar"))}, nil, nil),
			},
		},
		{
			name: "Regular use with package",
			str:  "Package foo! Use bar!",
			want: []ast.Decl{
				makeDecl(t, ast.NewAtom("Package"), []ast.Atom{ast.NewAtom("name", ast.String("foo"))}, nil, nil),
				makeDecl(t, ast.NewAtom("Use"), []ast.Atom{ast.NewAtom("name", ast.String("bar"))}, nil, nil),
			},
		},
		{
			name: "Use with atoms (trailing comma)",
			str:  `Use foo.bar [atom1("hello"), atom2(),]!`,
			want: []ast.Decl{
				makeDecl(t, ast.NewAtom("Package"), []ast.Atom{ast.NewAtom("name", ast.String(""))}, nil, nil),
				makeDecl(t, ast.NewAtom("Use"), []ast.Atom{ast.NewAtom("name", ast.String("foo.bar")), ast.NewAtom("atom1", ast.String("hello")), ast.NewAtom("atom2")}, nil, nil),
			},
		},
		{
			name: "One decl",
			str:  `Decl pred().`,
			want: []ast.Decl{
				makeDecl(t, ast.NewAtom("Package"), []ast.Atom{ast.NewAtom("name", ast.String(""))}, nil, nil),
				makeDecl(t, ast.NewAtom("pred"), nil, nil, nil),
			},
		},
		{
			name: "Multiple uses",
			str:  `Use foo! Use bar!`,
			want: []ast.Decl{
				makeDecl(t, ast.NewAtom("Package"), []ast.Atom{ast.NewAtom("name", ast.String(""))}, nil, nil),
				makeDecl(t, ast.NewAtom("Use"), []ast.Atom{ast.NewAtom("name", ast.String("foo"))}, nil, nil),
				makeDecl(t, ast.NewAtom("Use"), []ast.Atom{ast.NewAtom("name", ast.String("bar"))}, nil, nil),
			},
		},
	}
	for _, test := range tests {
		t.Run(test.name, func(t *testing.T) {
			got, err := Unit(strings.NewReader(test.str))
			if err != nil {
				t.Fatalf("Unit(%v) failed with %v", test.str, err)
			}
			if diff := cmp.Diff(test.want, got.Decls, cmp.Comparer(equals)); diff != "" {
				t.Errorf("Unit(%v) = %v want %v diff %v", test.str, got, test.want, diff)
			}
		})