def test_display_cleanup(self):
with current_app.test_request_context():
new_form = CategoryForm()
self.assertFalse(hasattr(new_form, '_categories'))
self.assertFalse(hasattr(new_form, '_subcategories'))
new_form.display_cleanup(all_categories=self.all_categories)
self.assertTrue(hasattr(new_form, '_categories'))
self.assertTrue(hasattr(new_form, '_subcategories'))
categories = json.loads(new_form.get_categories())
self.assertEquals(len(categories), 2)
self.assertTrue('one' in categories)
self.assertTrue('ten' in categories)
subcategories = json.loads(new_form.get_subcategories())
self.assertEquals(len(subcategories), 3)
def test_display_cleanup(self):
with current_app.test_request_context():
new_form = CategoryForm()
self.assertFalse(hasattr(new_form, '_categories'))
self.assertFalse(hasattr(new_form, '_subcategories'))
new_form.display_cleanup(all_categories=self.all_categories)
self.assertTrue(hasattr(new_form, '_categories'))
self.assertTrue(hasattr(new_form, '_subcategories'))
categories = json.loads(new_form.get_categories())
self.assertEquals(len(categories), 2)
self.assertTrue('one' in categories)
self.assertTrue('ten' in categories)
subcategories = json.loads(new_form.get_subcategories())
self.assertEquals(len(subcategories), 3)
def test_build_categories(self):
with current_app.test_request_context():
new_form = CategoryForm()
subcats = new_form.build_categories(self.all_categories)
self.assertEquals(len(subcats), 3)
self.assertEquals(len(subcats['Select All']), 4)
def test_build_categories(self):
with current_app.test_request_context():
new_form = CategoryForm()
subcats = new_form.build_categories(self.all_categories)
self.assertEquals(len(subcats), 3)
self.assertEquals(len(subcats['Select All']), 4)
