package utils

import (
	"testing"

	"github.com/stretchr/testify/assert"

	"edge-infra.dev/pkg/edge/constants/api/banner"
)

func TestGetBannerTypeOrg(t *testing.T) {
	isOrgBanner := true
	bannertype := GetBannerType(&isOrgBanner)
	assert.Equal(t, bannertype, banner.Org)
}

func TestGetBannerTypeEU(t *testing.T) {
	isOrgBanner := false
	bannertype := GetBannerType(&isOrgBanner)
	assert.Equal(t, bannertype, banner.EU)
}