fixup! Add support for retries

kraken-tech · Sep 19, 2024 · 9c0784b · 9c0784b
1 parent 2365f1e
commit 9c0784b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/django_pg_migration_tools/management/commands/test_migrate_with_timeouts.py b/tests/django_pg_migration_tools/management/commands/test_migrate_with_timeouts.py
@@ -163,7 +163,7 @@ def test_retry_callback_is_called(self, mock_handle, mock_sleep):
                 "migrate_with_timeouts",
                 lock_timeout_in_ms=50_000,
                 lock_timeout_max_retries=2,
-                retry_callback_path="tests.django_pg_migration_tools.management.commands.test_migrate_with_timeouts.example_callback",
+                retry_callback_path=f"{__name__}.example_callback",
             )