def test_crud(self): with self.app_context(): # Create event and assert that it is not in the database mock_event = EventTest.create_mock_event() self.assertIsNone(EventModel.find_by_id(mock_event.id)) # Save event to database and assert that it is in the database mock_event.save_to_db() self.assertIsNotNone(EventModel.find_by_id(mock_event.id)) # Update event details and save to database then assert that details have been updated mock_event.title = "Updated title" mock_event.description = "Updated description" mock_event.set_occurred_on("1755-01-11") mock_event.save_to_db() updated_mock_event = EventModel.find_by_id(mock_event.id) self.assertEqual("Updated title", updated_mock_event.title) self.assertEqual("Updated description", updated_mock_event.description) self.assertEqual("Sat Jan 11 00:00:00 1755", updated_mock_event.occurred_on.ctime()) # Delete event from database and assert event is no longer in database updated_mock_event.delete_from_db() self.assertIsNone(EventModel.find_by_id(updated_mock_event.id))
def get(self, id_all): event_data = Event.find_by_id(id_all) or Event.find_by_author(id_all) if event_data: return event_data.json() else: return {'message': 'Event not found'}, 404
def mutate_and_get_payload(cls, root, info, **event_input): status = None event = EventModel.find_by_id(event_input.get('id')) if event is not None: event.delete_from_db() status = "ok" else: status = "event with ID {} not found.".format(event_input.get('id')) return DeleteEvent(status=status)
def mutate_and_get_payload(cls, root, info, **event_input): status = None event = EventModel.find_by_id(event_input.get('id')) if event is not None: event.title = event_input.get('title') event.description = event_input.get('description') event.set_occurred_on(event_input.get('occurred_on')) event.save_to_db() status = "ok" else: status = "event with ID {} not found.".format(event_input.get('id')) return UpdateEvent(event=event, status=status)
def delete(self, id_all): event_data = Event.find_by_id(id_all) if event_data: try: event_data.delete_to_db() except: return {"message": "An error occurred update the event."}, 500 return {'message': 'Event succes delete'}, 200 else: return {'message': 'Event succes delete'}, 404
def put(self, id_all): data = Events.parse.parse_args() event_data = Event.find_by_id(id_all) if event_data: event_data.title = data['title'] event_data.content = data['content'] try: event_data.save_to_db() except: return {"message": "An error occurred update the event."}, 500 return {'message': 'event succes update'}, 200 else: return {'message': 'event not found'}, 404