Merge pull request #98 from arasheghdam84/test/modify-tests

mohammadll · web-flow · commit 20a94536c748 · 2024-11-26T23:11:32.000+03:30
Modify Tests
diff --git a/app/tests/conftest.py b/app/tests/conftest.py
@@ -5,14 +5,16 @@
 from app.main import app
 from app.models import (
     IaCTemplateGenerationDocker, IaCTemplateGenerationEC2, IaCTemplateGenerationS3, IaCTemplateGenerationIAM,
-    IaCTemplateGenerationArgoCD, IaCTemplateGenerationELB, IaCTemplateGenerationEFS, SyncPolicy, ArgoApplication
+    IaCTemplateGenerationArgoCD, IaCTemplateGenerationELB, IaCTemplateGenerationEFS, SyncPolicy, ArgoApplication,
+    HelmTemplateGeneration, Pod, Persistance, Ingress, Environment, IaCBasicInput, IaCBugfixInput, IaCInstallationInput
 )
 
 
 @pytest.fixture
 def client():
-    with TestClient(app) as client:
-        yield client
+    # with TestClient(app) as client:
+    #     yield client
+    return TestClient(app)
 
 
 @pytest.fixture
@@ -50,3 +52,106 @@ def iac_template_elb_sample_input():
 @pytest.fixture
 def iac_template_efs_sample_input():
     return IaCTemplateGenerationEFS().model_dump()
+
+
+@pytest.fixture
+def helm_template_sample_input():
+    nginx_pod = Pod(
+        name='nginx',
+        image='nginx:latest',
+        target_port=80,
+        replicas=2,
+        persistance=Persistance(),
+        environment=[Environment(name='DEBUG', value='true')],
+        stateless=True,
+        ingress=Ingress(enabled=True)
+    )
+
+    redis_pod = Pod(
+        name='redis',
+        image='redis:latest',
+        target_port=6379,
+        replicas=1,
+        persistance=Persistance(),
+        environment=[],
+        stateless=False,
+        ingress=Ingress(enabled=False)
+    )
+
+    return HelmTemplateGeneration(api_version=3, pods=[nginx_pod, redis_pod]).model_dump()
+
+@pytest.fixture
+def helm_template_invalid_sample_input():
+    return {
+        'api_version': 0,
+        'pods': [
+            {
+                'name': '',
+                'image': 'nginx:latest',
+                'target_port': 70000,
+                'replicas': 0,
+                'persistance': {'size': 'invalid_size', 'accessModes': 'InvalidMode'},
+                'environment': [{'name': '', 'value': ''}],
+                'stateless': True,
+                'ingress': {'enabled': True, 'host': ''}
+            }
+        ]
+    }
+
+
+@pytest.fixture
+def iac_basic_sample_input():
+    return IaCBasicInput(
+        input='How do I manage state effectively in Terraform?',
+        service='terraform',
+        min_tokens=100,
+        max_tokens=500
+    ).model_dump()
+
+
+@pytest.fixture
+def iac_basic_invalid_sample_input():
+    return {
+        'input': 'Create a basic configuration',
+        'service': 'invalid_service',
+        'min_tokens': 100,
+        'max_tokens': 500
+    }
+
+
+@pytest.fixture
+def ias_bugfix_sample_input():
+    return IaCBugfixInput(
+        bug_description='Application fails to start on version latest',
+        version='latest',
+        service='terraform',
+        min_tokens=100,
+        max_tokens=500
+    ).model_dump()
+
+
+@pytest.fixture
+def iac_bugfix_invalid_sample_input():
+    return {
+        'bug_description': '',
+        'version': 'latest',
+        'service': 'terraform',
+        'min_tokens': 100,
+        'max_tokens': 500
+    }
+
+
+@pytest.fixture
+def iac_install_sample_input():
+    return IaCInstallationInput(
+        os='ubuntu',
+        service='terraform'
+    ).model_dump()
+
+
+@pytest.fixture
+def iac_install_invalid_sample_input():
+    return {
+        'os': 'Kali',  # Unsupported OS
+        'service': 'terraform',
+    }
diff --git a/app/tests/test_helm_template.py b/app/tests/test_helm_template.py
@@ -1,111 +1,34 @@
-import pytest
-from unittest.mock import patch
-from fastapi.testclient import TestClient
-from app.main import app
-from app.models import HelmTemplateGeneration, Pod, Persistance, Ingress, Environment
+from unittest.mock import MagicMock, patch, mock_open
 
