Add == in MerkleNode

software-mansion · Oct 28, 2024 · 3d0393b · 3d0393b
1 parent 009f27f
commit 3d0393b
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/Sources/Starknet/Data/StorageProof/MerkleNode.swift b/Sources/Starknet/Data/StorageProof/MerkleNode.swift
@@ -37,6 +37,17 @@ public enum MerkleNode: Codable, Equatable {
             try container.encode(edgeNode)
         }
     }
+
+    public static func == (lhs: MerkleNode, rhs: MerkleNode) -> Bool {
+        switch (lhs, rhs) {
+        case let (.binaryNode(lhsBinaryNode), .binaryNode(rhsBinaryNode)):
+            lhsBinaryNode == rhsBinaryNode
+        case let (.edgeNode(lhsEdgeNode), .edgeNode(rhsEdgeNode)):
+            lhsEdgeNode == rhsEdgeNode
+        default:
+            false
+        }
+    }
 }
 
 public struct BinaryNode: Codable, Equatable {