test(main) remove content asset in test_iac_install

Author: maralzar

app/tests/test_iac_install.py

@@ -5,7 +5,6 @@
 from app.models import IaCInstallationInput
 
 client = TestClient(app)
-mocked_gpt_response = "Mocked shell script for installing terraform on Ubuntu."
 
 @pytest.fixture
 def valid_installation_data():
@@ -15,37 +14,29 @@ def valid_installation_data():
         min_tokens=100,
         max_tokens=500
     )
-@patch('app.main.gpt_service', return_value=mocked_gpt_response)
+
+@patch('app.main.gpt_service')
 def test_install(mock_gpt_service, valid_installation_data):
     """
-    Test the /IaC-install/ endpoint with valid input data to ensure correct output.
+    Test the /IaC-install/ endpoint with valid input data to ensure it returns a 200 status code.
     """
+    mock_gpt_service.return_value = "Mocked shell script for installing terraform on Ubuntu."
+
     response = client.post("/IaC-install/", json=valid_installation_data.model_dump())
     assert response.status_code == 200
-    assert response.json() == {"output": mocked_gpt_response}
+
 
 @patch('app.main.gpt_service')
 def test_install_invalid(mock_gpt_service):
     """
-    Test the /IaC-install/ endpoint with an invalid 'os' value to ensure proper validation.
+    Test the /IaC-install/ endpoint with an invalid 'os' value to ensure it returns a 422 status code.
     """
     invalid_input = {
-        "os": "Kali",  
-        "service": "terraform"
+        "os": "Kali",  # Unsupported OS
+        "service": "terraform",
+        "min_tokens": 100,
+        "max_tokens": 500
     }
 
     response = client.post("/IaC-install/", 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", "os"]
-    expected_error_msg = "OS must be one of ['ubuntu', 'centos', 'debian']."
-
-    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