Add tests for custom blake hash

Signed-off-by: Lukas Puehringer <lukas.puehringer@nyu.edu>

Author: Lukas Puehringer

tests/test_api.py

@@ -895,6 +895,12 @@ def test_length_and_hash_validation(self) -> None:
             # test with data as bytes
             snapshot_metafile.verify_length_and_hashes(data)
 
+            # test with custom blake algorithm
+            snapshot_metafile.hashes = {
+                "blake2b-256": "963a3c31aad8e2a91cfc603fdba12555e48dd0312674ac48cce2c19c243236a1"
+            }
+            snapshot_metafile.verify_length_and_hashes(data)
+
             # test exceptions
             expected_length = snapshot_metafile.length
             snapshot_metafile.length = 2345
@@ -987,6 +993,12 @@ def test_targetfile_from_data(self) -> None:
         targetfile_from_data = TargetFile.from_data(target_file_path, data)
         targetfile_from_data.verify_length_and_hashes(data)
 
+        # Test with custom blake hash algorithm
+        targetfile_from_data = TargetFile.from_data(
+            target_file_path, data, ["blake2b-256"]
+        )
+        targetfile_from_data.verify_length_and_hashes(data)
+
     def test_metafile_from_data(self) -> None:
         data = b"Inline test content"
 
@@ -1010,6 +1022,10 @@ def test_metafile_from_data(self) -> None:
             ),
         )
 
+        # Test with custom blake hash algorithm
+        metafile = MetaFile.from_data(1, data, ["blake2b-256"])
+        metafile.verify_length_and_hashes(data)
+
     def test_targetfile_get_prefixed_paths(self) -> None:
         target = TargetFile(100, {"sha256": "abc", "md5": "def"}, "a/b/f.ext")
         self.assertEqual(