Polarion script: add tests (#27)

* polarion script: add tests

* polarion script: add tests

* Fix polarion config test

---------

Co-authored-by: Debarati Basu-Nag <[email protected]>

Author: myakove

apps/polarion/polarion_utils.py

@@ -23,9 +23,13 @@ def git_diff_lines():
     return diff
 
 
-def validate_polarion_requirements(polarion_test_ids, polarion_project_id):
+def validate_polarion_requirements(
+    polarion_test_ids,
+    polarion_project_id,
+):
+    from pylero.work_item import TestCase, Requirement
+
     tests_with_missing_requirements = []
-    from pylero.work_item import Requirement, TestCase
 
     for _id in polarion_test_ids:
         has_req = False

apps/polarion/polarion_verify_tc_requirements.py

@@ -3,14 +3,13 @@
 from simple_logger.logger import get_logger
 import os
 import click
-
 from apps.polarion.polarion_utils import (
     git_diff_lines,
     validate_polarion_requirements,
 )
 from apps.utils import get_util_config
 
-LOGGER = get_logger(name=__name__)
+LOGGER = get_logger(name="polarion-verify-tc-requirements")
 
 
 @click.command()
@@ -25,11 +24,14 @@
 def has_verify(config_file_path, project_id, verbosity):
     if verbosity:
         LOGGER.setLevel(logging.DEBUG)
+
     polarion_project_id = project_id or get_util_config(
-        util_name="pyutils-polarion-verify-tc-requirements", config_file_path=config_file_path
+        util_name="pyutils-polarion-verify-tc-requirements",
+        config_file_path=config_file_path,
     ).get("project_id")
+
     if not polarion_project_id:
-        click.echo("Polarion project id must be passed via config file or command line")
+        LOGGER.error("Polarion project id must be passed via config file or command line")
         raise click.Abort()
 
     if added_ids := re.findall(
@@ -39,7 +41,8 @@ def has_verify(config_file_path, project_id, verbosity):
     ):
         LOGGER.debug(f"Checking following ids: {added_ids}")
         if tests_with_missing_requirements := validate_polarion_requirements(
-            polarion_test_ids=added_ids, polarion_project_id=polarion_project_id
+            polarion_test_ids=added_ids,
+            polarion_project_id=polarion_project_id,
         ):
             click.echo(f"TestCases with missing requirement: {tests_with_missing_requirements}")
             raise click.Abort()

poetry.lock

@@ -62,6 +62,7 @@ pattern = "((?P<epoch>\\d+)!)?(?P<base>\\d+(\\.\\d+)*)"
 
 
 
+
 [tool.poetry.group.test.dependencies]
 pytest = "^8.0.0"
 pytest-cov = "^4.1.0"

pyproject.toml

@@ -0,0 +1,30 @@
+import shlex
+import subprocess
+from pyhelper_utils.shell import run_command
+
+BASE_COMMAND = "poetry run python apps/polarion/polarion_verify_tc_requirements.py"
+
+
+def test_missing_project_id():
+    rc, _, err = run_command(
+        command=shlex.split(BASE_COMMAND),
+        verify_stderr=False,
+        check=False,
+        capture_output=False,
+        stderr=subprocess.PIPE,
+    )
+    assert "Polarion project id must be passed via config file or command line" in err
+    assert not rc
+
+
+def test_project_id_from_config():
+    command = f"{BASE_COMMAND} --config-file-path=config.example.yaml"
+    rc, _, err = run_command(
+        command=shlex.split(command),
+        verify_stderr=False,
+        check=False,
+        capture_output=False,
+        stderr=subprocess.PIPE,
+    )
+    assert not err
+    assert rc