✅ Added additional unit test for removing keys from proof

LiskArchive · Sep 12, 2023 · 1dae25a · 1dae25a
1 parent 5489d7b
commit 1dae25a
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/test/sparse_merkle_tree.spec.js b/test/sparse_merkle_tree.spec.js
@@ -154,6 +154,16 @@ describe('SparseMerkleTree', () => {
             await expect(smt.removeKeysFromProof(proof, removeKeys)).rejects.toThrow("Invalid input: `Not all sibling hashes were used`");
         });
 
+		it('should reject by empty queries in proof', async () => {
+			const smt = new SparseMerkleTree(32);
+			const proof = {
+				siblingHashes: [getRandomBytes()],
+				queries: [],
+			};
+			const removeKeys = FixturesInclusionProof.testCases[0].input.queryKeys.map(keyHex => Buffer.from(keyHex, 'hex'));
+			await expect(smt.removeKeysFromProof(proof, removeKeys)).rejects.toThrow("Invalid input: `Empty`");
+		});
+
 		describe('when there is too many sibling hashes', () => {
 			for (const test of FixturesInclusionProof.testCases) {
 				// eslint-disable-next-line no-loop-func