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)
