def _ext(content): # Create a document. path = mock_osp.add_file(content=content) document = Document.create(path=path) # Extract text, then date. ext_text(document.id) ext_semester(document.id) # Pop out the new row. return ( Document_Date_Semester .select() .where(Document_Date_Semester.document==document) .first() )
def test_link_with_document(models, mock_osp): """ When a semester marker is found, the metadata row should be associated with the document that was passed to the job. """ # 2 document rows. doc1 = Document.create(path='path1') doc2 = Document.create(path='path2') # Just 1 text row. doc_text = Document_Text.create(document=doc2, text='Fall 2012') assert doc_text.id != doc_text.document.id row = ext_semester(doc2.id) assert row.document == doc2