def test_instance_callbacks_are_executed_once(self):
ODMModel.set_pm(MagicMock())
dic = {"_id" : "2323", "pippo" : "pluto"}
odm = ODMModel(dic)
odm.pippo = MagicMock()
odm.pluto = MagicMock()
sempronio = MagicMock()
odm.listen_once("before_save", "pippo")
odm.listen_once("before_save", "pluto")
odm.listen_once("before_save", sempronio)
odm.save()
MagicMock.assert_called_once_with(odm.pippo, odm)
MagicMock.assert_called_once_with(odm.pluto, odm)
MagicMock.assert_called_once_with(sempronio, odm)
self.assertEqual(odm.before_callbacks_single["save"], [])
odm.pippo.reset_mock()
odm.pluto.reset_mock()
sempronio.reset_mock()
odm.listen_once("after_destroy", "pippo")
odm.listen_once("after_destroy", sempronio)
odm.destroy()
MagicMock.assert_called_once_with(odm.pippo, odm)
MagicMock.assert_called_once_with(sempronio, odm)
self.assertEqual(odm.pluto.call_count, 0)
self.assertEqual(odm.after_callbacks_single["destroy"], [])
def test_instance_callbacks_are_executed_once(self):
ODMModel.set_pm(MagicMock())
dic = {"_id": "2323", "pippo": "pluto"}
odm = ODMModel(dic)
odm.pippo = MagicMock()
odm.pluto = MagicMock()
sempronio = MagicMock()
odm.listen_once("before_save", "pippo")
odm.listen_once("before_save", "pluto")
odm.listen_once("before_save", sempronio)
odm.save()
MagicMock.assert_called_once_with(odm.pippo, odm)
MagicMock.assert_called_once_with(odm.pluto, odm)
MagicMock.assert_called_once_with(sempronio, odm)
self.assertEqual(odm.before_callbacks_single["save"], [])
odm.pippo.reset_mock()
odm.pluto.reset_mock()
sempronio.reset_mock()
odm.listen_once("after_destroy", "pippo")
odm.listen_once("after_destroy", sempronio)
odm.destroy()
MagicMock.assert_called_once_with(odm.pippo, odm)
MagicMock.assert_called_once_with(sempronio, odm)
self.assertEqual(odm.pluto.call_count, 0)
self.assertEqual(odm.after_callbacks_single["destroy"], [])
def test_class_callbacks_are_called(self):
ODMModel.set_pm(MagicMock())
dic = {"_id" : "2323", "pippo" : "pluto"}
odm = ODMModel(dic)
odm.pippo = MagicMock()
odm.pluto = MagicMock()
odm.sempronio = MagicMock()
ODMModel.listen("before_save", "sempronio")
ODMModel.listen("before_save", "pippo")
ODMModel.listen("after_save", "pluto")
ODMModel.listen("after_save", "sempronio")
odm.save()
MagicMock.assert_called_once_with(odm.pippo, odm)
MagicMock.assert_called_once_with(odm.pluto, odm)
self.assertEqual(odm.sempronio.call_count, 2)
def test_class_callbacks_are_called(self):
ODMModel.set_pm(MagicMock())
dic = {"_id": "2323", "pippo": "pluto"}
odm = ODMModel(dic)
odm.pippo = MagicMock()
odm.pluto = MagicMock()
odm.sempronio = MagicMock()
ODMModel.listen("before_save", "sempronio")
ODMModel.listen("before_save", "pippo")
ODMModel.listen("after_save", "pluto")
ODMModel.listen("after_save", "sempronio")
odm.save()
MagicMock.assert_called_once_with(odm.pippo, odm)
MagicMock.assert_called_once_with(odm.pluto, odm)
self.assertEqual(odm.sempronio.call_count, 2)
