def test_in_progress_documents(self):
template = insert_new_template(published=True)
insert_new_document(template.id, name='test')
insert_new_document(template.id, name='test2')
response = self.client.get('/generate/edit')
assert response.status_code == 200
self.assertEquals(len(self.get_context_variable('documents')), 2)
def test_get_documents(self):
documents = dm.get_all_documents()
self.assertEquals(len(documents), 1)
util.insert_new_document(self.template.id, name='second test doc')
documents = dm.get_all_documents()
self.assertEquals(len(documents), 2)
document = dm.get_single_document(2)
self.assertEquals(document.template_id, self.template.id)
def test_edit_in_progress_documents(self):
template = insert_new_template(published=True)
document = insert_new_document(template.id, name='test')
insert_new_document(template.id, name='test2')
self.assertEquals(len(DocumentBase.query.all()), 2)
response = self.client.post('/generate/edit/' + str(document.id) + '?method=DELETE')
self.assertEquals(len(DocumentBase.query.all()), 1)
self.assertEquals(response.status_code, 302)
self.assertEquals(response.location, 'http://localhost/generate/edit')
def test_get_documents_from_single_parent(self):
template_two = util.insert_new_template(title='test two')
util.insert_new_document(template_two.id)
util.insert_new_document(template_two.id)
documents = dm.get_documents_and_parent_templates()
test_results = defaultdict(int)
for document in documents:
test_results[document.title] += 1
self.assertEquals(test_results['test'], 1)
self.assertEquals(test_results['test two'], 2)
self.assertEquals(len(test_results.keys()), 2)
self.assertEquals(sum(test_results.values()), 3)
document = dm.get_single_document_and_parent_template(self.document.id)
self.assertEquals(document[2], 'test')
def test_delete_template_cascade(self):
template = insert_new_template(published=True)
insert_new_document(template.id, name='test')
self.client.get('/build/1/edit?method=DELETE')
self.assertEquals(len(DocumentBase.query.all()), 0)
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)
