More test

danielgtaylor · Aug 14, 2024 · d51a92f · d51a92f
1 parent 3a5d527
commit d51a92f
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/tests/inputs/deprecated/deprecated.proto b/tests/inputs/deprecated/deprecated.proto
@@ -16,5 +16,6 @@ message Message {
 message Empty {}
 
 service TestService {
+    rpc func(Empty) returns (Empty);
     rpc deprecated_func(Empty) returns (Empty) { option deprecated = true; };
 }
diff --git a/tests/test_deprecated.py b/tests/test_deprecated.py
@@ -50,11 +50,15 @@ def test_message_with_deprecated_field_not_set_default(message):
 
 @pytest.mark.asyncio
 async def test_service_with_deprecated_method():
-
-    stub = TestServiceStub(MockChannel([Empty()]))
+    stub = TestServiceStub(MockChannel([Empty(), Empty()]))
 
     with pytest.warns(DeprecationWarning) as record:
         await stub.deprecated_func(Empty())
 
     assert len(record) == 1
     assert str(record[0].message) == f"TestService.deprecated_func is deprecated"
+
+    with pytest.warns(None) as record:
+        await stub.func(Empty())
+
+    assert not record