alita-sdk 0.3.130__py3-none-any.whl → 0.3.132__py3-none-any.whl

tests/test_ado_analysis.py

@@ -0,0 +1,99 @@
+import os
+import pytest
+from dotenv import load_dotenv
+
+from elitea_analyse.ado.azure_search import AzureSearch
+
+from ..alita_sdk.community.analysis.ado_analyse.api_wrapper import (
+    GetAdoWorkItemsArgs, AdoCommitsArgs, AdoAnalyseWrapper)
+
+from ..alita_sdk.clients.client import AlitaClient
+from ..alita_sdk.tools.artifact import ArtifactWrapper
+from ..alita_sdk.community.utils import check_schema
+
+# Load environment variables from .env file
+load_dotenv()
+
+
+@pytest.fixture
+def ado_api_wrapper():
+    base_url = os.getenv("DEPLOYMENT_URL")
+    project_id = os.getenv("PROJECT_ID")
+    api_key = os.getenv("API_KEY")
+
+    if not base_url or not project_id or not api_key:
+        raise ValueError("Environment variables DEPLOYMENT_URL, PROJECT_ID, and API_KEY must be set.")
+
+    client = AlitaClient(
+        base_url=base_url,
+        project_id=int(project_id),
+        auth_token=api_key,
+    )
+
+    artifacts_wrapper = ArtifactWrapper(
+        client=client, bucket=os.getenv("ARTIFACT_BUCKET_PATH", "analyse-ado")
+    )
+    check_schema(artifacts_wrapper)
+    ado_search = AzureSearch(
+        organization=os.getenv("ADO_ORGANIZATION", "john.doe@epam.com"),
+        user=os.getenv("ADO_USER", "ado_user"),
+        token=os.getenv("ADO_TOKEN", "1111"),
+    )
+
+    ado_wrapper = AdoAnalyseWrapper(
+        artifacts_wrapper=artifacts_wrapper,
+        project_keys=os.getenv("ADO_PROJECTS", "project1,project2"),
+        default_branch_name=os.getenv("ADO_DEFAULT_BRANCH", "main"),
+        area=os.getenv("ADO_AREA", ""),
+        ado_search=ado_search,
+    )
+    check_schema(ado_wrapper)
+
+    return ado_wrapper
+
+def test_get_projects_list(ado_api_wrapper):
+    result = ado_api_wrapper.get_projects_list()
+    assert isinstance(result, str)
+    assert "You have access to" in result
+
+def test_get_work_items(ado_api_wrapper):
+    args = GetAdoWorkItemsArgs(
+        resolved_after="2023-01-01",
+        updated_after="2023-01-01",
+        created_after="2023-01-01",
+    )
+    result = ado_api_wrapper.get_work_items(
+        resolved_after=args.resolved_after,
+        updated_after=args.updated_after,
+        created_after=args.created_after,
+        area=args.area,
+        project_keys=args.project_keys
+    )
+    assert isinstance(result, str)
+    assert f"Work items for {ado_api_wrapper.project_keys} have been successfully retrieved and saved to the bucket" in result
+
+@pytest.mark.asyncio
+async def test_get_commits(ado_api_wrapper):
+    args = AdoCommitsArgs(
+        since_date="2023-01-01",
+    )
+
+    result = await ado_api_wrapper.get_commits(since_date=args.since_date)
+    # breakpoint()
+    assert isinstance(result, str)
+    assert f"Commits for {ado_api_wrapper.project_keys} have been successfully" in result
+
+
+def test_get_merge_requests(ado_api_wrapper):
+    args = AdoCommitsArgs(
+        since_date="2023-01-01",
+    )
+
+    result = ado_api_wrapper.get_merge_requests(since_date=args.since_date)
+    assert isinstance(result, str)
+    assert f"Pull requests for {ado_api_wrapper.project_keys} have been successfully retrieved" in result
+
+def test_get_pipelines_runs(ado_api_wrapper):
+    result = ado_api_wrapper.get_pipelines_runs()
+    assert isinstance(result, str)
+    assert f"Pipeline runs for {ado_api_wrapper.project_keys} have been successfully retrieved " in result

tests/test_github_analysis.py

