whatcanGOwrong

go/pkg/mod/go.starlark.net@v0.0.0-20231101134539-556fd59b42f6/starlark/value_test.go

@@ -0,0 +1,157 @@
+// Copyright 2017 The Bazel Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style
+// license that can be found in the LICENSE file.
+
+package starlark_test
+
+// This file defines tests of the Value API.
+
+import (
+	"fmt"
+	"testing"
+
+	"github.com/google/go-cmp/cmp"
+	"go.starlark.net/starlark"
+)
+
+func TestStringMethod(t *testing.T) {
+	s := starlark.String("hello")
+	for i, test := range [][2]string{
+		// quoted string:
+		{s.String(), `"hello"`},
+		{fmt.Sprintf("%s", s), `"hello"`},
+		{fmt.Sprintf("%+s", s), `"hello"`},
+		{fmt.Sprintf("%v", s), `"hello"`},
+		{fmt.Sprintf("%+v", s), `"hello"`},
+		// unquoted:
+		{s.GoString(), `hello`},
+		{fmt.Sprintf("%#v", s), `hello`},
+	} {
+		got, want := test[0], test[1]
+		if got != want {
+			t.Errorf("#%d: got <<%s>>, want <<%s>>", i, got, want)
+		}
+	}
+}
+
+func TestListAppend(t *testing.T) {
+	l := starlark.NewList(nil)
+	l.Append(starlark.String("hello"))
+	res, ok := starlark.AsString(l.Index(0))
+	if !ok {
+		t.Errorf("failed list.Append() got: %s, want: starlark.String", l.Index(0).Type())
+	}
+	if res != "hello" {
+		t.Errorf("failed list.Append() got: %+v, want: hello", res)
+	}
+}
+
+func TestParamDefault(t *testing.T) {
+	tests := []struct {
+		desc         string
+		starFn       string
+		wantDefaults []starlark.Value
+	}{
+		{
+			desc:         "function with all required params",
+			starFn:       "all_required",
+			wantDefaults: []starlark.Value{nil, nil, nil},
+		},
+		{
+			desc:   "function with all optional params",
+			starFn: "all_opt",
+			wantDefaults: []starlark.Value{
+				starlark.String("a"),
+				starlark.None,
+				starlark.String(""),
+			},
+		},
+		{
+			desc:   "function with required and optional params",
+			starFn: "mix_required_opt",
+			wantDefaults: []starlark.Value{
+				nil,
+				nil,
+				starlark.String("c"),
+				starlark.String("d"),
+			},
+		},
+		{
+			desc:   "function with required, optional, and varargs params",
+			starFn: "with_varargs",
+			wantDefaults: []starlark.Value{
+				nil,
+				starlark.String("b"),
+				nil,
+			},
+		},
+		{
+			desc:   "function with required, optional, varargs, and keyword-only params",
+			starFn: "with_varargs_kwonly",
+			wantDefaults: []starlark.Value{
+				nil,
+				starlark.String("b"),
+				starlark.String("c"),
+				nil,
+				nil,
+			},
+		},
+		{
+			desc:   "function with required, optional, and keyword-only params",
+			starFn: "with_kwonly",
+			wantDefaults: []starlark.Value{
+				nil,
+				starlark.String("b"),
+				starlark.String("c"),
+				nil,
+			},
+		},
+		{
+			desc:   "function with required, optional, and kwargs params",
+			starFn: "with_kwargs",
+			wantDefaults: []starlark.Value{
+				nil,
+				starlark.String("b"),
+				starlark.String("c"),
+				nil,
+			},
+		},
+		{
+			desc:   "function with required, optional, varargs, kw-only, and kwargs params",
+			starFn: "with_varargs_kwonly_kwargs",
+			wantDefaults: []starlark.Value{
+				nil,
+				starlark.String("b"),
+				starlark.String("c"),
+				nil,
+				starlark.String("e"),
+				nil,
+				nil,
+			},
+		},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.desc, func(t *testing.T) {
+			thread := &starlark.Thread{}
+			filename := "testdata/function_param.star"
+			globals, err := starlark.ExecFile(thread, filename, nil, nil)
+			if err != nil {
+				t.Fatalf("ExecFile(%v, %q) failed: %v", thread, filename, err)
+			}
+
+			fn, ok := globals[tt.starFn].(*starlark.Function)
+			if !ok {
+				t.Fatalf("value %v is not a Starlark function", globals[tt.starFn])
+			}
+
+			var paramDefaults []starlark.Value
+			for i := 0; i < fn.NumParams(); i++ {
+				paramDefaults = append(paramDefaults, fn.ParamDefault(i))
+			}
+			if diff := cmp.Diff(tt.wantDefaults, paramDefaults); diff != "" {
+				t.Errorf("param defaults got diff (-want +got):\n%s", diff)
+			}
+		})
+	}
+}