package common

import (
	"errors"
	"github.com/stretchr/testify/assert"
	"testing"
)

func TestCurrentArchAMD64(t *testing.T) {
	origRuntimeGOARCH := RuntimeGOARCH
	defer func() { RuntimeGOARCH = origRuntimeGOARCH }()
	RuntimeGOARCH = "amd64"
	assert.Equal(t, "x86_64", CurrentArch())
}

func TestCurrentArchARM64(t *testing.T) {
	origRuntimeGOARCH := RuntimeGOARCH
	defer func() { RuntimeGOARCH = origRuntimeGOARCH }()
	RuntimeGOARCH = "arm64"
	assert.Equal(t, "aarch64", CurrentArch())
}

func TestCurrentArchPPC64LE(t *testing.T) {
	origRuntimeGOARCH := RuntimeGOARCH
	defer func() { RuntimeGOARCH = origRuntimeGOARCH }()
	RuntimeGOARCH = "ppc64le"
	assert.Equal(t, "ppc64le", CurrentArch())
}

func TestCurrentArchS390X(t *testing.T) {
	origRuntimeGOARCH := RuntimeGOARCH
	defer func() { RuntimeGOARCH = origRuntimeGOARCH }()
	RuntimeGOARCH = "s390x"
	assert.Equal(t, "s390x", CurrentArch())
}

func TestCurrentArchUnsupported(t *testing.T) {
	origRuntimeGOARCH := RuntimeGOARCH
	defer func() { RuntimeGOARCH = origRuntimeGOARCH }()
	RuntimeGOARCH = "UKNOWN"
	assert.PanicsWithValue(t, "unsupported architecture", func() { CurrentArch() })
}

func TestPanicOnError(t *testing.T) {
	err := errors.New("Error message")
	assert.PanicsWithValue(t, err, func() { PanicOnError(err) })
}