def test_create_document_no_client():
event = Event(event_id='1')
document = event.create_document(
'plaintext', "“You're no help,” he told the lime. "
"This was unfair. It was only a lime; "
"there was nothing special about it at all. "
"It was doing the best it could.")
assert event.documents['plaintext'] == document
def test_create_document(mocker):
client = mocker.Mock(EventsClient)
client.get_local_instance.return_value = client
event = Event(event_id='1', client=client)
document = event.create_document(
'plaintext', "“You're no help,” he told the lime. "
"This was unfair. It was only a lime; "
"there was nothing special about it at all. "
"It was doing the best it could.")
assert event.documents['plaintext'] == document
client.add_document.assert_called_once_with(
'1', 'plaintext', "“You're no help,” he told the lime. "
"This was unfair. It was only a lime; "
"there was nothing special about it at all. "
"It was doing the best it could.")
def test_iter_documents(mocker):
client = mocker.Mock(EventsClient)
client.get_local_instance.return_value = client
client.get_all_document_names.return_value = ['plaintext', 'two', 'three']
event = Event(event_id='1', client=client)
documents = {document for document in event.documents}
assert documents == {'plaintext', 'two', 'three'}
def test_get_document_missing(mocker):
client = mocker.Mock(EventsClient)
client.get_local_instance.return_value = client
client.get_all_document_names.return_value = ['plaintext']
event = Event(event_id='1', client=client)
with pytest.raises(KeyError):
_ = event.documents['foo']
def test_get_document(mocker):
client = mocker.Mock(EventsClient)
client.get_local_instance.return_value = client
client.get_all_document_names.return_value = ['plaintext']
event = Event(event_id='1', client=client)
document = event.documents['plaintext']
assert document.document_name == 'plaintext'
def test_contains_documents(mocker):
client = mocker.Mock(EventsClient)
client.get_local_instance.return_value = client
client.get_all_document_names.return_value = ['plaintext', 'two', 'three']
event = Event(event_id='1', client=client)
assert 'plaintext' in event.documents
assert 'two' in event.documents
assert 'four' not in event.documents
def test_get_document_missing_no_client(mocker):
event = Event(event_id='1')
with pytest.raises(KeyError):
_ = event.documents['foo']
def test_created_indices_no_documents():
event = Event(event_id='1')
assert event.created_indices == {}
def test_documents_len(mocker):
client = mocker.Mock(EventsClient)
client.get_local_instance.return_value = client
client.get_all_document_names.return_value = ['plaintext', 'two', 'three']
event = Event(event_id='1', client=client)
assert len(event.documents) == 3