eeva/gRPC-haskell
1
0
Fork 0
mirror of https://github.com/unclechu/gRPC-haskell.git synced 2024-07-01 08:58:35 +02:00
Code Issues Projects Releases Wiki Activity

Fix python test client to send done message on exception paths

Browse Source
This commit is contained in:
Joel Stanley 2017-03-10 15:05:32 -06:00
parent 1009a2aceb
commit cdad0fcd3d
No known key found for this signature in database
GPG Key ID: 38D3B3C63EFFAB64
1 changed files with 60 additions and 57 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

71
tests/test-client.py
View File

@ -8,42 +8,36 @@ print "Starting python client"
channel = beta_implementations.insecure_channel('localhost', 50051) channel = beta_implementations.insecure_channel('localhost', 50051)
stub = beta_create_SimpleService_stub(channel) stub = beta_create_SimpleService_stub(channel)
# Test normal call: return a sum of all numbers sent to it def runTests():
print "Test 100 random sums" # Test normal call: return a sum of all numbers sent to it
for i in xrange(100): print "Test 100 random sums"
for i in xrange(100):
randints = [random.randint(0, 1000) for _ in xrange(random.randint(10, 1000))] randints = [random.randint(0, 1000) for _ in xrange(random.randint(10, 1000))]
name = "test%d" % i name = "test%d" % i
response = stub.normalCall(SimpleServiceRequest(request = name, num = randints), 10) response = stub.normalCall(SimpleServiceRequest(request = name, num = randints), 10)
assert response.response == name assert response.response == name
assert response.num == sum(randints) assert response.num == sum(randints)
# Test streaming call: The server response will be the sum of all numbers sent in the request along with a concatenation of the request name # Test streaming call: The server response will be the sum of all numbers sent in the request along with a concatenation of the request name
print "Test 100 random sums (client streaming)" print "Test 100 random sums (client streaming)"
for i in xrange(100): for i in xrange(100):
expected_sum = 0 # avoid globals
expected_response_name = '' expected_sum = [0]
expected_response_name = ['']
def send_requests(): def send_requests(esum, ename):
global expected_sum
global expected_response_name
for _ in xrange(random.randint(5, 50)): for _ in xrange(random.randint(5, 50)):
nums = [random.randint(0, 1000) for _ in xrange(random.randint(10, 100))] nums = [random.randint(0, 1000) for _ in xrange(random.randint(10, 100))]
name = str(uuid4()) name = str(uuid4())
esum[0] += sum(nums)
expected_sum += sum(nums) ename[0] += name
expected_response_name += name
yield SimpleServiceRequest(request = name, num = nums) yield SimpleServiceRequest(request = name, num = nums)
response = stub.clientStreamingCall(send_requests(expected_sum, expected_response_name), 10)
assert response.response == expected_response_name[0]
assert response.num == expected_sum[0]
response = stub.clientStreamingCall(send_requests(), 10) # Test server streaming call: The server should respond once for each number in the request
assert response.response == expected_response_name print "Test 100 random server streaming calls"
assert response.num == expected_sum for i in xrange(100):
# Test server streaming call: The server should respond once for each number in the request
print "Test 100 random server streaming calls"
for i in xrange(100):
nums = [random.randint(0, 1000) for _ in xrange(random.randint(0, 1000))] nums = [random.randint(0, 1000) for _ in xrange(random.randint(0, 1000))]
for response in stub.serverStreamingCall(SimpleServiceRequest(request = "server streaming", num = nums), 60): for response in stub.serverStreamingCall(SimpleServiceRequest(request = "server streaming", num = nums), 60):
@ -51,15 +45,11 @@ for i in xrange(100):
assert response.response == "server streaming" assert response.response == "server streaming"
nums = nums[1:] nums = nums[1:]
# Test bidirectional streaming: for each request, we should get a response indicating the sum of all numbers sent in the last request # Test bidirectional streaming: for each request, we should get a response indicating the sum of all numbers sent in the last request
print "Test bidirectional streaming" print "Test bidirectional streaming"
for i in xrange(100): for i in xrange(100):
requests = Queue.Queue() requests = Queue.Queue()
def send_requests(): def send_requests():
global cur_request
global cur_nums
global requests
for _ in xrange(random.randint(5, 50)): for _ in xrange(random.randint(5, 50)):
nums = [random.randint(0, 1000) for _ in xrange(random.randint(10, 100))] nums = [random.randint(0, 1000) for _ in xrange(random.randint(10, 100))]
name = str(uuid4()) name = str(uuid4())
@ -74,5 +64,18 @@ for i in xrange(100):
assert response.response == exp_name assert response.response == exp_name
assert response.num == exp_sum assert response.num == exp_sum
# Signal the ending of the test print "Sending DONE message to server"
stub.done(SimpleServiceDone(), 10) stub.done(SimpleServiceDone(), 10)
print "All python client tests against the generated server were successful"
try:
runTests()
except Exception as e:
print "Python client test exception caught: "
print e.__doc__
print e.message
print "Sending DONE message to server before exiting"
stub.done(SimpleServiceDone(), 10)
print "Exiting with failure status"
exit(1)