def test_execute_propagates_connection_error_on_connect(self, mock_session, mock_try_connect):
mock_try_connect.side_effect = NailgunClient.NailgunConnectionError(
"127.0.0.1:31337", 31337, -31336, Exception("oops"),
)
with self.assertRaises(NailgunClient.NailgunConnectionError):
self.nailgun_client.execute("test")
def test_execute_propagates_connection_error_on_connect(
self, mock_session, mock_try_connect):
mock_try_connect.side_effect = NailgunClient.NailgunConnectionError(
'oops')
with self.assertRaises(NailgunClient.NailgunConnectionError):
self.nailgun_client.execute('test')
