captured warnings in pytest

tests/test_RestServer.py

 import os
 import json
 import pytest
+import warnings
 from fastapi.testclient import TestClient
 from dccXMLJSONConv.dccServer import app
 
@@ -18,17 +19,26 @@ def test_root():
         "message": "Hello!\nBetter use the URL /json2dcc/?js={...} \n or /dcc2json (POST method)"}
 
 
-@pytest.mark.skipif(not os.path.exists(os.path.join(data_dir, "APITestXMLStub.xml")), reason="Test XML file not found")
+@pytest.mark.skipif(not os.path.exists(os.path.join(data_dir, "APITestXMLStub.xml")),
+                    reason="Test XML file not found")
 def test_dcc2json_with_valid_xml():
     xml_data_path = os.path.join(data_dir, "APITestXMLStub.xml")
     with open(xml_data_path, "r") as f:
         xml_data = f.read()
 
-    response = client.post("/dcc2json/", json={"xml": xml_data})
+    # Capture the expected warning
+    with pytest.warns(UserWarning, match="XML->JSON Validation error:.*"):
+        response = client.post("/dcc2json/", json={"xml": xml_data})
+
+    # Validate response
     response_json = response.json()
+    expected_json_path = os.path.join(data_dir, "APITestJSONStub.json")
+
+    with open(expected_json_path, "r") as f:
+        expected_json = json.load(f)
 
     assert response.status_code == 200
-    assert response_json == json.loads(open(os.path.join(data_dir, "APITestJSONStub.json"), "r").read())
+    assert response_json == expected_json
 
 
 def test_dcc2json_with_empty_xml():
@@ -43,9 +53,17 @@ def test_json2dcc_with_valid_json():
     with open(json_data_path, "r") as f:
         json_data = json.load(f)
 
-    response = client.post("/json2dcc/", json={"js": json_data})
+    # Capture the expected warning during the request
+    with pytest.warns(UserWarning, match="JSON->XML Validation error:.*"):
+        response = client.post("/json2dcc/", json={"js": json_data})
+
+    # Validate response
     assert response.status_code == 200
-    assert response.text == open(os.path.join(data_dir, "JSONTOXMLRespons.xml"), "r").read()
+    expected_xml_path = os.path.join(data_dir, "JSONTOXMLRespons.xml")
+    with open(expected_xml_path, "r") as f:
+        expected_xml = f.read()
+
+    assert response.text == expected_xml
 
 
 def test_json2dcc_with_empty_json():