From b98ec2ebf9a5afdf20e29013c99188dfc8ee6d5f Mon Sep 17 00:00:00 2001
From: bogdan-rosianu <bogdan.rosianu@yahoo.com>
Date: Thu, 19 May 2022 14:26:55 +0300
Subject: [PATCH] fix token type log

---
 vm/systemSmartContracts/esdt.go      |  3 ++-
 vm/systemSmartContracts/esdt_test.go | 13 ++++++++++---
 2 files changed, 12 insertions(+), 4 deletions(-)

diff --git a/vm/systemSmartContracts/esdt.go b/vm/systemSmartContracts/esdt.go
index ffba114d3b4..06aa2a9e4e2 100644
--- a/vm/systemSmartContracts/esdt.go
+++ b/vm/systemSmartContracts/esdt.go
@@ -531,7 +531,7 @@ func (e *esdt) registerAndSetRoles(args *vmcommon.ContractCallInput) vmcommon.Re
 	logEntry := &vmcommon.LogEntry{
 		Identifier: []byte(args.Function),
 		Address:    args.CallerAddr,
-		Topics:     [][]byte{tokenIdentifier, args.Arguments[0], args.Arguments[1], []byte(metaESDT)},
+		Topics:     [][]byte{tokenIdentifier, args.Arguments[0], args.Arguments[1], tokenType},
 	}
 	e.eei.Finish(tokenIdentifier)
 	e.eei.AddLogEntry(logEntry)
@@ -553,6 +553,7 @@ func getAllRolesForTokenType(tokenType string) ([][]byte, error) {
 }
 
 func getTokenType(compressed []byte) (bool, []byte, error) {
+	// TODO: might extract the compressed constants to core, alongside metaESDT
 	switch string(compressed) {
 	case "NFT":
 		return false, []byte(core.NonFungibleESDT), nil
diff --git a/vm/systemSmartContracts/esdt_test.go b/vm/systemSmartContracts/esdt_test.go
index 551278d747a..5a5e05230bd 100644
--- a/vm/systemSmartContracts/esdt_test.go
+++ b/vm/systemSmartContracts/esdt_test.go
@@ -4592,9 +4592,16 @@ func TestEsdt_ExecuteRegisterAndSetSemiFungible(t *testing.T) {
 	assert.Equal(t, token.TokenType, []byte(core.SemiFungibleESDT))
 }
 
-func TestEsdt_ExecuteRegisterAndSetMetaESDT(t *testing.T) {
+func TestEsdt_ExecuteRegisterAndSetMetaESDTShouldSetType(t *testing.T) {
 	t.Parallel()
 
+	registerAndSetAllRolesWithTypeCheck(t, []byte("NFT"), []byte(core.NonFungibleESDT))
+	registerAndSetAllRolesWithTypeCheck(t, []byte("SFT"), []byte(core.SemiFungibleESDT))
+	registerAndSetAllRolesWithTypeCheck(t, []byte("META"), []byte(metaESDT))
+	registerAndSetAllRolesWithTypeCheck(t, []byte("FNG"), []byte(core.FungibleESDT))
+}
+
+func registerAndSetAllRolesWithTypeCheck(t *testing.T, typeArgument []byte, expectedType []byte) {
 	args := createMockArgumentsForESDT()
 	eei, _ := NewVMContext(
 		&mock.BlockChainHookStub{},
@@ -4608,7 +4615,7 @@ func TestEsdt_ExecuteRegisterAndSetMetaESDT(t *testing.T) {
 	vmInput := getDefaultVmInputForFunc("registerAndSetAllRoles", nil)
 	vmInput.CallValue = big.NewInt(0).Set(e.baseIssuingCost)
 
-	vmInput.Arguments = [][]byte{[]byte("tokenName"), []byte("TICKER"), []byte("META"), big.NewInt(10).Bytes()}
+	vmInput.Arguments = [][]byte{[]byte("tokenName"), []byte("TICKER"), typeArgument, big.NewInt(10).Bytes()}
 	eei.gasRemaining = 9999
 	eei.returnMessage = ""
 	output := e.Execute(vmInput)
@@ -4617,5 +4624,5 @@ func TestEsdt_ExecuteRegisterAndSetMetaESDT(t *testing.T) {
 	assert.True(t, strings.Contains(string(eei.output[0]), "TICKER-"))
 
 	token, _ := e.getExistingToken(eei.output[0])
-	assert.Equal(t, token.TokenType, []byte(metaESDT))
+	assert.Equal(t, expectedType, token.TokenType)
 }