fix: resolve pylint/pyright issues in new test files
lint / lint (push) Successful in 2m7s
test / unit (pull_request) Successful in 57s
test / integration (pull_request) Successful in 17s
test / coverage (pull_request) Successful in 1m4s

- test_contrib_gitea_client: remove unused Any import, fix _mock_response
  to use return_value instead of lambda (unknown lambda type), narrow
  HTTPError hdrs type, add type annotations to fake_urlopen helpers,
  suppress protected-access for _request tests
- test_bootstrap: annotate **kw as **kw: object, use dict literal,
  unpack server_address via index to avoid tuple type mismatch
- test_main: remove unused MagicMock import
- test_watchdog: guard store.get() result before accessing .status

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
This commit is contained in:
2026-07-01 19:47:31 +00:00
parent 57290da1e8
commit 71699b3ecd
4 changed files with 26 additions and 29 deletions
+9 -9
View File
@@ -35,12 +35,12 @@ def _config(tmp: str) -> Config:
)
def _record(**kw) -> RunRecord:
defaults: dict[str, object] = dict(
owner="o", repo="r", issue_number=1, slug="s1", agent_name="a",
bottle_names=["claude"], backend_name="docker", agent_git_user="bot",
pr_number=5, status="running", last_checkin_at="2026-01-01T00:00:00+00:00",
)
def _record(**kw: object) -> RunRecord:
defaults: dict[str, object] = {
"owner": "o", "repo": "r", "issue_number": 1, "slug": "s1", "agent_name": "a",
"bottle_names": ["claude"], "backend_name": "docker", "agent_git_user": "bot",
"pr_number": 5, "status": "running", "last_checkin_at": "2026-01-01T00:00:00+00:00",
}
defaults.update(kw)
return RunRecord(**defaults) # type: ignore[arg-type]
@@ -169,10 +169,10 @@ class BuildTest(unittest.TestCase):
config = _config(tmp)
with patch.dict(os.environ, {"GITEA_TOKEN": "tok"}):
server, _, _ = build(config)
host, port = server.server_address
addr = server.server_address
server.server_close()
self.assertEqual("127.0.0.1", host)
self.assertGreater(port, 0)
self.assertEqual("127.0.0.1", addr[0])
self.assertGreater(addr[1], 0)
if __name__ == "__main__":
+1 -1
View File
@@ -4,7 +4,7 @@ from __future__ import annotations
import io
import unittest
from unittest.mock import MagicMock, patch
from unittest.mock import patch
from bot_bottle.orchestrator.__main__ import main
from bot_bottle.orchestrator.config import Config
+3 -1
View File
@@ -71,7 +71,9 @@ class WatchdogSweepTest(unittest.TestCase):
self.wd.start()
time.sleep(0.05) # enough for several iterations at 0.01s tick
self.wd.stop()
self.assertEqual(STATUS_FROZEN, self.store.get("o", "r", 5).status)
rec = self.store.get("o", "r", 5)
assert rec is not None
self.assertEqual(STATUS_FROZEN, rec.status)
if __name__ == "__main__":
+13 -18
View File
@@ -17,12 +17,13 @@ def _client() -> GiteaClient:
def _mock_response(body: bytes) -> MagicMock:
resp = MagicMock()
resp.read.return_value = body
resp.__enter__ = lambda s: s
resp.__exit__ = MagicMock(return_value=False)
resp.__enter__.return_value = resp
resp.__exit__.return_value = False
return resp
class GiteaClientTest(unittest.TestCase):
# pylint: disable=protected-access
def setUp(self):
self.client = _client()
@@ -46,10 +47,8 @@ class GiteaClientTest(unittest.TestCase):
self.assertTrue(self.client.is_org_member("myorg", "alice"))
def test_is_org_member_false_on_http_error(self):
with patch(
"urllib.request.urlopen",
side_effect=urllib.error.HTTPError("url", 404, "Not Found", {}, None),
):
err = urllib.error.HTTPError("url", 404, "Not Found", None, None) # type: ignore[arg-type]
with patch("urllib.request.urlopen", side_effect=err):
self.assertFalse(self.client.is_org_member("myorg", "nobody"))
def test_get_issue(self):
@@ -84,34 +83,30 @@ class GiteaClientTest(unittest.TestCase):
mock_open.assert_called_once()
def test_request_builds_correct_url(self):
captured: list[object] = []
import urllib.request as ureq
captured: list[ureq.Request] = []
def fake_urlopen(req, timeout):
def fake_urlopen(req: ureq.Request, timeout: float) -> MagicMock: # pylint: disable=unused-argument
captured.append(req)
return _mock_response(b"{}")
with patch("urllib.request.urlopen", side_effect=fake_urlopen):
self.client.get_issue(5)
import urllib.request as ureq
req = captured[0]
assert isinstance(req, ureq.Request)
self.assertIn("/issues/5", req.full_url)
self.assertIn("/issues/5", captured[0].full_url)
def test_request_sends_auth_header(self):
captured: list[object] = []
import urllib.request as ureq
captured: list[ureq.Request] = []
def fake_urlopen(req, timeout):
def fake_urlopen(req: ureq.Request, timeout: float) -> MagicMock: # pylint: disable=unused-argument
captured.append(req)
return _mock_response(b"{}")
with patch("urllib.request.urlopen", side_effect=fake_urlopen):
self.client.get_issue(1)
import urllib.request as ureq
req = captured[0]
assert isinstance(req, ureq.Request)
self.assertEqual("token tok", req.get_header("Authorization"))
self.assertEqual("token tok", captured[0].get_header("Authorization"))
class GiteaForgeTest(unittest.TestCase):