-client = TestClient(app)
-@pytest.fixture
-def sample_helm_template_input():
-    """
-    Provides a valid HelmTemplateGeneration object as input for tests.
-    """
-    return HelmTemplateGeneration(
-        api_version=3,
-        pods=[
-            Pod(
-                name="nginx",
-                image="nginx:latest",
-                target_port=80,
-                replicas=2,
-                persistance=Persistance(enabled=False),
-                environment=[Environment(name="DEBUG", value="true")],
-                stateless=True,
-                ingress=Ingress(enabled=True)
-            ),
-            Pod(
-                name="redis",
-                image="redis:latest",
-                target_port=6379,
-                replicas=1,
-                persistance=Persistance(enabled=True),
-                environment=[],
-                stateless=False,
-                ingress=Ingress(enabled=False)
-            )
-        ]
-    )
 
-@pytest.fixture
-def invalid_input():
-    """
-    Provides an invalid input dictionary for tests, containing various validation errors.
-    """
-    return {
-        "api_version": 0,  
-        "pods": [
-            {
-                "name": "",  
-                "image": "nginx:latest",
-                "target_port": 70000,
-                "replicas": 0, 
-                "persistance": {"size": "invalid_size", "accessModes": "InvalidMode"},  
-                "environment": [{"name": "", "value": ""}],  
-                "stateless": True,
-                "ingress": {"enabled": True, "host": ""} 
-            }
-        ]
-    }
+class TestHelmTemplate:
+    def setup_method(self):
+        mock_client_instance = MagicMock()
+        mock_client_instance.chat.completions.create.return_value = MagicMock(
+            choices=[MagicMock(message=MagicMock(content='Mocked OpenAI Response'))]
+        )
 
-@patch("app.main.execute_pythonfile")
-@patch("app.main.edit_directory_generator")
-@patch("app.main.gpt_service")
-def test_helm_template_generation(
-    mock_gpt_service,
-    mock_edit_directory_generator,
-    mock_execute_pythonfile,
-    sample_helm_template_input
-):
-    """
-    Tests the /Helm-template/ endpoint with valid input.
-    """
-    
-    mock_gpt_service.return_value = "Generated Python Code"
-    response = client.post("/Helm-template/", json=sample_helm_template_input.model_dump())
-    assert response.status_code == 200
-    mock_gpt_service.assert_called_once()
-    mock_edit_directory_generator.assert_called_once_with("helm_generator", "Generated Python Code")
-    mock_execute_pythonfile.assert_called_once_with("MyHelm", "helm_generator")
+        self.mock_execute_python_file = patch('app.main.execute_pythonfile').start()
+        self.mock_edit_directory_generator = patch('app.main.edit_directory_generator').start()
+        self.mock_gpt_service = patch('app.main.gpt_service', return_value='Mocked GPT Response').start()
+        self.mock_openai = patch('app.gpt_services.OpenAI', return_value=mock_client_instance).start()
+        self.mock_builtin_open = patch('builtins.open', mock_open()).start()
+        self.mock_shutil_rm = patch('shutil.rmtree').start()
 
-def test_helm_invalid_api(invalid_input):
-    """
-    Tests the /Helm-template/ endpoint with an invalid API version.
-    """
-    invalid_input = invalid_input.copy()
-    invalid_input["api_version"] = 0  
+        self.url = '/Helm-template/'
 
+    def teardown_method(self):
+        patch.stopall()
 
-    response = client.post("/Helm-template/", json=invalid_input)
+    def test_helm_template_generation(self, client, helm_template_sample_input):
+        response = client.post(self.url, json=helm_template_sample_input)
+        assert response.status_code == 200
 
-    assert response.status_code == 422
+    def test_helm_invalid_api(self, client, helm_template_invalid_sample_input):
+        resource = client.post(self.url, json=helm_template_invalid_sample_input)
+        assert resource.status_code == 422
 
-
-@patch("app.main.execute_pythonfile")
-@patch("app.main.edit_directory_generator")
-@patch("app.main.gpt_service")
-def test_helm_invalid_port(
-    mock_gpt_service,
-    mock_edit_directory_generator,
-    mock_execute_pythonfile,
-    invalid_input
-):
-    """
-    Tests the /Helm-template/ endpoint with an invalid target port.
-    """
-    invalid_input = invalid_input.copy()
-    invalid_input["pods"][0]["target_port"] = 70000  
-    response = client.post("/Helm-template/", json=invalid_input)
-    assert response.status_code == 422
-    mock_gpt_service.assert_not_called()
-    mock_edit_directory_generator.assert_not_called()
-    mock_execute_pythonfile.assert_not_called()
+    def test_helm_invalid_port(self, client, helm_template_invalid_sample_input):
+        helm_template_invalid_sample_input['pods'][0]['target_port'] = 70000
+        resource = client.post(self.url, json=helm_template_invalid_sample_input)
+        assert resource.status_code == 422
diff --git a/app/tests/test_iac_basic.py b/app/tests/test_iac_basic.py
@@ -1,42 +1,25 @@
-import pytest
-from fastapi.testclient import TestClient
-from unittest.mock import patch
-from app.main import app
-from app.models import IaCBasicInput
+from unittest.mock import MagicMock, patch
 
