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