def test_crud(self):
with self.app_context():
room = RoomModel(_id=1,
company='Test Company',
location='Test Location',
name='Test Name',
description='Some Test Description',
rateofescape='10/10',
image='test.jpg',
duration='60 minutes',
playedon='01/01/2018')
self.assertIsNone(RoomModel.find_by_id(1))
room.save_to_db()
self.assertIsNotNone(
RoomModel.find_by_name_company('Test Name', 'Test Company'))
self.assertIsNotNone(RoomModel.find_by_id(1))
self.assertIsNotNone(RoomModel.get_all_rooms())
room.delete_from_db()
self.assertIsNone(RoomModel.find_by_id(1))
self.assertEqual(RoomModel.get_all_rooms(), {'rooms': []})
def get(self):
rooms = RoomModel.get_all_rooms()
if rooms:
return rooms
return {'message': 'No rooms to display'}, 404
