def test_category_get_method(self):
''' Add a category to the database then query. '''
category_slug = 'test-slug'
db_loader.create_category(slug=category_slug)
response = generic_service_method_success_test(self, 'category', 'get', params={'slug':category_slug})
self.assertEqual(response['response']['type'], 'Category',
'Category get service method failed.')
self.assertEqual(response['response']['content']['slug'], category_slug,
'Category get method returned the wrong category.')
def test_category_list_method(self):
''' Add a category to the database then query. '''
slug = 'test-slug'
db_loader.create_category(slug=slug)
response = generic_service_method_success_test(self, 'category', 'list')
self.assertEqual(response['response']['type'], 'Categories',
'System category list service method failed.')
self.assertEqual(len(response['response']['content']['categories']), 1,
'Failed to return the correct number of categories.')
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 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.")
