Generalise vcr cassettes path

uktrade · Dec 7, 2024 · 1f549d0 · 1f549d0
1 parent 251b6c8
commit 1f549d0
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/test/client/test_handler.py b/test/client/test_handler.py
@@ -1,10 +1,16 @@
 from functools import wraps
+from os import path
+from pathlib import Path
 
 from matchbox.client._handler import get_resolution_graph
 from vcr import use_cassette
 
 
 def vcr_cassette(func):
+    test_dir = Path(__file__).resolve().parents[1]
+    cassette_name = f"{func.__name__}.yaml"
+    path.join(test_dir, "fixtures", cassette_name)
+
     @wraps(func)
     @use_cassette(f"test/fixtures/vcr_cassettes/{func.__name__}.yaml")
     def wrapper(*args, **kwargs):