tests/unit/test_backend_freezer.py

@@ -213,7 +213,8 @@ class TestFirecrackerFreezer(_FakeHomeMixin, unittest.TestCase):
         freezer = FirecrackerFreezer()
         agent = _make_agent(slug, "firecracker")
 
-        with patch("bot_bottle.backend.firecracker.freezer._commit_rootfs_via_ssh") as mock_commit, \
+        commit_fn = "bot_bottle.backend.firecracker.freezer._commit_rootfs_via_ssh"
+        with patch(commit_fn) as mock_commit, \
              patch("bot_bottle.backend.freeze.info"), \
              patch("bot_bottle.backend.firecracker.freezer.info"):
             freezer.commit(agent)

tests/unit/test_firecracker_helpers.py

@@ -11,6 +11,7 @@ import subprocess
 import tempfile
 import unittest
 from pathlib import Path
+from typing import Any
 from unittest.mock import patch
 
 from bot_bottle.backend.firecracker import firecracker_vm, freezer, netpool, util
@@ -189,7 +190,7 @@ class TestBuildCommittedRootfsDir(unittest.TestCase):
             real_run = subprocess.run
             calls = {"n": 0}
 
-            def counting_run(argv, *a, **k):
+            def counting_run(argv: list[str], *a: Any, **k: Any) -> Any:
                 if argv and argv[0] == "tar":
                     calls["n"] += 1
                 return real_run(argv, *a, **k)