def test_destroy(self):
pm = MagicMock()
ODMModel.set_pm(pm)
dic = {"_id" : "2323", "pippo" : "pluto"}
odm = ODMModel(dic)
odm.destroy()
pm.destroy_by_id.assert_called_once_with(odm.collection_name(), "2323")
def test_destroy(self):
pm = MagicMock()
ODMModel.set_pm(pm)
dic = {"_id": "2323", "pippo": "pluto"}
odm = ODMModel(dic)
odm.destroy()
pm.destroy_by_id.assert_called_once_with(odm.collection_name(), "2323")
class ODMModelTest(unittest.TestCase):
def setUp(self):
self.odm = ODMModel( { "pippo" : "pluto", "mickey": "paperino" } )
self.old_before_callbacks = ODMModel.before_callbacks
self.old_after_callbacks = ODMModel.after_callbacks
ODMModel.before_callbacks = {}
ODMModel.after_callbacks = {}
def tearDown(self):
ODMModel.before_callbacks = self.old_before_callbacks
ODMModel.after_callbacks = self.old_after_callbacks
def test_collection_name(self):
self.assertEqual(self.odm.collection_name(), 'odmmodel')
self.assertEqual(ODMModel.collection_name(), 'odmmodel')
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_destroy(self):
pm = MagicMock()
ODMModel.set_pm(pm)
dic = {"_id" : "2323", "pippo" : "pluto"}
odm = ODMModel(dic)
odm.destroy()
pm.destroy_by_id.assert_called_once_with(odm.collection_name(), "2323")
def test_class_callbacks_are_saved(self):
ODMModel.listen("before_save", "pippo")
ODMModel.listen("before_save", "pluto")
ODMModel.listen("before_save", "sempronio")
self.assertTrue("pippo" in ODMModel.before_callbacks["ODMModel"]["save"])
self.assertTrue("pluto" in ODMModel.before_callbacks["ODMModel"]["save"])
self.assertTrue("sempronio" in ODMModel.before_callbacks["ODMModel"]["save"])
self.assertEqual(["pippo", "pluto", "sempronio"], ODMModel.before_callbacks["ODMModel"]["save"] )
ODMModel.listen("after_save", "pippo")
ODMModel.listen("after_save","pluto")
ODMModel.listen("after_save", "sempronio")
self.assertTrue("pippo" in ODMModel.after_callbacks["ODMModel"]["save"])
self.assertTrue("pluto" in ODMModel.after_callbacks["ODMModel"]["save"])
self.assertTrue("sempronio" in ODMModel.after_callbacks["ODMModel"]["save"])
self.assertEqual(["pippo", "pluto", "sempronio"], ODMModel.after_callbacks["ODMModel"]["save"] )
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_destroy_callbacks_are_called(self):
ODMModel.set_pm(MagicMock())
dic = {"_id" : "2323", "pippo" : "pluto"}
odm = ODMModel(dic)
pippo = MagicMock()
pluto = MagicMock()
sempronio = MagicMock()
ODMModel.listen("before_destroy", pippo)
ODMModel.listen("before_destroy", sempronio)
ODMModel.listen("after_destroy", sempronio)
ODMModel.listen("after_destroy", pluto)
odm.destroy()
MagicMock.assert_called_once_with(pippo, odm)
MagicMock.assert_called_once_with(pluto, odm)
MagicMock.assert_any_call(sempronio, odm)
def test_instance_callbacks_are_saved(self):
ODMModel.set_pm(MagicMock())
dic = {"_id" : "2323", "pippo" : "pluto"}
odm = ODMModel(dic)
pippo = MagicMock()
pluto = MagicMock()
sempronio = MagicMock()
odm.listen_once("before_save", "pippo")
odm.listen_once("before_save", "pluto")
odm.listen_once("before_save", "sempronio")
self.assertEqual(
odm.before_callbacks_single["save"],
["pippo", "pluto", "sempronio"]
)
odm.listen_once("after_save", "pippo")
odm.listen_once("after_save", "sempronio")
self.assertEqual(
odm.after_callbacks_single["save"],
["pippo", "sempronio"]
)
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_collection_name(self):
self.assertEqual(self.odm.collection_name(), 'odmmodel')
self.assertEqual(ODMModel.collection_name(), 'odmmodel')
class ODMModelTest(unittest.TestCase):
def setUp(self):
self.odm = ODMModel({"pippo": "pluto", "mickey": "paperino"})
self.old_before_callbacks = ODMModel.before_callbacks
self.old_after_callbacks = ODMModel.after_callbacks
ODMModel.before_callbacks = {}
ODMModel.after_callbacks = {}
def tearDown(self):
ODMModel.before_callbacks = self.old_before_callbacks
ODMModel.after_callbacks = self.old_after_callbacks
def test_collection_name(self):
self.assertEqual(self.odm.collection_name(), 'odmmodel')
self.assertEqual(ODMModel.collection_name(), 'odmmodel')
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_destroy(self):
pm = MagicMock()
ODMModel.set_pm(pm)
dic = {"_id": "2323", "pippo": "pluto"}
odm = ODMModel(dic)
odm.destroy()
pm.destroy_by_id.assert_called_once_with(odm.collection_name(), "2323")
def test_class_callbacks_are_saved(self):
ODMModel.listen("before_save", "pippo")
ODMModel.listen("before_save", "pluto")
ODMModel.listen("before_save", "sempronio")
self.assertTrue(
"pippo" in ODMModel.before_callbacks["ODMModel"]["save"])
self.assertTrue(
"pluto" in ODMModel.before_callbacks["ODMModel"]["save"])
self.assertTrue(
"sempronio" in ODMModel.before_callbacks["ODMModel"]["save"])
self.assertEqual(["pippo", "pluto", "sempronio"],
ODMModel.before_callbacks["ODMModel"]["save"])
ODMModel.listen("after_save", "pippo")
ODMModel.listen("after_save", "pluto")
ODMModel.listen("after_save", "sempronio")
self.assertTrue(
"pippo" in ODMModel.after_callbacks["ODMModel"]["save"])
self.assertTrue(
"pluto" in ODMModel.after_callbacks["ODMModel"]["save"])
self.assertTrue(
"sempronio" in ODMModel.after_callbacks["ODMModel"]["save"])
self.assertEqual(["pippo", "pluto", "sempronio"],
ODMModel.after_callbacks["ODMModel"]["save"])
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_destroy_callbacks_are_called(self):
ODMModel.set_pm(MagicMock())
dic = {"_id": "2323", "pippo": "pluto"}
odm = ODMModel(dic)
pippo = MagicMock()
pluto = MagicMock()
sempronio = MagicMock()
ODMModel.listen("before_destroy", pippo)
ODMModel.listen("before_destroy", sempronio)
ODMModel.listen("after_destroy", sempronio)
ODMModel.listen("after_destroy", pluto)
odm.destroy()
MagicMock.assert_called_once_with(pippo, odm)
MagicMock.assert_called_once_with(pluto, odm)
MagicMock.assert_any_call(sempronio, odm)
def test_instance_callbacks_are_saved(self):
ODMModel.set_pm(MagicMock())
dic = {"_id": "2323", "pippo": "pluto"}
odm = ODMModel(dic)
pippo = MagicMock()
pluto = MagicMock()
sempronio = MagicMock()
odm.listen_once("before_save", "pippo")
odm.listen_once("before_save", "pluto")
odm.listen_once("before_save", "sempronio")
self.assertEqual(odm.before_callbacks_single["save"],
["pippo", "pluto", "sempronio"])
odm.listen_once("after_save", "pippo")
odm.listen_once("after_save", "sempronio")
self.assertEqual(odm.after_callbacks_single["save"],
["pippo", "sempronio"])
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_collection_name(self):
self.assertEqual(self.odm.collection_name(), 'odmmodel')
self.assertEqual(ODMModel.collection_name(), 'odmmodel')
