def test__connect_slave_ok_on_first_try(self): pool = pyredis.pool.SentinelPool(sentinels=[("host1", 12345)], name="mymaster", slave_ok=True) pool._get_master = Mock() client_mock = Mock() pool._get_slave = Mock() pool._get_slave.return_value = client_mock client = pool._connect() pool._get_slave.assert_called_with() self.assertEqual(client, client_mock)
def test__connect_slave_ok_on_second_try(self): pool = pyredis.pool.SentinelPool(sentinels=[("host1", 12345)], name="mymaster", slave_ok=True) pool._get_master = Mock() client_mock = Mock() pool._get_slave = Mock() pool._get_slave.side_effect = [None, client_mock] client = pool._connect() pool._get_slave.assert_has_calls([call(), call()]) self.assertEqual(client, client_mock)
def test__connect_slave_ok_on_first_try(self): pool = pyredis.pool.SentinelPool(sentinels=[('host1', 12345)], name='mymaster', slave_ok=True) pool._get_master = Mock() client_mock = Mock() pool._get_slave = Mock() pool._get_slave.return_value = client_mock client = pool._connect() pool._get_slave.assert_called_with() self.assertEqual(client, client_mock)
def test__connect_master_ok_on_second_try(self): pool = pyredis.pool.SentinelPool(sentinels=[('host1', 12345)], name='mymaster') pool._get_master = Mock() client_mock = Mock() pool._get_master.side_effect = [None, client_mock] pool._get_slave = Mock() client = pool._connect() pool._get_master.assert_has_calls([call(), call()]) self.assertEqual(client, client_mock)