def list_shop_api():
try:
shops = Shop.find_shops_by_user(g.current_user)
return dumps([shop.as_json() for shop in shops]), 200
except ValueError as e:
return dumps(e.message), 400
except Exception as e:
print(e.message)
print(traceback.print_exc())
return dumps(e.message), 400
def test_get_home_api(self):
user_json = self.get_user_json()
user = create_user(user_json)
headers_dict = {'EMAIL': user.email}
shop_1 = create_shop(fk_user_id=user.user_id, fk_category_id=self.category.category_id)
shop_2 = create_shop(fk_user_id=user.user_id, fk_category_id=self.category.category_id)
uri = "/user/{user_id}/home".format(user_id=user.user_id)
home_response = self.app.get(
uri,
headers=headers_dict
)
self.assertEqual(200, home_response.status_code)
json_home = json.loads(home_response.data)
home = {
'banners': [],
'shops': dumps([shop.as_json() for shop in Shop.find_shops_by_user(user=user)]),
'offers': [],
'recommended': [],
'recent_view': [],
}
self.assertEqual(dumps(home), dumps(json_home))
def test_should_return_list_of_shop_by_user(self):
create_shop(fk_user_id=self.user.user_id)
create_shop(fk_user_id=self.user.user_id)
create_shop(fk_user_id=self.user.user_id)
shop = Shop.find_shops_by_user(self.user)
self.assertEqual(len(shop), 4)