Demonstrate pytest with docker

Author: mkundu1

.github/workflows/ci.yml

@@ -174,6 +174,9 @@ jobs:
       - name: Install pyvistaqt requirements
         run: make install-pyvistaqt-requirements
 
+      - name: Pull docker image with test grpc server
+        run: docker pull mkundu1/test-grpc-server
+
       - name: Unit Testing
         run: make unittest
 

tests/docker/helloworld_pb2.py

@@ -0,0 +1,46 @@
+# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
+import grpc
+
+import helloworld_pb2 as helloworld__pb2
+
+
+class GreeterStub(object):
+  """The greeting service definition.
+  """
+
+  def __init__(self, channel):
+    """Constructor.
+
+    Args:
+      channel: A grpc.Channel.
+    """
+    self.SayHello = channel.unary_unary(
+        '/helloworld.Greeter/SayHello',
+        request_serializer=helloworld__pb2.HelloRequest.SerializeToString,
+        response_deserializer=helloworld__pb2.HelloReply.FromString,
+        )
+
+
+class GreeterServicer(object):
+  """The greeting service definition.
+  """
+
+  def SayHello(self, request, context):
+    """Sends a greeting
+    """
+    context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+    context.set_details('Method not implemented!')
+    raise NotImplementedError('Method not implemented!')
+
+
+def add_GreeterServicer_to_server(servicer, server):
+  rpc_method_handlers = {
+      'SayHello': grpc.unary_unary_rpc_method_handler(
+          servicer.SayHello,
+          request_deserializer=helloworld__pb2.HelloRequest.FromString,
+          response_serializer=helloworld__pb2.HelloReply.SerializeToString,
+      ),
+  }
+  generic_handler = grpc.method_handlers_generic_handler(
+      'helloworld.Greeter', rpc_method_handlers)
+  server.add_generic_rpc_handlers((generic_handler,))

tests/docker/helloworld_pb2_grpc.py

@@ -0,0 +1,29 @@
+import subprocess
+from time import sleep
+
+import grpc
+
+import helloworld_pb2
+import helloworld_pb2_grpc
+
+
+def test_server():
+    subprocess.run(
+        [
+            "docker",
+            "run",
+            "--name",
+            "test_server",
+            "-d",
+            "--rm",
+            "-p",
+            "50051:50051",
+            "mkundu1/test-grpc-server",
+        ]
+    )
+    sleep(2)
+    with grpc.insecure_channel("localhost:50051") as channel:
+        stub = helloworld_pb2_grpc.GreeterStub(channel)
+        response = stub.SayHello(helloworld_pb2.HelloRequest(name="World"))
+        assert response.message == "Hello, World!"
+    subprocess.run(["docker", "stop", "test_server"])