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 setUp(self):
self._pm = MongoDBPersistenceManager({
"db": {
"url": "localhost",
"port": 27017,
"db_name": "campus_unimi_test"
}
})
ODMModel.set_pm(self._pm)
self._pm.clean_collection("building")
def setUp(self):
self._pm = MongoDBPersistenceManager({
"db": {
"url" : "localhost",
"port" : 27017,
"db_name" : "campus_unimi_test"
}
})
ODMModel.set_pm(self._pm)
self._pm.clean_collection("building")
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 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 run_csv(self, files):
"""
Process csv files.
Arguments:
- files: files we want process.
Returns: None
Instantiates a MongoDBPersistenceManager and a CSVTask to process the csv
files whit the apposite procedure.
"""
persistence = MongoDBPersistenceManager(self._config)
ODMModel.set_pm( persistence )
task = CSVTask(self._config)
task.perform_updates_on_files(files)
self.run_lookup()
def run_svg(self, b_ids):
"""
Create svg files representing maps of buildings.
Arguments:
- b_ids: a list of string representing the building ids of the buildings
we want create/update svg files.
Returns: None
Instantiates a MongoDBPersistenceManager and a SVGTask to create the svg
files whit the apposite procedure.
If b_ids is None the CSVTask create svg files of every building in the DB.
"""
persistence = MongoDBPersistenceManager(self._config)
ODMModel.set_pm(persistence)
task = SVGTask(self._config)
task.perform_svg_update(b_ids)
def run_csv(self, files):
"""
Process csv files.
Arguments:
- files: files we want process.
Returns: None
Instantiates a MongoDBPersistenceManager and a CSVTask to process the csv
files whit the apposite procedure.
"""
persistence = MongoDBPersistenceManager(self._config)
ODMModel.set_pm(persistence)
task = CSVTask(self._config)
task.perform_updates_on_files(files)
self.run_lookup()
def run_svg(self, b_ids):
"""
Create svg files representing maps of buildings.
Arguments:
- b_ids: a list of string representing the building ids of the buildings
we want create/update svg files.
Returns: None
Instantiates a MongoDBPersistenceManager and a SVGTask to create the svg
files whit the apposite procedure.
If b_ids is None the CSVTask create svg files of every building in the DB.
"""
persistence = MongoDBPersistenceManager(self._config)
ODMModel.set_pm( persistence )
task = SVGTask(self._config)
task.perform_svg_update(b_ids)
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 setUp(self):
pm = MongoDBPersistenceManager({
"db": {
"url" : "localhost",
"port" : 27017,
"db_name" : "campus_unimi_test"
}
})
self.old_pm = getattr(ODMModel, "_pm", None)
ODMModel.set_pm(pm)
self.db_building = {
"_id" : "21030",
"dxf" : {
"l_b_id" : "5703",
"floors" : [
{ "f_id" : "-05", "rooms": {}, "unidentified_rooms": [] },
]
},
"edilizia" : {
"l_b_id" : "5703",
"b_id" : "21030",
"lat" : "",
"lon" : "",
"address" : ""
},
"easyroom" : {
"b_id" : "21030",
"address" : "Via Celoria"
}
}
pm.clean_collection("building")
self.edil_up = EdiliziaDataUpdater()
self.easy_up = EasyroomDataUpdater()
self.dxf_updater = DXFDataUpdater()
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)
from persistence import MongoDBPersistenceManager
from model import RoomCategory,Building,BuildingView,AvailableService
from tasks import LookupTableTask
from model.odm import ODMModel
from bson.json_util import dumps
from flask import Flask, jsonify,abort,request,send_from_directory,render_template,Markup
from datetime import datetime
from api.model import RoomTimeTable
app = Flask(__name__,static_url_path='')
app.app_config = ConfigManager('config/general.json')
app.persistence = MongoDBPersistenceManager(app.app_config)
app.api_namespace = '/api'
app.api_version = 'v1.0'
ODMModel.set_pm( app.persistence )
BuildingView.setup_collection()
# radius used with GeoSpatial Query (meters)
app.radius = 2000
app.maps_folder = 'static-maps'
app.lookup_table_folder = 'static-table'
###########
# HELPERS #
###########
def url_for_endpoint(url_endpoint):
return '/'.join( [app.api_namespace,app.api_version,url_endpoint] )+'/'
def filter_buildings_by_service(buildings,service,remove_floor_details=False):
return [ b for b in buildings if len([ f for f in b['floors'] if service in f['available_services'] ])>0 ]
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)
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_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)
from utils.config_manager import ConfigManager
from persistence.db import MongoDBPersistenceManager
from model.odm import ODMModel
from model import Building
from utils.logger import Logger
from analysis import DXFAnalysis
from analysis import FloorMergeAnalysis
from analysis import BuildingIdAnalysis
persistence = MongoDBPersistenceManager(ConfigManager("config/general.json"))
ODMModel.set_pm( persistence )
def data_and_percent(value, total, num_padding = 3,):
if(total > 0):
return ("{:0>"+str(num_padding)+"} ({:>0.1f}%)").format(value, value/total*100.0)
else:
return ("{:0>"+str(num_padding)+"} ").format(value)
class GeneralReport:
merge_tuples = [
("edilizia", "dxf"),
("edilizia", "easyroom"),
("easyroom", "dxf"),
("easyroom", "edilizia")
]
@classmethod
def report_building(klass, building):
BuildingIdAnalysis.analyse_building_id(building)
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_collection_name(self):
self.assertEqual(self.odm.collection_name(), 'odmmodel')
self.assertEqual(ODMModel.collection_name(), 'odmmodel')
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_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_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 __init__(self):
self._config = ConfigManager("config/general.json")
self._data_persistence = MongoDBPersistenceManager(self._config)
ODMModel.set_pm(self._data_persistence)
BuildingView.setup_collection()
def __init__(self):
self._config = ConfigManager("config/general.json")
self._data_persistence = MongoDBPersistenceManager(self._config)
ODMModel.set_pm( self._data_persistence )
BuildingView.setup_collection()
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"], [])
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')
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_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 tearDown(self):
ODMModel.set_pm(self.old_pm)
def setUp(self):
pm = MongoDBPersistenceManager({
"db": {
"url" : "localhost",
"port" : 27017,
"db_name" : "campus_unimi_test"
}
})
ODMModel.set_pm(pm)
self.edil_rooms = [
{
'room_name' : 'Aula Seminari',
'cat_name' : 'Aula',
'r_id' : 'T-065',
'b_id' : '21030',
'capacity' : '52',
'l_floor' : 'T'
},
{
'room_name' : 'Aula Pippo',
'cat_name' : 'Aula',
'r_id' : 'T066',
'b_id' : '21030',
'capacity' : '52',
'l_floor' : 'T'
},
{
'room_name' : 'Aula Pippo Sdentato',
'cat_name' : 'Aula',
'r_id' : '1066',
'b_id' : '21030',
'capacity' : '52',
'l_floor' : '1'
}
]
self.easy_rooms = [
{
'room_name' : 'Aula Seminari',
'cat_name' : 'Aula',
'r_id' : '21030#T-065',
'b_id' : '21030',
'capacity' : '52',
'l_floor' : '0'
},
{
'room_name' : 'Aula Pippo',
'cat_name' : 'Aula',
'r_id' : '21030#T066',
'b_id' : '21030',
'capacity' : '52',
'l_floor' : '0'
},
{
'room_name' : 'Aula Pippo Sdentato',
'cat_name' : 'Aula',
'r_id' : '21030#1066',
'b_id' : '21030',
'capacity' : '52',
'l_floor' : '10'
}
]
pm.clean_collection("building")
self.edil_up = EdiliziaDataUpdater()
self.easy_up = EasyroomDataUpdater()
Building.__str__ = lambda x: ""
