fix missing tokens from es

multiversx · Dec 27, 2024 · 6e5bcf5 · 6e5bcf5
1 parent 54bce4b
commit 6e5bcf5
Show file tree

Hide file tree

Showing 2 changed files with 44 additions and 1 deletion.
diff --git a/outport/process/alteredaccounts/tokensProcessor.go b/outport/process/alteredaccounts/tokensProcessor.go
@@ -137,7 +137,7 @@ func (tp *tokensProcessor) processMultiTransferEvent(event data.EventHandler, ma
 
 		if string(tokenID) == vmcommon.EGLDIdentifier {
 			tp.processNativeEGLDTransferWithMultiTransfer(destinationAddress, markedAlteredAccounts)
-			return
+			continue
 		}
 
 		// process event for the sender address

diff --git a/outport/process/alteredaccounts/tokensProcessor_test.go b/outport/process/alteredaccounts/tokensProcessor_test.go
@@ -97,3 +97,46 @@ func TestTokenProcessorProcessEventMultiTransferV2WithEGLD(t *testing.T) {
 	}
 	require.Equal(t, markedAccount2, markedAccounts["receiver"])
 }
+
+func TestTokenProcessorProcessEventMultiTransferV2WithEGLDAndMoreTokens(t *testing.T) {
+	t.Parallel()
+
+	tp := newTokensProcessor(&mock.ShardCoordinatorStub{})
+
+	markedAccounts := make(map[string]*markedAlteredAccount)
+	tp.processEvent(&transaction.Event{
+		Identifier: []byte(core.BuiltInFunctionMultiESDTNFTTransfer),
+		Address:    []byte("addr"),
+		Topics:     [][]byte{[]byte("token1"), big.NewInt(0).Bytes(), []byte("2"), []byte(vmcommon.EGLDIdentifier), big.NewInt(0).Bytes(), []byte("3"), []byte("token2"), big.NewInt(0).Bytes(), []byte("2"), []byte("receiver")},
+	}, markedAccounts)
+
+	require.Equal(t, 2, len(markedAccounts))
+	markedAccount1 := &markedAlteredAccount{
+		tokens: map[string]*markedAlteredAccountToken{
+			"token1": {
+				identifier: "token1",
+				nonce:      0,
+			},
+			"token2": {
+				identifier: "token2",
+				nonce:      0,
+			},
+		},
+	}
+	require.Equal(t, markedAccount1, markedAccounts["addr"])
+
+	markedAccount2 := &markedAlteredAccount{
+		balanceChanged: true,
+		tokens: map[string]*markedAlteredAccountToken{
+			"token1": {
+				identifier: "token1",
+				nonce:      0,
+			},
+			"token2": {
+				identifier: "token2",
+				nonce:      0,
+			},
+		},
+	}
+	require.Equal(t, markedAccount2, markedAccounts["receiver"])
+}