1 package utils 2 3 import ( 4 "testing" 5 6 "github.com/stretchr/testify/assert" 7 8 "edge-infra.dev/pkg/edge/constants/api/banner" 9 ) 10 11 func TestGetBannerTypeOrg(t *testing.T) { 12 isOrgBanner := true 13 bannertype := GetBannerType(&isOrgBanner) 14 assert.Equal(t, bannertype, banner.Org) 15 } 16 17 func TestGetBannerTypeEU(t *testing.T) { 18 isOrgBanner := false 19 bannertype := GetBannerType(&isOrgBanner) 20 assert.Equal(t, bannertype, banner.EU) 21 } 22