Exemplos de get_all_by_restaurant_key em Python

Exemplo n.º 1

Arquivo: category_logic_test.py Projeto: niczy/hyo

  def test_add_and_get_categories(self):
    restaurant = restaurant_logic.add('FulinMen', 'restaurant', self.image_key) 
    category_logic.add(restaurant.key, "Category 1")
    category_logic.add(restaurant.key, "Cateogyr 2")
    categories = category_logic.get_all_by_restaurant_key(restaurant.key)
    self.assertEqual(2, len(categories))

    self.assertIsNotNone(category_logic.add('FulinMen', 'Category 3'))
    category_logic.add('FulinMen', 'Category 4')
    categories = category_logic.get_all_by_restaurant_key(restaurant.key)
    self.assertEqual(4, len(categories))

Exemplo n.º 2

Arquivo: category_logic_test.py Projeto: niczy/hyo

 def test_get_category_not_exist(self):
   try:
     categories = category_logic.get_all_by_restaurant_key(
         ndb.Key('Restaurant', 'id'))
     self.fail('Should throw RestaurantNotExistError')
   except RestaurantNotExistError as e:
     pass