Function TestSubsetFunction

common/env/env_test.go:1142–1241

(t *testing.T)

}

func TestSubsetFunction(t *testing.T) {
	tests := []struct {
		name     string
		lib      *LibrarySubset
		orig     *decls.FunctionDecl
		subset   *decls.FunctionDecl
		included bool
	}{
		{
			name:     "nil lib, included",
			orig:     mustNewFunction(t, "size", decls.Overload("size_string", []*types.Type{types.StringType}, types.IntType)),
			subset:   mustNewFunction(t, "size", decls.Overload("size_string", []*types.Type{types.StringType}, types.IntType)),
			included: true,
		},
		{
			name:     "empty, included",
			lib:      NewLibrarySubset(),
			orig:     mustNewFunction(t, "size", decls.Overload("size_string", []*types.Type{types.StringType}, types.IntType)),
			subset:   mustNewFunction(t, "size", decls.Overload("size_string", []*types.Type{types.StringType}, types.IntType)),
			included: true,
		},
		{
			name:     "empty, disabled",
			lib:      NewLibrarySubset().SetDisabled(true),
			orig:     mustNewFunction(t, "size", decls.Overload("size_string", []*types.Type{types.StringType}, types.IntType)),
			included: false,
		},
		{
			name: "lib, not included allow-list",
			lib: NewLibrarySubset().AddIncludedFunctions([]*Function{
				{Name: "int"},
			}...),
			orig:     mustNewFunction(t, "size", decls.Overload("size_string", []*types.Type{types.StringType}, types.IntType)),
			included: false,
		},
		{
			name: "lib, included whole function",
			lib: NewLibrarySubset().AddIncludedFunctions([]*Function{
				{Name: "size"},
			}...),
			orig:     mustNewFunction(t, "size", decls.Overload("size_string", []*types.Type{types.StringType}, types.IntType)),
			subset:   mustNewFunction(t, "size", decls.Overload("size_string", []*types.Type{types.StringType}, types.IntType)),
			included: true,
		},
		{
			name: "lib, included overload subset",
			lib: NewLibrarySubset().AddIncludedFunctions([]*Function{
				{Name: "size", Overloads: []*Overload{{ID: "size_string"}}},
			}...),
			orig: mustNewFunction(t, "size",
				decls.Overload("size_string", []*types.Type{types.StringType}, types.IntType),
				decls.Overload("size_list", []*types.Type{types.NewListType(types.NewTypeParamType("T"))}, types.IntType),
			),
			subset:   mustNewFunction(t, "size", decls.Overload("size_string", []*types.Type{types.StringType}, types.IntType)),
			included: true,
		},
		{
			name: "lib, included deny-list",