test(main): remove message content assert in test_iac_basic

Author: maralzar

app/tests/test_iac_basic.py

@@ -5,7 +5,6 @@
 from app.models import IaCBasicInput
 
 client = TestClient(app)
-mocked_gpt_response = "Mocked GPT response for IaC-basic"
 
 @pytest.fixture
 def valid_iac_basic_data():
@@ -16,19 +15,21 @@ def valid_iac_basic_data():
         max_tokens=500
     )
 
-@patch('app.main.gpt_service', return_value=mocked_gpt_response)
+@patch('app.main.gpt_service')
 def test_iac_basic_generation(mock_gpt_service, valid_iac_basic_data):
     """
-    Test the /IaC-basic/ endpoint with valid input data to ensure correct output.
+    Test the /IaC-basic/ endpoint with valid input data to ensure it returns a 200 status code.
     """
+    mock_gpt_service.return_value = "Mocked GPT response for IaC-basic"
+
     response = client.post("/IaC-basic/", json=valid_iac_basic_data.model_dump())
     assert response.status_code == 200
-    assert response.json() == {"output": mocked_gpt_response}
-    
-@patch('app.main.gpt_service')  
-def test_basic_generation(mock_gpt_service):
+
+
+@patch('app.main.gpt_service')
+def test_basic_generation_invalid_service(mock_gpt_service):
     """
-    Test the /IaC-basic/ endpoint with an invalid service to ensure proper validation.
+    Test the /IaC-basic/ endpoint with an invalid service to ensure it returns a 422 status code.
     """
     invalid_input = {
         "input": "Create a basic configuration",
@@ -38,15 +39,4 @@ def test_basic_generation(mock_gpt_service):
     }
 
     response = client.post("/IaC-basic/", json=invalid_input)
-    assert response.status_code == 422, f"Expected status code 422, got {response.status_code}"
-    assert "detail" in response.json(), "Response JSON does not contain 'detail'"
-    errors = response.json()["detail"]
-    expected_error_loc = ["body", "service"]
-    expected_error_msg = "Service must be one of ['terraform']."
-    assert any(
-        error["loc"] == expected_error_loc and error_msg in error["msg"]
-        for error in errors
-        for error_msg in [expected_error_msg]
-    ), f"Expected error message '{expected_error_msg}' at location {expected_error_loc}, but not found."
-
-    mock_gpt_service.assert_not_called()
+    assert response.status_code == 422