package service import ( "testing" "github.com/stretchr/testify/assert" ) func resetAssetAdapterRegistryForTest(t interface{ Cleanup(func()) }) { oldRegistry := assetAdapters assetAdapters = map[int]AssetAdapter{} t.Cleanup(func() { assetAdapters = oldRegistry }) } func RegisterAssetAdapterForTest(channelType int, adapter AssetAdapter) { assetAdapters[channelType] = adapter } func TestRegisteredAssetChannelTypesAreSorted(t *testing.T) { resetAssetAdapterRegistryForTest(t) RegisterAssetAdapterForTest(61, fakeAssetAdapter{}) RegisterAssetAdapterForTest(7, fakeAssetAdapter{}) RegisterAssetAdapterForTest(16, fakeAssetAdapter{}) assert.Equal(t, []int{7, 16, 61}, RegisteredAssetChannelTypes()) }