Update core/trie/snap_support.go

Co-authored-by: Pawel Nowosielski <pnowosie@users.noreply.github.com>

core/trie/snap_support.go

@@ -14,7 +14,7 @@
 		return consumer(key, value)
 	})
 	if err != nil {
 		return nil, false, err
 	}
 
 	proofset := map[felt.Felt]ProofNode{}
@@ -27,7 +27,7 @@
 		// actually check that the server did not skip leafs.
 		leftProof, err := GetProof(&startKey, t)
 		if err != nil {
 			return nil, false, err
 		}
 		for _, proof := range leftProof {
 			// Well.. using the trie hash here is kinda slow... but I just need it to work right now.
@@ -40,7 +40,7 @@
 		lastKey := NewKey(t.height, feltBts[:])
 		rightProof, err := GetProof(&lastKey, t)
 		if err != nil {
 			return nil, false, err
 		}
 
 		for _, proof := range rightProof {
@@ -71,26 +71,24 @@
 
 		tempTrie, err := newTrie(newMemStorage(), treeHeight, hash)
 		if err != nil {
 			return false, false, err
 		}
 
 		for i, key := range keys {
 			_, err = tempTrie.Put(key, values[i])
 			if err != nil {
 				return false, false, err
 			}
 		}
 
 		recalculatedRoot, err := tempTrie.Root()
 		if err != nil {
 			return false, false, err
 		}
 
-		if !root.Equal(recalculatedRoot) {
-			return false, false, nil
-		}
+		rootMatched := root.Equal(recalculatedRoot)
 
-		return false, true, nil
+		return false, rootMatched, nil
 	}
 
 	if _, ok := proofMap[*root]; !ok {
@@ -101,7 +99,7 @@
 	proofKeys := map[felt.Felt]Key{}
 	err := buildKeys(NewKey(0, []byte{}), root, proofMap, proofKeys, 0)
 	if err != nil {
 		return false, false, err
 	}
 
 	// TODO: Verify here proof here
@@ -138,7 +136,7 @@
 		ch := proofNode.Edge.Child
 		err := buildKeys(chKey, ch, proofMap, keys, depth+1)
 		if err != nil {
 			return err
 		}
 	} else {
 		binary := proofNode.Binary
@@ -147,14 +145,14 @@
 		ch := binary.LeftHash
 		err := buildKeys(chKey, ch, proofMap, keys, depth+1)
 		if err != nil {
 			return err
 		}
 
 		chKey = currentKey.AppendBit(true)
 		ch = binary.RightHash
 		err = buildKeys(chKey, ch, proofMap, keys, depth+1)
 		if err != nil {
 			return err
 		}
 	}