def test_office_load_state(self):
path = "db/"
file_name = "mydb"
file_ = file_name + ".db"
#clean up to avoid conflict between tests
os.remove(path + file_)
self.clear_stores()
#memory
office1 = Office('ooskks9')
Office.add(office1)
Office.save_state(file_name)
#db
engine = create_engine("sqlite:///" + path + file_, echo=False)
Session = sessionmaker(bind=engine)
session = Session()
db_office = session.query(Room).filter_by(
roomname='ooskks9'.upper()).first()
#clear memory stores
self.clear_stores()
#memory
Office.load_state(file_name)
office = Office.from_name('ooskks9')
#compare
full_office = [office.name, office.capacity, office.type_]
full_db_office = [
db_office.roomname, db_office.roomcapacity, db_office.roomtype
]
session.close()
self.assertEqual(full_db_office, full_office)
def test_instantiate_room_from_name(self):
livingspace = LivingSpace('MyLLLLL')
LivingSpace.add(livingspace)
office = Office('MyJJJSS')
Office.add(office)
livingspace1 = LivingSpace.from_name("MyLLLLL")
office1 = Office.from_name("MyJJJSS")
self.assertEqual(office.name, office1.name)
def get_room(room_name):
try:
return LivingSpace.from_name(room_name)
except ValueError:
pass
try:
return Office.from_name(room_name)
except ValueError:
raise ValueError("specifed room is unknown")