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.")