def Status(self, request, context):
self.count += 1
if request.HasField("count"):
return AddTypes_pb2.StatusResponse(count={"number": self.count})
if request.HasField("state"):
response = AddTypes_pb2.StatusResponse()
response.state.states[
"sum_engine"].state = AddTypes_pb2.ServiceState.READY
response.state.states[
"difference_engine"].state = AddTypes_pb2.ServiceState.BUSY_COMPUTING_ONE
response.state.states[
"chaos_engine"].state = AddTypes_pb2.ServiceState._UNKNOWN
return response
raise NotImplementedError("request not implemented: " +
request.WhichOneof("query"))
def test_AddOnes(grpc_channel):
from Testing_protos import Add_One_pb2_grpc
from Testing_protos import AddTypes_pb2
stub = Add_One_pb2_grpc.AdditionStub(grpc_channel)
responses = stub.AddOnes(AddTypes_pb2.Addend(number=5))
for i, response in enumerate(responses):
assert response.number == 6 + i
assert i == 4
def test_AddOne(grpc_channel, lazy_cleaner):
yagrc_importer.add_lazy_packages(["Testing_protos"])
from Testing_protos import Add_One_pb2_grpc
from Testing_protos import AddTypes_pb2
yagrc_importer.resolve_lazy_imports(grpc_channel)
stub = Add_One_pb2_grpc.AdditionStub(grpc_channel)
response = stub.AddOne(AddTypes_pb2.Addend(number=5))
assert response.number == 6
def test_AddsOnes(grpc_channel, grpc_importer):
from Testing_protos import Add_One_pb2_grpc
from Testing_protos import AddTypes_pb2
stub = Add_One_pb2_grpc.AdditionStub(grpc_channel)
responses = stub.AddsOnes(
(AddTypes_pb2.Addend(number=x) for x in range(5)))
for i, response in enumerate(responses):
assert response.number == i + 1
assert i == 4
def test_StatusQuery(grpc_channel):
from Testing_protos import Add_One_pb2_grpc
from Testing_protos import AddTypes_pb2
stub = Add_One_pb2_grpc.AdditionStub(grpc_channel)
response = stub.Status(AddTypes_pb2.StatusQuery(state={}))
assert response.state.states[
"sum_engine"].state == AddTypes_pb2.ServiceState.READY
assert AddTypes_pb2.ServiceState.Name(
response.state.states["difference_engine"].state
) == "BUSY_COMPUTING_ONE"
assert response.state.states["chaos_engine"].state == AddTypes_pb2._UNKNOWN
def test_resolve_fail(grpc_channel, lazy_cleaner):
yagrc_importer.add_lazy_packages(["Testing_protos"])
from Testing_protos import AddTypes_pb2
try:
yagrc_importer.resolve_lazy_imports(None)
assert False
except AttributeError:
pass
# Test failure again to verify first failure did not lose state.
try:
yagrc_importer.resolve_lazy_imports(None)
assert False
except AttributeError:
pass
# Finally, test success case to make sure prior failures did not break
# things permanently, including the ability to do more lazy imports.
from Testing_protos import Add_One_pb2_grpc
yagrc_importer.resolve_lazy_imports(grpc_channel)
stub = Add_One_pb2_grpc.AdditionStub(grpc_channel)
response = stub.AddOne(AddTypes_pb2.Addend(number=5))
assert response.number == 6
def main():
with grpc.insecure_channel("localhost:19001") as channel:
stub = Add_One_pb2_grpc.AdditionStub(channel)
response = stub.AddOne(AddTypes_pb2.Addend(number=5))
print(response)
responses = stub.AddOnes(AddTypes_pb2.Addend(number=5))
print(*responses)
response = stub.AddsOne(
(AddTypes_pb2.Addend(number=x) for x in range(5)))
print(response)
responses = stub.AddsOnes(
(AddTypes_pb2.Addend(number=x) for x in range(5)))
print(*responses)
response = stub.Status(AddTypes_pb2.StatusQuery(count={}))
print(response)
response = stub.Status(AddTypes_pb2.StatusQuery(state={}))
print(response)
def test_CountQuery(grpc_channel):
from Testing_protos import Add_One_pb2_grpc
from Testing_protos import AddTypes_pb2
stub = Add_One_pb2_grpc.AdditionStub(grpc_channel)
response = stub.Status(AddTypes_pb2.StatusQuery(count={}))
assert response.count.number > 0
def test_AddsOne(grpc_channel):
from Testing_protos import Add_One_pb2_grpc
from Testing_protos import AddTypes_pb2
stub = Add_One_pb2_grpc.AdditionStub(grpc_channel)
response = stub.AddsOne((AddTypes_pb2.Addend(number=x) for x in range(5)))
assert response.number == 0 + 1 + 2 + 3 + 4 + 1
def test_AddOne(grpc_channel):
from Testing_protos import Add_One_pb2_grpc
from Testing_protos import AddTypes_pb2
stub = Add_One_pb2_grpc.AdditionStub(grpc_channel)
response = stub.AddOne(AddTypes_pb2.Addend(number=5))
assert response.number == 6
def AddsOnes(self, request_iterator, context):
self.count += 1
return (AddTypes_pb2.Sum(number=request.number + 1)
for request in request_iterator)
def AddsOne(self, request_iterator, context):
self.count += 1
return AddTypes_pb2.Sum(number=sum(request.number
for request in request_iterator) +
1)
def AddOnes(self, request, context):
self.count += 1
return (AddTypes_pb2.Sum(number=request.number + x + 1)
for x in range(request.number))
def AddOne(self, request, context):
self.count += 1
return AddTypes_pb2.Sum(number=request.number + 1)
