test: ensure err works

ApeWorX · Sep 11, 2024 · 54c04c3 · 54c04c3
1 parent c1efd86
commit 54c04c3
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/tests/functional/test_exceptions.py b/tests/functional/test_exceptions.py
@@ -12,6 +12,7 @@
     ContractNotFoundError,
     NetworkNotFoundError,
     TransactionError,
+    UnknownSnapshotError,
     handle_ape_exception,
 )
 from ape.types import SourceTraceback
@@ -247,3 +248,15 @@ def test_fork_network(self):
             "Try installing an explorer plugin using \x1b[32mape plugins install etherscan"
             "\x1b[0m, or using a network with explorer support."
         )
+
+
+class TestUnknownSnapshotError:
+    def test_bytes(self):
+        snapshot_id = b"asdfasdfasdf"
+        err = UnknownSnapshotError(snapshot_id)
+        assert str(err) == "Unknown snapshot ID '6173..6466'."
+
+    @pytest.mark.parametrize("snapshot_id", (123, "123"))
+    def test_not_bytes(self, snapshot_id):
+        err = UnknownSnapshotError(snapshot_id)
+        assert str(err) == "Unknown snapshot ID '123'."