@@ -254,7 +254,6 @@ def test_node_validate_tls_certificate_no_address_book():
254254@patch ("grpc.secure_channel" )
255255@patch ("grpc.insecure_channel" )
256256def test_node_get_channel_secure (
257- _mock_insecure , mock_secure , mock_node_with_address_book
258257 mock_insecure , mock_secure , mock_node_with_address_book
259258):
260259 """Test channel creation for secure connection."""
@@ -273,25 +272,24 @@ def test_node_get_channel_secure(
273272 channel = node ._get_channel ()
274273
275274 mock_secure .assert_called_once ()
276- _mock_insecure .assert_not_called ()
275+ mock_insecure .assert_not_called ()
277276 assert channel is not None
278277
279278
280279@patch ("grpc.secure_channel" )
281280@patch ("grpc.insecure_channel" )
282281def test_node_get_channel_insecure (
283- _mock_insecure , mock_secure , mock_node_without_address_book
284282 mock_insecure , mock_secure , mock_node_without_address_book
285283):
286284 """Test channel creation for insecure connection."""
287285 node = mock_node_without_address_book
288286
289287 mock_channel = Mock ()
290- _mock_insecure .return_value = mock_channel
288+ mock_insecure .return_value = mock_channel
291289
292290 channel = node ._get_channel ()
293291
294- _mock_insecure .assert_called_once ()
292+ mock_insecure .assert_called_once ()
295293 mock_insecure .return_value = mock_channel
296294
297295 channel = node ._get_channel ()
@@ -304,7 +302,6 @@ def test_node_get_channel_insecure(
304302@patch ("grpc.secure_channel" )
305303@patch ("grpc.insecure_channel" )
306304def test_node_get_channel_reuses_existing (
307- __mock_insecure , mock_secure , mock_node_with_address_book
308305 mock_insecure , mock_secure , mock_node_with_address_book
309306):
310307 """Test that channel is reused when already created."""
0 commit comments