def test_text_extraction_fails(models, mock_osp):
"""
If no text can be extracted, don't write the row.
"""
# Add an empty file.
path = mock_osp.add_file(content="")
document = Document.create(path=path)
ext_text(document.id)
# Shouldn't write a row.
assert Document_Text.select().count() == 0
def test_text_extraction_succeeds(models, mock_osp):
"""
read_text() should extract text for a document and write the result into
the `document_text` table.
"""
# Add a file, create a document row.
path = mock_osp.add_file(content="text")
document = Document.create(path=path)
ext_text(document.id)
# Pop out the new row.
row = Document_Text.get(Document_Text.document == document)
assert row.text == "text"
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 _doc(content='content'):
# Write a file.
path = mock_osp.add_file(content=content)
syllabus = Syllabus(path)
# Insert the document row.
document = Document.create(path=syllabus.relative_path)
# Extract text.
text = ext_text(document.id)
return document
