Create a custom exception type for this library

3 files changed, 25 insertions, 2 deletions

diff --git a/tests/test_engine.py b/tests/test_engine.py
index a3d2898..8a7421c 100644
--- a/tests/test_engine.py
+++ b/tests/test_engine.py
@@ -52,6 +52,17 @@ def test_serde_file(tmpdir):
     assert deserialization_result is None
 
 
+def test_deserialize_corrupt(tmpdir):
+    path = str(tmpdir / "corrupt_cache.dat")
+    with open(path, "w", encoding="utf-8") as f:
+        f.write("abc")
+
+    engine = empty_engine()
+    with pytest.raises(adblock.DeserializationError):
+        engine.deserialize_from_file(path)
+    with pytest.raises(adblock.DeserializationError):
+        engine.deserialize(b"abc")
+
 def test_serde():
     engine = empty_engine()
     serialization_result = engine.serialize()
diff --git a/tests/test_exceptions.py b/tests/test_exceptions.py
new file mode 100644
index 0000000..497b3ef
--- /dev/null
+++ b/tests/test_exceptions.py
@@ -0,0 +1,10 @@
+import adblock
+
+def test_correct_baseclasses():
+    assert issubclass(adblock.AdblockException, Exception)
+    assert issubclass(adblock.BlockerException, adblock.AdblockException)
+    assert issubclass(adblock.SerializationError, adblock.BlockerException)
+    assert issubclass(adblock.DeserializationError, adblock.BlockerException)
+    assert issubclass(adblock.OptimizedFilterExistence, adblock.BlockerException)
+    assert issubclass(adblock.BadFilterAddUnsupported, adblock.BlockerException)
+    assert issubclass(adblock.FilterExists, adblock.BlockerException)
\ No newline at end of file
diff --git a/tests/test_imports.py b/tests/test_imports.py
index 7692483..09ef5d5 100644
--- a/tests/test_imports.py
+++ b/tests/test_imports.py
@@ -14,16 +14,18 @@ def get_added_classes():
             match = re.match(r"m\.add_class::<(.+)>\(\)\?;", line.strip())
             if match is not None:
                 classes.append(match.group(1))
+                continue
     return classes
 
 
 def test_added_classes():
     """
     Make sure that there's no class that we added in Rust but didn't import in
-    `__init__.py` and vice versa.
+    `__init__.py`.
     """
     added_classes = get_added_classes()
-    assert added_classes == list(adblock.__all__)
+    for c in added_classes:
+        assert c in adblock.__all__
 
 
 def test_dunder_all_classes_imported():