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_clean(self):
pm = MongoDBPersistenceManager({
"db": {
"url" : "localhost",
"port" : 27017,
"db_name" : "campus_unimi_test"
}
})
ODMModel.set_pm(pm)
dic = {"_id" : "2323", "pippo" : "pluto"}
odm = ODMModel(dic)
odm.save()
self.assertEqual(ODMModel._pm.get_collection("odmmodel").find_one({"_id" : "2323"}), dic)
odm.clean()
self.assertEqual(ODMModel._pm.get_collection("odmmodel").find_one({"_id" : "2323"}), None)
def test_clean(self):
pm = MongoDBPersistenceManager({
"db": {
"url": "localhost",
"port": 27017,
"db_name": "campus_unimi_test"
}
})
ODMModel.set_pm(pm)
dic = {"_id": "2323", "pippo": "pluto"}
odm = ODMModel(dic)
odm.save()
self.assertEqual(
ODMModel._pm.get_collection("odmmodel").find_one({"_id": "2323"}),
dic)
odm.clean()
self.assertEqual(
ODMModel._pm.get_collection("odmmodel").find_one({"_id": "2323"}),
None)
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)
