def register(email, password):
"""Handle POST request at /users/register."""
try:
user = User(email=email, password=password)
user.save()
response = {'message': 'You registered successfully. Please log in.'}
return response, 201
except sqlalchemy.exc.IntegrityError:
response = {'message': 'User already exists.'}
return response, 409
except AssertionError as e:
response = {'message': str(e)}
return response, 400
def test_ticket_cancel(self):
"""Test can cancel ticket."""
with self.app.app_context():
user = UserFactory()
user.is_admin = True
access_token = User.generate_token(user.id)
ticket = TicketFactory.build()
ticket.booked_by = user
ticket.save()
response = self.client.delete(f'/api/tickets/cancel/{ticket.id}',
headers={
'Authorization': access_token,
'content-type':
'application/json'
})
result = json.loads(response.data.decode())
self.assertEqual(response.status_code, 200)
self.assertEqual(result['message'],
'Ticket successfully cancelled.')
with self.subTest('Ticket no longer exists'):
response = self.client.get(f'/api/tickets/{user.id}',
headers={
'Authorization': access_token,
'content-type':
'application/json'
})
result = json.loads(response.data.decode())
self.assertEqual(response.status_code, 200)
self.assertEqual(len(result['tickets']), 0)
def test_create_flight(self):
"""Test flight can be created."""
with self.app.app_context():
user = UserFactory()
user.is_admin = True
access_token = User.generate_token(user.id)
route_1 = RouteFactory()
route_2 = RouteFactory()
flight = FlightFactory.build()
response = self.client.post('/api/flights',
data=json.dumps(
{
'origin_id': route_1.id,
'destination_id': route_2.id,
'departure': flight.departure,
'arrival': flight.arrival,
'price': flight.price,
'capacity': flight.capacity
},
cls=JsonEncoderWithDatetime),
headers={
'Authorization': access_token,
'content-type': 'application/json'
})
result = json.loads(response.data.decode())
self.assertEqual(response.status_code, 201)
self.assertEqual(result['message'], 'Flight created.')
def wrapper(*args, **kwargs):
access_token = request.headers.get('Authorization')
if access_token:
user_id = User.decode_token(access_token)
if isinstance(user_id, str):
abort(404, user_id)
request.user_id = user_id
return func(*args, **kwargs)
def wrapper(*args, **kwargs):
access_token = request.headers.get('Authorization')
if access_token:
user_id = User.decode_token(access_token)
user = User.query.get(user_id)
if not user.is_admin:
abort(401, 'You\'re not authorized to perform this action.')
request.user_id = user_id
return func(*args, **kwargs)
def create(db_session, *, user_in: UserInCreate) -> User:
user = User(
email=user_in.email,
hashed_password=get_password_hash(user_in.password),
)
db_session.add(user)
db_session.commit()
db_session.refresh(user)
return user
def test_get_all_flights_with_date_range(self):
"""Test can get all flights filtered by date range."""
with self.app.app_context():
user = UserFactory()
access_token = User.generate_token(user.id)
flight_1 = FlightFactory()
with self.subTest('Raise an error for dates with wrong format'):
response = self.client.get('/api/flights',
query_string={
'from': 'not-a-real-date',
'to': 'also-not-real'
},
headers={
'Authorization': access_token,
'content-type': 'application/json'
})
result = json.loads(response.data.decode())
self.assertEqual(response.status_code, 422)
self.assertIn("does not match format '%Y-%m-%dT%H:%M:%S'", result['messages'])
with self.subTest('No results for flights in the past'):
response = self.client.get('/api/flights',
query_string={
'from': '2018-07-19T00:00:00',
'to': '2018-07-23T00:00:00'
},
headers={
'Authorization': access_token,
'content-type': 'application/json'
})
result = json.loads(response.data.decode())
self.assertEqual(response.status_code, 200)
self.assertEqual(len(result['flights']), 0)
with self.subTest('Results adhere to date range filter'):
from_date = flight_1.departure - timedelta(days=1)
to_date = flight_1.arrival + timedelta(days=1)
response = self.client.get('/api/flights',
query_string={
'from': from_date.strftime('%Y-%m-%dT%H:%M:%S'),
'to': to_date.strftime('%Y-%m-%dT%H:%M:%S')
},
headers={
'Authorization': access_token,
'content-type': 'application/json'
})
result = json.loads(response.data.decode())
self.assertEqual(response.status_code, 200)
self.assertGreater(len(result['flights']), 0)
def test_photo_download(self):
"""Test user can download photo."""
with self.app.app_context():
photo = PhotoFactory()
access_token = User.generate_token(photo.uploaded_by_id)
response = self.client.get('/api/users/photo/download',
headers={
'Authorization': access_token,
})
self.assertEqual(response.status_code, 200)
self.assertEqual(response.data, photo.data)
def update(db_session, *, user: User, user_in: UserInUpdate) -> User:
user_data = jsonable_encoder(user)
for field in user_data:
if field in user_in.fields:
value_in = getattr(user_in, field)
if value_in is not None:
setattr(user, field, value_in)
if user_in.password:
passwordhash = get_password_hash(user_in.password)
user.hashed_password = passwordhash
db_session.add(user)
db_session.commit()
db_session.refresh(user)
return user
def test_get_by_destination(self):
"""Test can get flight by destination."""
with self.app.app_context():
user = UserFactory()
access_token = User.generate_token(user.id)
flight = FlightFactory()
response = self.client.get(f'/api/flights/destination/{flight.destination_id}',
headers={
'Authorization': access_token,
'content-type': 'application/json'
})
result = json.loads(response.data.decode())
self.assertEqual(response.status_code, 200)
self.assertEqual(len(result['flights']), 1)
self.assertEqual(result['flights'][0]['origin_id'], flight.origin_id)
self.assertEqual(result['flights'][0]['destination_id'], flight.destination_id)
def test_get_by_user_id(self):
"""Test can get ticket by user id."""
with self.app.app_context():
user = UserFactory()
user.is_admin = True
access_token = User.generate_token(user.id)
ticket = TicketFactory()
response = self.client.get(f'/api/tickets/{ticket.booked_by_id}',
headers={
'Authorization': access_token,
'content-type': 'application/json'
})
result = json.loads(response.data.decode())
self.assertEqual(response.status_code, 200)
self.assertEqual(len(result['tickets']), 1)
self.assertEqual(result['tickets'][0]['flight_id'],
ticket.flight_id)
def test_create_ticket(self):
"""Test ticket can be created."""
with self.app.app_context():
user = UserFactory()
user.is_admin = True
access_token = User.generate_token(user.id)
flight = FlightFactory()
response = self.client.post('/api/tickets/book',
data=json.dumps(
{'flight_id': flight.id}),
headers={
'Authorization': access_token,
'content-type': 'application/json'
})
result = json.loads(response.data.decode())
self.assertEqual(response.status_code, 201)
self.assertEqual(result['message'], 'Ticket successfully booked.')
def test_get_mine(self):
"""Test user can get own ticket details."""
with self.app.app_context():
user = UserFactory()
access_token = User.generate_token(user.id)
ticket = TicketFactory.build()
ticket.booked_by = user
ticket.save()
response = self.client.get('/api/tickets/mine',
headers={
'Authorization': access_token,
'content-type': 'application/json'
})
result = json.loads(response.data.decode())
self.assertEqual(response.status_code, 200)
self.assertEqual(len(result['tickets']), 1)
self.assertEqual(result['tickets'][0]['flight_id'],
ticket.flight_id)
def test_photo_upload(self):
"""Test user can upload photo."""
with self.app.app_context():
user = UserFactory()
access_token = User.generate_token(user.id)
with self.subTest('Jpgs are uploaded correctly'):
response = self.client.post(
'/api/users/photo/upload',
data={'data': (io.BytesIO(b'yaddayadda'), 'test_photo.jpg')},
headers={
'Authorization': access_token,
'content-type': 'multipart/form-data'
})
result = json.loads(response.data.decode())
self.assertEqual(response.status_code, 201)
self.assertEqual(result['message'],
'Photo test_photo.jpg successfully uploaded.')
with self.app.app_context():
photo = Photo.query.get(1)
self.assertEqual(photo.data, b'yaddayadda')
with self.subTest('None Jpgs are NOT uploaded'):
response = self.client.post(
'/api/users/photo/upload',
data={'data': (io.BytesIO(b'yaddayadda'), 'test_photo.txt')},
headers={
'Authorization': access_token,
'content-type': 'multipart/form-data'
})
result = json.loads(response.data.decode())
self.assertEqual(response.status_code, 422)
self.assertEqual(result['messages']['data'], ['Invalid value.'])