views_test.py 文件源码

def test_valid_delete_reservation(monkeypatch):
    with app.app_context():
        with app.test_request_context(method='DELETE'):
            def reservation_not_found(_):
                room = Room(1)
                user = User('buddy', 'boy')
                time = Timeslot(1, 2, datetime(2020, 01, 01), '', 1, 1)
                equipment = Equipment("EquipmentID_iionask")
                return Reservation(room, user, time, 'description', equipment,'test')

            def empty_return(*args, **kwargs):
                return

            monkeypatch.setattr(ReservationMapper, 'find', reservation_not_found)
            monkeypatch.setattr(ReservationMapper, 'delete', empty_return)
            monkeypatch.setattr(ReservationMapper, 'done', empty_return)
            monkeypatch.setattr(WaitingMapper, 'find', reservation_not_found)
            monkeypatch.setattr(WaitingMapper, 'delete', empty_return)
            monkeypatch.setattr(WaitingMapper, 'done', empty_return)
            monkeypatch.setattr(TimeslotMapper, 'delete', empty_return)
            monkeypatch.setattr(TimeslotMapper, 'done', empty_return)
            monkeypatch.setattr(EquipmentMapper, 'delete', empty_return)
            monkeypatch.setattr(EquipmentMapper, 'done', empty_return)

            views.session.clear()
            views.session.update({'logged_in': True, 'username': 'pasta'})
            response = views.delete_reservation('test')
            assert (response.status_code == views.STATUS_CODE['OK'])
            response_data = json.loads(response.get_data())
            assert (isinstance(response_data, dict))
            assert ('reservationId' in response_data or 'waitingId' in response_data)