// Copyright 2026 The Gitea Authors. All rights reserved.
// SPDX-License-Identifier: MIT

package container

import (
	"testing"

	specs "github.com/opencontainers/image-spec/specs-go/v1"
	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
)

func TestParsePlatform(t *testing.T) {
	t.Run("empty input returns nil platform without error", func(t *testing.T) {
		got, err := parsePlatform("")
		require.NoError(t, err)
		assert.Nil(t, got)
	})

	for input, want := range map[string]specs.Platform{
		"linux/amd64":    {OS: "linux", Architecture: "amd64"},
		"linux/arm/v7":   {OS: "linux", Architecture: "arm", Variant: "v7"},
		"Linux/AMD64/V8": {OS: "linux", Architecture: "amd64", Variant: "v8"},
	} {
		t.Run(input, func(t *testing.T) {
			got, err := parsePlatform(input)
			require.NoError(t, err)
			assert.Equal(t, &want, got)
		})
	}

	for _, bad := range []string{
		"amd64",
		"linux",
		"linux/",
		"/amd64",
		"/",
		"//",
		"linux/arm/",
		"linux/arm/v7/extra",
	} {
		t.Run("rejects "+bad, func(t *testing.T) {
			got, err := parsePlatform(bad)
			require.Error(t, err)
			assert.Nil(t, got)
		})
	}
}

func TestPlatformMatches(t *testing.T) {
	for _, testcase := range []struct {
		requested, image string
		want             bool
	}{
		{"linux/amd64", "linux/amd64", true},
		{"linux/amd64/v1", "linux/amd64", true},
		{"linux/arm64", "linux/amd64", false},
		{"linux/arm64/v8", "linux/arm64", true},
		{"linux/arm/v7", "linux/arm/v7", true},
		{"linux/arm", "linux/arm/v7", true},
		{"linux/arm/v6", "linux/arm/v7", false},
		{"linux/arm", "linux/arm/v6", true},
		{"windows/arm/v7", "linux/arm/v7", false},
	} {
		requested, err := parsePlatform(testcase.requested)
		require.NoError(t, err)
		image, err := parsePlatform(testcase.image)
		require.NoError(t, err)
		assert.Equal(t, testcase.want, platformMatches(requested, image), "%s on %s", testcase.requested, testcase.image)
	}
}