def test_GetDocumentText_bad_event(events_server):
_, _, status_code, _ = events_server.invoke_unary_unary(
events_pb2.DESCRIPTOR.services_by_name['Events'].
methods_by_name['GetDocumentText'], {},
events_pb2.GetDocumentTextRequest(event_id='1',
document_name='plaintext'),
None).termination()
assert status_code == grpc.StatusCode.NOT_FOUND
def test_GetDocumentText(events_server):
events_server.invoke_unary_unary(
events_pb2.DESCRIPTOR.services_by_name['Events'].
methods_by_name['OpenEvent'], {},
events_pb2.OpenEventRequest(event_id='1'), None)
events_server.invoke_unary_unary(
events_pb2.DESCRIPTOR.services_by_name['Events'].
methods_by_name['AddDocument'], {},
events_pb2.AddDocumentRequest(event_id='1',
document_name='plaintext',
text=PHASERS), None)
response, _, _, _ = events_server.invoke_unary_unary(
events_pb2.DESCRIPTOR.services_by_name['Events'].
methods_by_name['GetDocumentText'], {},
events_pb2.GetDocumentTextRequest(event_id='1',
document_name='plaintext'),
None).termination()
assert response.text == PHASERS
def get_document_text(self, event_id, document_name):
request = events_pb2.GetDocumentTextRequest(
event_id=event_id, document_name=document_name)
response = self.stub.GetDocumentText(request)
return response.text