-client = TestClient(app)
 
-@pytest.fixture
-def valid_iac_basic_data():
-    return IaCBasicInput(
-        input="How do I manage state effectively in Terraform?",
-        service="terraform",
-        min_tokens=100,
-        max_tokens=500
-    )
+class TestIaCBasic:
+    def setup_method(self):
+        mock_client_instance = MagicMock()
+        mock_client_instance.chat.completions.create.return_value = MagicMock(
+            choices=[MagicMock(message=MagicMock(content='Mocked OpenAI 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 it returns a 200 status code.
-    """
-    mock_gpt_service.return_value = "Mocked GPT response for IaC-basic"
+        self.mock_gpt_service = patch('app.main.gpt_service', return_value='Mocked GPT Response').start()
+        self.mock_openai = patch('app.gpt_services.OpenAI', return_value=mock_client_instance).start()
 
-    response = client.post("/IaC-basic/", json=valid_iac_basic_data.model_dump())
-    assert response.status_code == 200
+        self.url = '/IaC-basic/'
 
+    def teardown_method(self):
+        patch.stopall()
 
-@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 it returns a 422 status code.
-    """
-    invalid_input = {
-        "input": "Create a basic configuration",
-        "service": "invalid_service",  # Unsupported service (invalid)
-        "min_tokens": 100,
-        "max_tokens": 500
-    }
+    def test_iac_basic_generation(self, client, iac_basic_sample_input):
+        response = client.post(self.url, json=iac_basic_sample_input)
+        assert response.status_code == 200
 
-    response = client.post("/IaC-basic/", json=invalid_input)
-    assert response.status_code == 422
+    def test_basic_generation_invalid_service(self, client, iac_basic_invalid_sample_input):
+        response = client.post(self.url, json=iac_basic_invalid_sample_input)
+        assert response.status_code == 422
diff --git a/app/tests/test_iac_bugfix.py b/app/tests/test_iac_bugfix.py
@@ -1,44 +1,25 @@
-import pytest
-from fastapi.testclient import TestClient
-from unittest.mock import patch
-from app.main import app
-from app.models import IaCBugfixInput
+from unittest.mock import MagicMock, patch
 
-client = TestClient(app)
 
-@pytest.fixture
-def valid_bugfix_data():
-    return IaCBugfixInput(
-        bug_description="Application fails to start on version latest",
-        version="latest",
-        service="terraform",
-        min_tokens=100,
-        max_tokens=500
-    )
+class TestIaCBugFix:
+    def setup_method(self):
+        mock_client_instance = MagicMock()
+        mock_client_instance.chat.completions.create.return_value = MagicMock(
+            choices=[MagicMock(message=MagicMock(content='Mocked OpenAI Response'))]
+        )
 
-@patch('app.main.gpt_service')
-def test_bugfix(mock_gpt_service, valid_bugfix_data):
-    """
-    Test the /IaC-bugfix/ endpoint with valid input data to ensure it returns a 200 status code.
-    """
-    mock_gpt_service.return_value = "Mocked GPT response for IaC-bugfix"
+        self.mock_gpt_service = patch('app.main.gpt_service', return_value='Mocked GPT Response').start()
+        self.mock_openai = patch('app.gpt_services.OpenAI', return_value=mock_client_instance).start()
 
-    response = client.post("/IaC-bugfix/", json=valid_bugfix_data.model_dump())
-    assert response.status_code == 200
+        self.url = '/IaC-bugfix/'
 
-
-@patch('app.main.gpt_service')
-def test_bugfix_invalid(mock_gpt_service):
-    """
-    Test the /IaC-bugfix/ endpoint with invalid input data to ensure it returns a 422 status code.
-    """
-    invalid_input = {
-        "bug_description": "",  # Empty description
-        "version": "latest",
-        "service": "terraform",
-        "min_tokens": 100,
-        "max_tokens": 500
-    }
-
-    response = client.post("/IaC-bugfix/", json=invalid_input)
-    assert response.status_code == 422
+    def teardown_method(self):
+        patch.stopall()
+    
+    def test_iac_bugfix(self, client, ias_bugfix_sample_input):
+        response = client.post(self.url, json=ias_bugfix_sample_input)
+        assert response.status_code == 200
+    
+    def test_iac_bugfix_invalid_input(self, client, iac_bugfix_invalid_sample_input):
+        response = client.post(self.url, json=iac_bugfix_invalid_sample_input)
+        assert response.status_code == 422
diff --git a/app/tests/test_iac_install.py b/app/tests/test_iac_install.py
diff --git a/app/tests/test_iac_template.py b/app/tests/test_iac_template.py
