def test_add_restaurant(self):
letters = string.ascii_letters
random_name = ''.join(random.choice(letters) for i in range(12))
new_r = Restaurant(name=random_name, cuisine="POST_TEST_CUISINE")
res = self.client().post(f'/restaurants', json=new_r.format(),
headers=self.headers)
data = json.loads(res.data)
restaurant = Restaurant.query.filter_by(name=random_name).one_or_none()
self.assertEqual(res.status_code, 200)
self.assertEqual(data['restaurants'][0]['name'], restaurant.name)
self.assertTrue(restaurant.id)
self.assertTrue(data['success'])
def post_restaurant(payload):
try:
data = request.get_json()
name = data['name']
cuisine = data['cuisine']
new_restaurant = Restaurant(name=name, cuisine=cuisine)
new_restaurant.insert()
except Exception:
abort(400)
return jsonify({
'success': True,
'restaurants': [new_restaurant.format()]
})
def post_restaurants(token):
try:
owner_id = token.get('sub')
body = request.get_json()
name = body.get('name')
address = body.get('address')
restaurant = Restaurant(owner_id=owner_id, name=name, address=address)
restaurant.insert()
users_restaurant = Restaurant.query.filter(Restaurant.owner_id==owner_id).all()
except:
abort(422)
return jsonify({
"owned_restaurants": [restaurant.format() for restaurant in users_restaurant],
"success": True
})
