From 963caa7eb63435efffb2942f2dc72409b58d8411 Mon Sep 17 00:00:00 2001
From: Michael Boulton <michaelboulton@gmail.com>
Date: Sat, 27 Jan 2024 14:33:56 +0000
Subject: [PATCH] Return self in grpc client to be consistent

---
 tavern/_plugins/grpc/client.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/tavern/_plugins/grpc/client.py b/tavern/_plugins/grpc/client.py
index 85b433d3..8e51b110 100644
--- 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,