def test_update_admin_raise_exception(self, generate_update_items, validate_updated_args): self.assertEqual(1, 1) db_adapter = Mock() args = {'test': 'test'} db_adapter.update_object.side_effect = [Exception] am = AdminManager(db_adapter) self.assertEqual(am.update_admin(args), internal_server_error(ANY)) generate_update_items.assert_called_once_with(args) validate_updated_args.assert_called_once_with(args)
def test_update_admin_common_logic(self, validate_updated_args): db_adapter = Mock() args = {'role_type': 1, 'email': 'test'} db_adapter.update_object.return_value = 'passed' am = AdminManager(db_adapter) with mock.patch('hackathon.admin.admin_mgr.get_now') as get_now: get_now.return_value = "now" self.assertEqual(am.update_admin(args), ok('update hackathon admin success')) validate_updated_args.assert_called_once_with(args) db_adapter.update_object.assert_called_once_with(ANY, role_type=1, update_time='now')