@@ -0,0 +1,87 @@
+import os
+from unittest.mock import patch
+import pytest
+from dotenv import load_dotenv
+
+from elitea_analyse.github.github_org import GitHubGetOrgLvl
+from ..alita_sdk.community.analysis.github_analyse.api_wrapper import (
+  GetGithubCommitsFromReposArgs, GetGithubRepositoriesListArgs, GitHubAnalyseWrapper
+)
+
+from ..alita_sdk.clients.client import AlitaClient
+from ..alita_sdk.tools.artifact import ArtifactWrapper
+from ..alita_sdk.community.utils import check_schema
+
+
+# Load environment variables from .env file
+load_dotenv()
+
+
+@pytest.fixture
+def github_api_wrapper():
+    base_url = os.getenv("DEPLOYMENT_URL")
+    project_id = os.getenv("PROJECT_ID")
+    api_key = os.getenv("API_KEY")
+
+    if not base_url or not project_id or not api_key:
+        raise ValueError("Environment variables DEPLOYMENT_URL, PROJECT_ID, and API_KEY must be set.")
+
+    client = AlitaClient(
+        base_url=base_url,
+        project_id=int(project_id),
+        auth_token=api_key,
+    )
+
+    artifacts_wrapper = ArtifactWrapper(
+        client=client, bucket=os.getenv("ARTIFACT_BUCKET_PATH", "analyse-gitlab")
+    )
+    check_schema(artifacts_wrapper)
+
+    owner = os.getenv("GITHUB_OWNER")
+    token = os.getenv("GITHUB_TOKEN")
+    if not owner or not token:
+        raise ValueError("GitHub owner and token are required")
+
+    git = GitHubGetOrgLvl(owner, token)
+
+    github_wrapper = GitHubAnalyseWrapper(
+        artifacts_wrapper=artifacts_wrapper,
+        repos=os.getenv("GITHUB_REPOS", ""),
+        git=git
+    )
+
+    check_schema(github_wrapper)
+
+    return github_wrapper
+
+def test_get_commits_from_repos(github_api_wrapper):
+    args = GetGithubCommitsFromReposArgs(since_after="2025-06-01")
+    result = github_api_wrapper.get_commits_from_repos(since_after=args.since_after)
+    assert isinstance(result, str)
+    assert f"GitHub commits data for {github_api_wrapper.repos} saved" in result
+
+def test_get_pull_requests_from_repos(github_api_wrapper):
+    args = GetGithubCommitsFromReposArgs(since_after="2025-06-01")
+    result = github_api_wrapper.get_pull_requests_from_repos(since_after=args.since_after)
+    assert isinstance(result, str)
+    assert "GitHub pull requests data saved" in result
+
+def test_get_repositories_list(github_api_wrapper):
+    with patch(
+        "src.alita_sdk.community.analysis.github_analyse.api_wrapper.dispatch_custom_event",
+        return_value=None,
+    ):
+        args = GetGithubRepositoriesListArgs(pushed_after="2025-06-01")
+        result = github_api_wrapper.get_repositories_list(pushed_after=args.pushed_after)
+        assert isinstance(result, str)
+        assert "GitHub repositories list saved" in result
+
+def test_get_repositories_extended_data(github_api_wrapper):
+    with patch(
+        "src.alita_sdk.community.analysis.github_analyse.api_wrapper.dispatch_custom_event",
+        return_value=None,
+    ):
+        args = GetGithubRepositoriesListArgs(pushed_after="2025-06-01")
+        result = github_api_wrapper.get_repositories_extended_data(pushed_after=args.pushed_after)
+        assert isinstance(result, str)
+        assert "Extended repository info that you have access saved" in result

tests/test_gitlab_analysis.py

