Skip to content

Commit 06083b1

Browse files
author
Casey Quinn
committed
test(shared): tighten method error assertions
1 parent d8fcc13 commit 06083b1

File tree

1 file changed

+7
-13
lines changed

1 file changed

+7
-13
lines changed

tests/shared/test_method_errors.py

Lines changed: 7 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@
99

1010

1111
def _ensure(condition: bool, message: str) -> None:
12-
if condition:
12+
if condition: # pragma: no branch
1313
return
1414
pytest.fail(message) # pragma: no cover
1515

@@ -20,9 +20,7 @@ def _assert_error(error: types.JSONRPCError, expected_code: int, expected_messag
2020
_ensure(error_payload.message == expected_message, f"unexpected error message: {error_payload.message}")
2121

2222

23-
async def _run_client_request(
24-
request: types.JSONRPCRequest, *, expected_error: tuple[int, str] | None = None
25-
) -> types.JSONRPCError:
23+
async def _run_client_request(request: types.JSONRPCRequest, *, expected_error: tuple[int, str]) -> types.JSONRPCError:
2624
request_send, request_receive = anyio.create_memory_object_stream[SessionMessage | Exception](1)
2725
response_send, response_receive = anyio.create_memory_object_stream[SessionMessage](1)
2826

@@ -48,16 +46,13 @@ async def _run_client_request(
4846
_ensure(isinstance(root, types.JSONRPCError), "expected a JSON-RPC error response")
4947
error = cast(types.JSONRPCError, root)
5048

51-
if expected_error is not None:
52-
expected_code, expected_message = expected_error
53-
_assert_error(error, expected_code, expected_message)
49+
expected_code, expected_message = expected_error
50+
_assert_error(error, expected_code, expected_message)
5451

5552
return error
5653

5754

58-
async def _run_server_request(
59-
request: types.JSONRPCRequest, *, expected_error: tuple[int, str] | None = None
60-
) -> types.JSONRPCError:
55+
async def _run_server_request(request: types.JSONRPCRequest, *, expected_error: tuple[int, str]) -> types.JSONRPCError:
6156
request_send, request_receive = anyio.create_memory_object_stream[SessionMessage | Exception](1)
6257
response_send, response_receive = anyio.create_memory_object_stream[SessionMessage](1)
6358

@@ -83,9 +78,8 @@ async def _run_server_request(
8378
_ensure(isinstance(root, types.JSONRPCError), "expected a JSON-RPC error response")
8479
error = cast(types.JSONRPCError, root)
8580

86-
if expected_error is not None:
87-
expected_code, expected_message = expected_error
88-
_assert_error(error, expected_code, expected_message)
81+
expected_code, expected_message = expected_error
82+
_assert_error(error, expected_code, expected_message)
8983

9084
return error
9185

0 commit comments

Comments
 (0)