Return self in grpc client to be consistent

taverntesting · Jan 27, 2024 · 963caa7 · 963caa7
1 parent 875e522
commit 963caa7
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tavern/_plugins/grpc/client.py b/tavern/_plugins/grpc/client.py
@@ -239,8 +239,9 @@ def _make_call_request(
 
         return self._get_grpc_service(channel, service, method)
 
-    def __enter__(self) -> None:
+    def __enter__(self) -> "GRPCClient":
         logger.debug("Connecting to GRPC")
+        return self
 
     def call(
         self,