class TestDelete(AddressimoTestCase): def setUp(self): self.patcher1 = patch('addressimo.resolvers.RedisResolver.Redis') self.mockRedis = self.patcher1.start() from addressimo.data import IdObject # Setup redis data self.mock_id_obj = IdObject('id') self.mock_id_obj.id = 'id' # Setup redis resolver self.rr = RedisResolver() def test_go_right(self): ret_val = self.rr.delete(self.mock_id_obj) # Validate calls self.assertEqual( self.mockRedis.from_url.return_value.delete.return_value, ret_val) self.assertEqual(3, self.mockRedis.from_url.call_count) self.assertEqual( 1, self.mockRedis.from_url.return_value.delete.call_count) call_args = self.mockRedis.from_url.return_value.delete.call_args[0] self.assertEqual(self.mock_id_obj.id, call_args[0]) def test_exception_saving_redis_data(self): # Setup test case self.mockRedis.from_url.return_value.delete.side_effect = Exception() self.assertRaises(Exception, self.rr.delete, self.mock_id_obj) # Validate calls self.assertEqual(3, self.mockRedis.from_url.call_count) self.assertEqual( 1, self.mockRedis.from_url.return_value.delete.call_count) call_args = self.mockRedis.from_url.return_value.delete.call_args[0] self.assertEqual(self.mock_id_obj.id, call_args[0])
class TestDelete(AddressimoTestCase): def setUp(self): self.patcher1 = patch('addressimo.resolvers.RedisResolver.Redis') self.mockRedis = self.patcher1.start() from addressimo.data import IdObject # Setup redis data self.mock_id_obj = IdObject('id') self.mock_id_obj.id = 'id' # Setup redis resolver self.rr = RedisResolver() def test_go_right(self): ret_val = self.rr.delete(self.mock_id_obj) # Validate calls self.assertEqual(self.mockRedis.from_url.return_value.delete.return_value, ret_val) self.assertEqual(1, self.mockRedis.from_url.call_count) self.assertEqual(1, self.mockRedis.from_url.return_value.delete.call_count) call_args = self.mockRedis.from_url.return_value.delete.call_args[0] self.assertEqual(self.mock_id_obj.id, call_args[0]) def test_exception_saving_redis_data(self): # Setup test case self.mockRedis.from_url.return_value.delete.side_effect = Exception() self.assertRaises(Exception, self.rr.delete, self.mock_id_obj) # Validate calls self.assertEqual(1, self.mockRedis.from_url.call_count) self.assertEqual(1, self.mockRedis.from_url.return_value.delete.call_count) call_args = self.mockRedis.from_url.return_value.delete.call_args[0] self.assertEqual(self.mock_id_obj.id, call_args[0])