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 })