Exposing function for test returning only test chainIds

selectors.go

@@ -13,17 +13,17 @@ var selectorsYml []byte
 //go:embed test_selectors.yml
 var testSelectorsYml []byte
 
+var selectorsMap = parseYml(selectorsYml)
+var testSelectorsMap = parseYml(testSelectorsYml)
+
 var evmChainIdToChainSelector = loadAllSelectors()
 
 func loadAllSelectors() map[uint64]uint64 {
-	selectors := parseYml(selectorsYml)
-	testSelectors := parseYml(testSelectorsYml)
-
-	output := make(map[uint64]uint64, len(selectors)+len(testSelectors))
-	for k, v := range selectors {
+	output := make(map[uint64]uint64, len(selectorsMap)+len(testSelectorsMap))
+	for k, v := range selectorsMap {
 		output[k] = v
 	}
-	for k, v := range testSelectors {
+	for k, v := range testSelectorsMap {
 		output[k] = v
 	}
 	return output
@@ -66,3 +66,11 @@ func SelectorFromChainId(chainId uint64) (uint64, error) {
 	}
 	return 0, fmt.Errorf("chain selector not found for chain %d", chainId)
 }
+
+func TestChainIds() []uint64 {
+	chainIds := make([]uint64, 0, len(testSelectorsMap))
+	for k := range testSelectorsMap {
+		chainIds = append(chainIds, k)
+	}
+	return chainIds
+}

selectors_test.go

@@ -1,6 +1,27 @@
 package chain_selectors
 
-import "testing"
+import (
+	"testing"
+)
+
+func TestNoSameChainSelectorsAreGenerated(t *testing.T) {
+	chainSelectors := map[uint64]bool{}
+
+	for k, v := range evmChainIdToChainSelector {
+		if _, exist := chainSelectors[v]; exist {
+			t.Errorf("Chain Selectors should be unique. Selector %d is duplicated for chain %d", v, k)
+		}
+		chainSelectors[v] = true
+	}
+}
+
+func TestNoOverlapBetweenRealAndTestChains(t *testing.T) {
+	for k, _ := range selectorsMap {
+		if _, exist := testSelectorsMap[k]; exist {
+			t.Errorf("Chain %d is duplicated between real and test chains", k)
+		}
+	}
+}
 
 func TestBothSelectorsYmlAndTestSelectorsYmlAreValid(t *testing.T) {
 	optimismGoerliSelector, err := SelectorFromChainId(420)
@@ -74,3 +95,16 @@ func TestSelectorFromChainId(t *testing.T) {
 		t.Error("Should return correct chain selector id")
 	}
 }
+
+func TestTestChainIds(t *testing.T) {
+	chainIds := TestChainIds()
+	if len(chainIds) != len(testSelectorsMap) {
+		t.Error("Should return correct number of test chain ids")
+	}
+
+	for _, chainId := range chainIds {
+		if _, exist := testSelectorsMap[chainId]; !exist {
+			t.Error("Should return correct test chain ids")
+		}
+	}
+}