test: Fix more cancellation tests

Signed-off-by: Graham King <[email protected]>

Author: grahamking

lib/bindings/python/tests/cancellation/test_cancellation.py

@@ -256,7 +256,7 @@ async def test_server_context_cancel(server, client):
     except ValueError as e:
         # Verify the expected cancellation exception is received
         # TODO: Should this be a asyncio.CancelledError?
-        assert str(e) == "Stream ended before generation completed"
+        assert str(e).startswith("Stream ended before generation completed")
 
     # Verify server context cancellation status
     assert handler.context_is_stopped

lib/bindings/python/tests/conftest.py

@@ -436,6 +436,6 @@ async def test_my_test(runtime):
             )
 
     loop = asyncio.get_running_loop()
-    runtime = DistributedRuntime(loop, "etcd", False)
+    runtime = DistributedRuntime(loop, "mem", True)
     yield runtime
     runtime.shutdown()