Correct return type for the serialize method

1 files changed, 29 insertions, 0 deletions

diff --git a/tests/test_engine.py b/tests/test_engine.py
index af15b6d..28e25e6 100644
--- a/tests/test_engine.py
+++ b/tests/test_engine.py
@@ -1,4 +1,5 @@
 import adblock
+import pytest
 
 
 def test_engine_arguments():
@@ -9,3 +10,31 @@ def test_engine_arguments():
     adblock.Engine(network_filters=[])
     adblock.Engine(load_network=False, load_cosmetic=True, debug=False)
     adblock.Engine(debug=True)
+
+
+def test_serde_file(tmpdir):
+    path = str(tmpdir / "cache.dat")
+
+    engine0 = adblock.Engine()
+    with pytest.raises(FileNotFoundError):
+        # We haven't created the cache.dat file, so we should get an exception
+        # when attempting to deserialize.
+        engine0.deserialize_from_file(path)
+
+    engine1 = adblock.Engine()
+    serialization_result = engine1.serialize_to_file(path)
+    assert serialization_result is None
+
+    engine2 = adblock.Engine()
+    deserialization_result = engine2.deserialize_from_file(path)
+    assert deserialization_result is None
+
+
+def test_serde():
+    engine = adblock.Engine()
+    serialization_result = engine.serialize()
+    assert isinstance(serialization_result, bytes)
+
+    engine2 = adblock.Engine()
+    deserialization_result = engine2.deserialize(serialization_result)
+    assert deserialization_result is None