add concurrency unit tests

trie/branchNode.go

@@ -461,6 +461,7 @@ func (bn *branchNode) updateNode(
 		go func(childPos int) {
 			updateFunc(data[childPos], childPos, goRoutinesManager, modifiedHashes, hasBeenModified, db)
 
+			goRoutinesManager.EndGoRoutineProcessing()
 			waitGroup.Done()
 		}(childPos)
 	}

trie/export_test.go

@@ -2,7 +2,8 @@ package trie
 
 import (
 	"time"
-
+
+	"github.com/multiversx/mx-chain-core-go/core"
 	"github.com/multiversx/mx-chain-core-go/marshal"
 	"github.com/multiversx/mx-chain-go/common"
 	"github.com/multiversx/mx-chain-go/common/statistics"
@@ -105,3 +106,16 @@ func ExecuteUpdatesFromBatch(tr common.Trie) {
 	pmt, _ := tr.(*patriciaMerkleTrie)
 	_ = pmt.updateTrie()
 }
+
+func KeyBytesToHex(str []byte) []byte {
+	return keyBytesToHex(str)
+}
+
+func GetBatchManager(tr common.Trie) common.TrieBatchManager {
+	return tr.(*patriciaMerkleTrie).batchManager
+}
+
+func SetGoRoutinesThrottlerToTrie(tr common.Trie, gt core.Throttler) {
+	pmt, _ := tr.(*patriciaMerkleTrie)
+	pmt.goroutinesThrottler = gt
+}

trie/interface.go

@@ -108,6 +108,7 @@ type IdleNodeProvider interface {
 	IsInterfaceNil() bool
 }
 
+// RootManager is used to manage the root node
 type RootManager interface {
 	GetRootNode() node
 	SetNewRootNode(newRoot node)