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")