Fix in equipLeafWithModExtensionNode; go fmt

Author: miha-stopar

geth-utils/gethutil/mpt/witness/branch.go

@@ -174,7 +174,7 @@ func getDriftedPosition(leafKeyRow []byte, numberOfNibbles int) byte {
 		} else {
 			keyLen = 1
 			nibbles = append(nibbles, leafKeyRow[1]-16)
-		}	
+		}
 	} else {
 		keyLen := int(leafKeyRow[2] - 128)
 		if (leafKeyRow[3] != 32) && (leafKeyRow[3] != 0) { // second term is for extension node

geth-utils/gethutil/mpt/witness/gen_witness_from_local_blockchain_test.go

@@ -933,7 +933,7 @@ func TestExtensionIntoBranch(t *testing.T) {
 	database := state.NewDatabase(blockHeaderParent)
 	statedb, _ := state.New(blockHeaderParent.Root, database, nil)
 	addr := common.HexToAddress("0x50efbf12580138bc623c95757286df4e24eb81c9")
-	
+
 	statedb.DisableLoadingRemoteAccounts()
 
 	statedb.CreateAccount(addr)

geth-utils/gethutil/mpt/witness/modified_extension_node.go

@@ -51,17 +51,6 @@ func equipLeafWithModExtensionNode(statedb *state.StateDB, leafNode Node, addr c
 		longExtNodeKey[j] = longNibbles[j-byte(keyIndex)]
 	}
 
-	k := trie.HexToKeybytes(longExtNodeKey)
-	ky := common.BytesToHash(k)
-	var proof [][]byte
-	var err error
-	if isAccountProof {
-		proof, _, _, _, _, err = statedb.GetProof(addr)
-	} else {
-		proof, _, _, _, _, err = statedb.GetStorageProof(addr, ky)
-	}
-	check(err)
-
 	// There is no short extension node when `len(longNibbles) - numberOfNibbles = 1`, in this case there
 	// is simply a branch instead.
 	shortExtNodeIsBranch := len(longNibbles)-numberOfNibbles == 1
@@ -71,6 +60,17 @@ func equipLeafWithModExtensionNode(statedb *state.StateDB, leafNode Node, addr c
 	var extValuesC [][]byte
 
 	if !shortExtNodeIsBranch {
+		k := trie.HexToKeybytes(longExtNodeKey)
+		ky := common.BytesToHash(k)
+		var proof [][]byte
+		var err error
+		if isAccountProof {
+			proof, _, _, _, _, err = statedb.GetProof(addr)
+		} else {
+			proof, _, _, _, _, err = statedb.GetStorageProof(addr, ky)
+		}
+		check(err)
+
 		if len2 > len1 {
 			isItBranch := isBranch(proof[len(proof)-1])