buildermanagerservicer implementation

buildman/buildmanagerservicer.py

@@ -103,7 +103,7 @@ def Heartbeat(self, request_iterator, context):
                 yield False
             else:
                 job_id = decoded_token["access"]["build_job_id"]
-                self._lifecycle_manager.job_heartbeat(job_id)
+                self._lifecycle_manager.heartbeat(job_id)
                 yield True
 
     def LogMessage(self, request, context):
@@ -228,3 +228,51 @@ def _decode_jwt_token(self, token, subject):
             return None, msg
 
         return decoded, msg
+
+
+class NoopServicer(buildman_pb2_grpc.BuildManagerServicer):
+    def __init__(self, container):
+        self._container = container
+
+    def Ping(self, request, context):
+        self._container.counter += 1
+        print("PINGING", self._container.counter)
+        return buildman_pb2.PingReply(reply="PONG")
+
+    def RegisterBuildJob(self, request, context):
+        self._container.counter += 1
+        print("REGISTERING BUILD", self._container.counter)
+
+    def Heartbeat(self, request_iterator, context):
+        self._container.counter += 1
+        print("HEARTBEATING", self._container.counter)
+
+    def LogMessage(self, request, context):
+        self._container.counter += 1
+        print("LOGGING", self._container.counter)
+
+    def DetermineCachedTag(self, request, context):
+        self._container.counter += 1
+        print("CACHEING", self._container.counter)
+
+
+if __name__ == "__main__":
+    from concurrent import futures
+
+    class ContainerClass(object):
+        def __init__ (self):
+            self.counter = 0
+
+        def _work_checker(self):
+            import time
+            while True:
+                print("CHECKING FOR WORK...")
+                time.sleep(5)
+
+    server = grpc.server(futures.ThreadPoolExecutor(max_workers=10))
+    container = ContainerClass()
+    buildman_pb2_grpc.add_BuildManagerServicer_to_server(NoopServicer(container), server)
+    server.add_insecure_port("[::]:50051")
+    server.start()
+
+    container._work_checker()