chore: fixing test

algorandfoundation · Oct 3, 2023 · 9e6cf5b · 9e6cf5b
1 parent 8f5cc72
commit 9e6cf5b
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 5 deletions.
diff --git a/poetry.lock b/poetry.lock
diff --git a/tests/tasks/test_vanity_address.py b/tests/tasks/test_vanity_address.py
@@ -101,8 +101,10 @@ def test_vanity_address_on_existing_alias(mock_keyring: dict[str, str]) -> None:
 
 def test_vanity_address_on_termination(mocker: MockerFixture) -> None:
     mock_pool = mocker.MagicMock()
-    mock_pool.apply_async.side_effect = KeyboardInterrupt()
-    mocker.patch("multiprocessing.Pool", return_value=mock_pool)
+    mock_pool.side_effect = KeyboardInterrupt()
+    # Mocking functions within the thread is tricky hence the use of side_effect on the range is used to simulate
+    # the KeyboardInterrupt
+    mocker.patch("algokit.core.tasks.vanity_address.range", side_effect=[range(2), KeyboardInterrupt()])
 
     result = invoke("task vanity-address AAAAAA")