@@ -0,0 +1,96 @@
+import os
+import pytest
+from dotenv import load_dotenv
+
+from elitea_analyse.git.git_search import GitLabV4Search
+
+
+from ..alita_sdk.community.analysis.gitlab_analyse.api_wrapper import (
+    GitLabProjectsListArgs,
+    GitLabCommitsArgs, GitLabAnalyseWrapper
+)
+
+from ..alita_sdk.clients.client import AlitaClient
+from ..alita_sdk.tools.artifact import ArtifactWrapper
+from ..alita_sdk.community.utils import check_schema
+
+
+# Load environment variables from .env file
+load_dotenv()
+
+
+@pytest.fixture
+def gitlab_api_wrapper():
+    base_url = os.getenv("DEPLOYMENT_URL")
+    project_id = os.getenv("PROJECT_ID")
+    api_key = os.getenv("API_KEY")
+
+    if not base_url or not project_id or not api_key:
+        raise ValueError("Environment variables DEPLOYMENT_URL, PROJECT_ID, and API_KEY must be set.")
+
+    client = AlitaClient(
+        base_url=base_url,
+        project_id=int(project_id),
+        auth_token=api_key,
+    )
+
+    artifacts_wrapper = ArtifactWrapper(
+        client=client, bucket=os.getenv("ARTIFACT_BUCKET_PATH", "analyse-gitlab")
+    )
+    check_schema(artifacts_wrapper)
+
+    gitlab_url = os.getenv("GITLAB_URL")
+    gitlab_token = os.getenv("GITLAB_TOKEN")
+
+    if not gitlab_url or not gitlab_token:
+        raise ValueError("Environment variables GITLAB_URL and GITLAB_TOKEN must be set.")
+
+    gitlab_search = GitLabV4Search(
+        url=gitlab_url,
+        default_branch_name=os.getenv("GITLAB_DEFAULT_BRANCH", "master"),
+        token=gitlab_token,
+    )
+
+    project_keys = os.getenv("GITLAB_JIRA_PROJECTS", "project1,project2")
+    project_ids = os.getenv("GITLAB_PROJECTS_IDS", "123")
+
+    gitlab_analyse_wrapper = GitLabAnalyseWrapper(
+        artifacts_wrapper=artifacts_wrapper,
+        jira_project_keys=project_keys,
+        project_ids=project_ids,
+        gitlab_search=gitlab_search,
+    )
+    check_schema(gitlab_analyse_wrapper)
+
+    return gitlab_analyse_wrapper
+
+def test_get_gitlab_projects_list(gitlab_api_wrapper):
+    args = GitLabProjectsListArgs(date="2025-05-01")
+    result = gitlab_api_wrapper.get_gitlab_projects_list(date=args.date)
+    assert isinstance(result, str)
+    assert "You have access to" in result
+
+def test_get_gitlab_projects_that_in_jira(gitlab_api_wrapper):
+    result = gitlab_api_wrapper.get_gitlab_projects_that_in_jira()
+    assert isinstance(result, str)
+    assert "GitLab projects that match Jira project names." in result
+
+def test_get_gitlab_commits(gitlab_api_wrapper):
+    project_ids = os.getenv("GITLAB_PROJECTS_IDS")
+    if not project_ids:
+        raise ValueError("Environment variable GITLAB_PROJECTS_IDS must be set.")
+
+    args = GitLabCommitsArgs(project_ids=project_ids, since_date="2010-01-01")
+    result = gitlab_api_wrapper.get_gitlab_commits(
+        project_ids=args.project_ids, since_date=args.since_date
+    )
+    assert isinstance(result, str)
+    assert f"Commits data for project {gitlab_api_wrapper.project_ids} has been saved. " in result
+
+
+def test_get_gitlab_merge_requests(gitlab_api_wrapper):
+    since_date = "2010-01-01"
+    args = GitLabCommitsArgs(since_date=since_date)
+    result = gitlab_api_wrapper.get_gitlab_merge_requests(project_ids=args.project_ids, since_date=args.since_date)
+    assert isinstance(result, str)
+    assert f"There are no merge requests in the project {gitlab_api_wrapper.project_ids} created after {since_date}"

tests/test_jira_analysis.py

@@ -18,13 +18,24 @@ load_dotenv()
 
 
 @pytest.fixture
-def eda_api_wrapper():
+def jira_api_wrapper():
+    base_url = os.getenv("DEPLOYMENT_URL")
+    project_id = os.getenv("PROJECT_ID")
+    api_key = os.getenv("API_KEY")
+    artifact_bucket_path = os.getenv("ARTIFACT_BUCKET_PATH", "analyse-jira")
+
+    if not base_url or not project_id or not api_key:
+        raise ValueError("Environment variables DEPLOYMENT_URL, PROJECT_ID, and API_KEY must be set.")
+
     client = AlitaClient(
-        base_url=os.getenv("DEPLOYMENT_URL"),
-        project_id=int(os.getenv("PROJECT_ID")),
-        auth_token=os.getenv("API_KEY"),
+        base_url=base_url,
+        project_id=int(project_id),
+        auth_token=api_key,
     )
 
