test: use async context manager for proper cleanup (review feedback)

Hellnight2005 · Hellnight2005 · commit f90331246c0e · 2026-02-04T16:52:56.000+05:30
diff --git a/tests/test_502_handling.py b/tests/test_502_handling.py
@@ -7,7 +7,6 @@
 async def test_502_html_handling():
     """Test that 502 errors with HTML bodies are sanitized"""
     config = {"api_key": "test_key", "api_url": "https://api.test.com"}
-    engine = LingoDotDevEngine(config)
 
     html_body = """<html><body><h1>502 Bad Gateway</h1></body></html>"""
 
@@ -22,8 +21,9 @@ async def test_502_html_handling():
         )  # simulating non-JSON response
         mock_post.return_value = mock_response
 
-        with pytest.raises(RuntimeError) as exc_info:
-            await engine.localize_text("hello", {"target_locale": "es"})
+        async with LingoDotDevEngine(config) as engine:
+            with pytest.raises(RuntimeError) as exc_info:
+                await engine.localize_text("hello", {"target_locale": "es"})
 
         error_msg = str(exc_info.value)
 
@@ -38,8 +38,6 @@ async def test_502_html_handling():
 async def test_500_json_handling():
     """Test that 500 errors with JSON bodies are preserved"""
     config = {"api_key": "test_key", "api_url": "https://api.test.com"}
-    engine = LingoDotDevEngine(config)
-
     error_json = {"error": "Specific internal error message"}
 
     with patch("lingodotdev.engine.httpx.AsyncClient.post") as mock_post:
@@ -50,8 +48,9 @@ async def test_500_json_handling():
         mock_response.json.return_value = error_json
         mock_post.return_value = mock_response
 
-        with pytest.raises(RuntimeError) as exc_info:
-            await engine.localize_text("hello", {"target_locale": "es"})
+        async with LingoDotDevEngine(config) as engine:
+            with pytest.raises(RuntimeError) as exc_info:
+                await engine.localize_text("hello", {"target_locale": "es"})
 
         error_msg = str(exc_info.value)
 
