def test_delete_month(self):
# Проверка , что лишнее не удаляется
PlanCtrl.delete_month(3000, 12)
plan = Plan.get(1)
self.assertEqual(plan.n, 1)
# А вот теперь должно удалиться
PlanCtrl.delete_month(2017, 12)
plans = PlanCtrl.find({})
self.assertEqual(len(plans), 0)
def delete(self):
"""
Удаление планов на год/месяц
"""
print(self.context.year, self.context.month)
try:
PlanCtrl.delete_month(self.context.year, self.context.month)
except Exception as e:
error_message = 'Error delete year={} month={}. Error: {}'.format(
self.context.year, self.context.month, e)
self.log.warning(error_message)
return MessageJSON(error_message)
return {'status': 'success'}