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)