new-api/service/asset_adapter_test.go

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())
}