starlark_import/starlark_import_test.go

// Copyright 2023 Google Inc. All rights reserved.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

package starlark_import

import (
	"strings"
	"testing"

	"go.starlark.net/starlark"
)

func TestBasic(t *testing.T) {
	globals, _, err := runStarlarkFileWithFilesystem("a.bzl", "", map[string]string{
		"a.bzl": `
my_string = "hello, world!"
`})
	if err != nil {
		t.Error(err)
		return
	}

	if globals["my_string"].(starlark.String) != "hello, world!" {
		t.Errorf("Expected %q, got %q", "hello, world!", globals["my_string"].String())
	}
}

func TestLoad(t *testing.T) {
	globals, _, err := runStarlarkFileWithFilesystem("a.bzl", "", map[string]string{
		"a.bzl": `
load("//b.bzl", _b_string = "my_string")
my_string = "hello, " + _b_string
`,
		"b.bzl": `
my_string = "world!"
`})
	if err != nil {
		t.Error(err)
		return
	}

	if globals["my_string"].(starlark.String) != "hello, world!" {
		t.Errorf("Expected %q, got %q", "hello, world!", globals["my_string"].String())
	}
}

func TestLoadRelative(t *testing.T) {
	globals, ninjaDeps, err := runStarlarkFileWithFilesystem("a.bzl", "", map[string]string{
		"a.bzl": `
load(":b.bzl", _b_string = "my_string")
load("//foo/c.bzl", _c_string = "my_string")
my_string = "hello, " + _b_string
c_string = _c_string
`,
		"b.bzl": `
my_string = "world!"
`,
		"foo/c.bzl": `
load(":d.bzl", _d_string = "my_string")
my_string = "hello, " + _d_string
`,
		"foo/d.bzl": `
my_string = "world!"
`})
	if err != nil {
		t.Error(err)
		return
	}

	if globals["my_string"].(starlark.String) != "hello, world!" {
		t.Errorf("Expected %q, got %q", "hello, world!", globals["my_string"].String())
	}

	expectedNinjaDeps := []string{
		"a.bzl",
		"b.bzl",
		"foo/c.bzl",
		"foo/d.bzl",
	}
	if !slicesEqual(ninjaDeps, expectedNinjaDeps) {
		t.Errorf("Expected %v ninja deps, got %v", expectedNinjaDeps, ninjaDeps)
	}
}

func TestLoadCycle(t *testing.T) {
	_, _, err := runStarlarkFileWithFilesystem("a.bzl", "", map[string]string{
		"a.bzl": `
load(":b.bzl", _b_string = "my_string")
my_string = "hello, " + _b_string
`,
		"b.bzl": `
load(":a.bzl", _a_string = "my_string")
my_string = "hello, " + _a_string
`})
	if err == nil || !strings.Contains(err.Error(), "cycle in load graph") {
		t.Errorf("Expected cycle in load graph, got: %v", err)
		return
	}
}

func slicesEqual[T comparable](a []T, b []T) bool {
	if len(a) != len(b) {
		return false
	}
	for i := range a {
		if a[i] != b[i] {
			return false
		}
	}
	return true
}