def test_single_host(self):
connector = MagicMock(RabbitConnection)
hosts = ['one']
conn = connect_to_hosts(connector, hosts, arg='arg')
connector.assert_called_once_with(host='one', arg='arg')
self.assertEqual(conn, connector())
def test_first_host_fails(self):
def fail_on_one(**kwargs):
if kwargs.get('host') == 'one':
raise socket.error('Expected error during test')
connector = MagicMock(RabbitConnection, side_effect=fail_on_one)
hosts = ['one', 'two']
conn = connect_to_hosts(connector, hosts, arg='arg')
connector.assert_called_with(host='two', arg='arg')
self.assertEqual(conn, connector())