def test_new_document(self): insert_placeholder_type() # insert a bunch of templates insert_new_template(published=True) insert_new_template(published=True) insert_new_template(published=True) template = insert_new_template(published=True) insert_new_template(published=True) # insert some placeholders insert_new_placeholder(template_id=template.id) insert_new_placeholder(template_id=template.id) url = '/generate/new/from-template-' + str(template.id) new = self.client.get(url) self.assertEquals(new.status_code, 200) post = self.client.post(url, data=dict( name='test' )) self.assertEquals(len(DocumentBase.query.all()), 1) self.assertEquals(len(DocumentPlaceholder.query.all()), 2) self.assertEquals(post.status_code, 302) self.assertEquals(post.location, 'http://localhost/generate/1/edit')
def test_get_template_placeholders(self): template = util.insert_new_template() self.assertEquals(template.section_order, None) # insert some sections and some placeholders section1 = util.insert_new_section() section2 = util.insert_new_section() util.insert_new_placeholder(section1.id) util.insert_new_placeholder(section2.id) placeholders = ph.get_template_placeholders(template.id) self.assertEquals(len(placeholders), 2)
def setUp(self): super(DocumentTest, self).setUp() self.template = util.insert_new_template() util.insert_new_placeholder(template_id=self.template.id) util.insert_new_section() self.document = util.insert_new_document(self.template.id)