add test for generator decorator return

kevin1024 · Jan 23, 2024 · 6252b92 · 6252b92
1 parent b1c45cd
commit 6252b92
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/tests/unit/test_vcr.py b/tests/unit/test_vcr.py
@@ -372,3 +372,19 @@ def test_path_class_as_cassette():
     )
     with use_cassette(path):
         pass
+
+
+def test_use_cassette_generator_return():
+    ret_val = object()
+
+    vcr = VCR()
+
+    @vcr.use_cassette("test")
+    def gen():
+        return ret_val
+        yield
+
+    with pytest.raises(StopIteration) as exc_info:
+        next(gen())
+
+    assert exc_info.value.value is ret_val