+    artifacts_wrapper = ArtifactWrapper(client=client, bucket=artifact_bucket_path)
+    check_schema(artifacts_wrapper)
+
     jira_credentials = {
         "username": os.getenv("JIRA_USER"),
         "base_url": os.getenv("JIRA_SERVER"),
@@ -33,24 +44,22 @@ def eda_api_wrapper():
         "verify_ssl": False,
     }
     jira = connect_to_jira(credentials=jira_credentials)
+    if not jira:
+        raise ValueError("Failed to connect to Jira. Please check your credentials.")
 
-    artifacts_wrapper = ArtifactWrapper(
-        client=client, bucket=os.getenv("ARTIFACT_BUCKET_PATH")
-    )
-
-    check_schema(artifacts_wrapper)
-    eda_wrapper = JiraAnalyseWrapper(
+    jira_wrapper = JiraAnalyseWrapper(
         artifacts_wrapper=artifacts_wrapper,
         jira=jira,
-        closed_status=os.getenv("JIRA_CLOSED_STATUS"),
-        defects_name=os.getenv("JIRA_DEFECTS_NAME"),
+        closed_status=os.getenv("JIRA_CLOSED_STATUS", "Done"),
+        defects_name=os.getenv("JIRA_DEFECTS_NAME", "Defect"),
         custom_fields={"team": "", "defects_environment": ""},
+        project_keys=os.getenv("JIRA_PROJECT", "CARRIER"),
     )
-    check_schema(eda_wrapper)
-    return eda_wrapper
+    check_schema(jira_wrapper)
+    return jira_wrapper
 
 
-def test_get_number_of_all_issues(eda_api_wrapper):
+def test_get_number_of_all_issues(jira_api_wrapper):
     with patch(
         "src.alita_sdk.community.analysis.jira_analyse.api_wrapper.dispatch_custom_event",
         return_value=None,
@@ -58,27 +67,14 @@ def test_get_number_of_all_issues(eda_api_wrapper):
         args = GetJiraFieldsArgs(
             project_keys=os.getenv("JIRA_PROJECT"), after_date="2025-01-01"
         )
-        result = eda_api_wrapper.get_number_off_all_issues(
+        result = jira_api_wrapper.get_number_off_all_issues(
             args.project_keys, args.after_date
         )
         assert "projects" in result
         assert "projects_summary" in result
 
 
-def test_get_all_jira_fields(eda_api_wrapper):
-    with patch(
-        "src.alita_sdk.community.analysis.jira_analyse.api_wrapper.dispatch_custom_event",
-        return_value=None,
-    ):
-        args = GetJiraFieldsArgs(
-            project_keys=os.getenv("JIRA_PROJECT"), after_date="2025-01-01"
-        )
-        result = eda_api_wrapper.get_all_jira_fields(args.project_keys, args.after_date)
-        assert "overall_stat" in result
-        assert "issue_types_stat" in result
-
-
-def test_get_jira_issues(eda_api_wrapper):
+def test_get_jira_issues(jira_api_wrapper):
     with patch(
         "src.alita_sdk.community.analysis.jira_analyse.api_wrapper.dispatch_custom_event",
         return_value=None,
@@ -91,12 +87,12 @@ def test_get_jira_issues(eda_api_wrapper):
             created_after="2025-01-01",
             add_filter="",
         )
-        result = eda_api_wrapper.get_jira_issues(
-            args.project_keys,
-            args.closed_issues_based_on,
-            args.resolved_after,
-            args.updated_after,
-            args.created_after,
-            args.add_filter,
+        result = jira_api_wrapper.get_jira_issues(
+            closed_issues_based_on=args.closed_issues_based_on,
+            resolved_after=args.resolved_after,
+            updated_after=args.updated_after,
+            created_after=args.created_after,
+            project_keys=args.project_keys,
+            add_filter=args.add_filter,
         )
         assert "Data has been extracted successfully." in result