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 test_reorder_sections(self):
template = util.insert_new_template()
self.assertEquals(template.section_order, None)
# insert some sections
util.insert_new_section()
util.insert_new_section()
util.insert_new_section()
new_order = [3,1,2]
sc.reorder_sections(template, new_order)
self.assertEquals(template.section_order, new_order)
def test_get_template_sections(self):
template = util.insert_new_template()
self.assertEquals(template.section_order, None)
# insert some sections
util.insert_new_section()
util.insert_new_section()
util.insert_new_section()
# assert that we get the sections
sections = sc.get_template_sections(template)
self.assertEquals(len(sections), 3)
# assert that if we have an order, the sections
# are returned in that order
new_order = [3,1,2]
sc.reorder_sections(template, new_order)
sections = sc.get_template_sections(template)
self.assertEquals(len(sections), 3)
self.assertEquals([i.id for i in sections], new_order)
def test_update_section(self):
# add the unicode placeholder type
db.session.execute('''INSERT INTO placeholder_types VALUES (1, 'unicode')''')
# generate a template, section, and document
template = util.insert_new_template()
section = util.insert_new_section()
document_id = dm.create_new_document(template.id, {'name': 'foobar'})
# add content to the section
new_content = {
'widget': 'this is a <span class="js-fr-placeholder">[[Text||foo]]</span>' +
'<span class="js-fr-placeholder">[[Text||bar]]</span>' +
'<span class="js-fr-placeholder">[[Text||baz]]</span>'
}
# update the section and document with our new content
sc.update_section(section, [], template.id, new_content)
dm.update_documents(template.id)
# test that the placeholders made it in ok
placeholders = TemplatePlaceholders.query.all()
self.assertEquals(len(placeholders), 3)
self.assertEquals(section.text, new_content.get('widget'))
# add values to each of the placeholders
dm.save_document_section(placeholders, {'[[foo]]': 'foo', '[[bar]]': 'foo', '[[baz]]': 'foo'})
# test that the values were set properly
document_placeholders = dm.get_document_placeholders(document_id)
self.assertEquals(len(document_placeholders), 3)
for placeholder in document_placeholders:
self.assertEquals(placeholder.value, 'foo')
# new content, deleting old placeholders
new_content2 = {
'widget': 'this is a section with fewer placeholders <span class="js-fr-placeholder">[[Text||foo]]</span>'
}
# update the section, documents
sc.update_section(section, placeholders, template.id, new_content2)
dm.update_documents(template.id)
# test that everything is correct with the section
self.assertEquals(len(TemplatePlaceholders.query.all()), 1)
self.assertEquals(section.text, new_content2.get('widget'))
# test that the documents were updated properly
placeholders = TemplatePlaceholders.query.all()
document_placeholders = dm.get_document_placeholders(document_id)
self.assertEquals(len(document_placeholders), 1)
self.assertEquals(document_placeholders[0].value, 'foo')
# update the section, additional content
# add content to the section
new_content = {
'widget': 'this is a <span class="js-fr-placeholder">[[Text||foo]]</span>' +
'<span class="js-fr-placeholder">[[Text||bar]]</span>'
}
# update the section and document with our new content
sc.update_section(section, placeholders, template.id, new_content)
dm.update_documents(template.id)
# test that the old value is still set
document_placeholders = dm.get_document_placeholders(document_id)
self.assertEquals(len(document_placeholders), 2)
for placeholder in document_placeholders:
if placeholder.display_name == '[[foo]]':
self.assertEquals(placeholder.value, 'foo')
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)