def list(self, request):
''' Return a list of categories. '''
categories = Category.query().fetch()
messages = []
for category in categories:
messages.append(category.to_message())
return common_messages.Categories(categories=messages)
def test_category_delete_method(self):
''' Add a category and then delete it through the api. '''
slug = 'test-slug'
category_key = db_loader.create_category(slug=slug)
params = {
'key': encrypt(category_key.urlsafe()),
}
response = generic_service_method_success_test(self, 'category', 'delete', params=params)
self.assertEqual(response['response']['type'], 'Echo',
'Category put service method failed.')
self.assertEqual(len(Category.query().fetch(1)), 0, 'Failed to delete category.')
def put(self, request):
''' Create or edit a category. '''
if not request.key:
# Create a new category.
category = Category(key=ndb.Key('Category', request.slug))
else:
# Get the category to edit.
category_key = ndb.Key(urlsafe=self.decrypt(request.key))
category = category_key.get()
if not category:
# TODO: What to do on error?
return common_messages.Category()
# Update the category.
category.mutate_from_message(request)
category.put()
return category.to_message()
def test_insert_entity(self):
category_key = db_loader.create_category()
self.assertTrue(category_key, "Failed to create an entity and return a key.")
self.assertEqual(1, len(Category.query().fetch(2)), "Failed to retrieve a stored entity.")
