class TestGetBranches(AddressimoTestCase):
def setUp(self):
self.patcher1 = patch('addressimo.resolvers.RedisResolver.Redis')
self.mockRedis = self.patcher1.start()
# Setup redis return data
self.mockRedis.from_url.return_value.hkeys.return_value = [
'123', '456'
]
# Setup redis resolver
self.rr = RedisResolver()
def test_go_right(self):
ret_val = self.rr.get_branches(111)
self.assertListEqual([123, 456], ret_val)
self.assertEqual(1,
self.mockRedis.from_url.return_value.hkeys.call_count)
self.assertEqual(
111, self.mockRedis.from_url.return_value.hkeys.call_args[0][0])
def test_no_branches_present(self):
# Setup test case
self.mockRedis.from_url.return_value.hkeys.return_value = None
ret_val = self.rr.get_branches(111)
self.assertListEqual([], ret_val)
self.assertEqual(1,
self.mockRedis.from_url.return_value.hkeys.call_count)
self.assertEqual(
111, self.mockRedis.from_url.return_value.hkeys.call_args[0][0])
def test_exception_retrieving_branches(self):
# Setup test case
self.mockRedis.from_url.return_value.hkeys.side_effect = Exception(
'Lookup failed')
ret_val = self.rr.get_branches(111)
self.assertListEqual([], ret_val)
self.assertEqual(1,
self.mockRedis.from_url.return_value.hkeys.call_count)
self.assertEqual(
111, self.mockRedis.from_url.return_value.hkeys.call_args[0][0])
class TestGetBranches(AddressimoTestCase):
def setUp(self):
self.patcher1 = patch('addressimo.resolvers.RedisResolver.Redis')
self.mockRedis = self.patcher1.start()
# Setup redis return data
self.mockRedis.from_url.return_value.hkeys.return_value = ['123', '456']
# Setup redis resolver
self.rr = RedisResolver()
def test_go_right(self):
ret_val = self.rr.get_branches(111)
self.assertListEqual([123, 456], ret_val)
self.assertEqual(1, self.mockRedis.from_url.return_value.hkeys.call_count)
self.assertEqual(111, self.mockRedis.from_url.return_value.hkeys.call_args[0][0])
def test_no_branches_present(self):
# Setup test case
self.mockRedis.from_url.return_value.hkeys.return_value = None
ret_val = self.rr.get_branches(111)
self.assertListEqual([], ret_val)
self.assertEqual(1, self.mockRedis.from_url.return_value.hkeys.call_count)
self.assertEqual(111, self.mockRedis.from_url.return_value.hkeys.call_args[0][0])
def test_exception_retrieving_branches(self):
# Setup test case
self.mockRedis.from_url.return_value.hkeys.side_effect = Exception('Lookup failed')
ret_val = self.rr.get_branches(111)
self.assertListEqual([], ret_val)
self.assertEqual(1, self.mockRedis.from_url.return_value.hkeys.call_count)
self.assertEqual(111, self.mockRedis.from_url.return_value.hkeys.call_args[0][0])
