def get_country(country_id=''):
response = 'undefined'
if not session.query(exists().where(Country.id == country_id)).scalar():
return response
country = session.query(Country).filter_by(id=country_id).first()
return country.description
def get_eye_color(eye_color_id=''):
response = 'undefined'
if not session.query(exists().where(EyeColor.id == eye_color_id)).scalar():
return response
eye_color = session.query(EyeColor).filter_by(id=eye_color_id).first()
return eye_color.description
def get_eye_colors(request):
cache_key = request.GET.get('cache_key', '')
data = cache.get(cache_key)
if not data:
data = []
for eye_color in session.query(EyeColor).all():
row = {}
row['id'] = eye_color.id
row['title'] = eye_color.title
row['description'] = eye_color.description
data.append(row)
# Create new cache
cache_key = uuid.uuid4().hex
cache_time = 86400 * 5
cache.set(cache_key, data, cache_time)
print(cache.get(cache_key))
response = {
'status_code': 200,
'status_message': 'OK',
'cache_key': cache_key,
'data': data
}
return JsonResponse(response)
def test_create_user_with_invalid_data(self):
"""Validates error when the route receives invalid data."""
data = {
'usrname': 'testuser',
'email': '*****@*****.**',
'password': '******'
}
res = self.client.post('/user', json=data)
query = session.query(User).all()
self.assertEqual(res.status_code, 400)
self.assertEqual(len(query), 0)
def test_duplicate_user_should_not_be_registered(self):
"""Validates error when sent data is already in database."""
data = {
'username': '******',
'email': '*****@*****.**',
'password': '******'
}
usr = User(**data)
session.add(usr)
session.commit()
res = self.client.post('/user', json=data)
query = session.query(User).all()
self.assertEqual(res.status_code, 409)
self.assertEqual(len(query), 1)