def test_matches(corpus_index, mock_hlom, add_doc, add_hlom):
"""
When OSP documents match the query, write link rows.
"""
d1 = add_doc('War and Peace, Leo Tolstoy 1')
d2 = add_doc('War and Peace, Leo Tolstoy 2')
d3 = add_doc('War and Peace, Leo Tolstoy 3')
d4 = add_doc('Anna Karenina, Leo Tolstoy 1')
d5 = add_doc('Anna Karenina, Leo Tolstoy 2')
Document_Text.es_insert()
record = add_hlom('War and Peace', 'Leo Tolstoy')
query(record.id)
# Should write 3 citation links.
assert HLOM_Citation.select().count() == 3
# Should match the right documents.
for doc in [d1, d2, d3]:
assert HLOM_Citation.select().where(
HLOM_Citation.document==doc,
HLOM_Citation.record==record
)
def test_no_matches(corpus_index, add_doc, add_hlom):
"""
When no documents match, don't write any rows.
"""
add_doc('War and Peace, Leo Tolstoy')
Document_Text.es_insert()
record = add_hlom('Master and Man', 'Leo Tolstoy')
query(record.id)
# Shouldn't write any rows.
assert HLOM_Citation.select().count() == 0