package bazel

import (
	"testing"

	"github.com/stretchr/testify/assert"
)

func TestSimplifyLabels(t *testing.T) {
	tests := map[string]string{
		"//pkg/edge/edgeadmin":           "pkg/edge/edgeadmin",
		"//pkg/edge/edgeadmin:edgeadmin": "pkg/edge/edgeadmin",
		"pkg/edge/edgeadmin":             "pkg/edge/edgeadmin",
		"//:helm":                        ":helm",
		"//helm":                         "helm",
		"//helm:helm":                    "helm",
		"//pkg/edge:edge_lib":            "pkg/edge:edge_lib",
	}

	for input, expected := range tests {
		t.Run(input, func(t *testing.T) {
			assert.Equal(t, expected, SimplifyLabel(input))
		})
	}
}

func TestPackageFromLabel(t *testing.T) {
	tcs := map[string]struct {
		label string
		pkg   string
	}{
		"empty":       {"", ""},
		"abbreviated": {"//cmd/f8n/foo", "//cmd/f8n/foo"},
		"standard":    {"//cmd/f8n/foo:label", "//cmd/f8n/foo"},
	}

	for name, tc := range tcs {
		t.Run(name, func(t *testing.T) {
			actual := PackageFromLabel(tc.label)
			assert.Equal(t, tc.pkg, actual)
		})
	}
}