diff --git a/app/main.py b/app/main.py index 150ce2e..b91a625 100644 --- a/app/main.py +++ b/app/main.py @@ -7,7 +7,7 @@ import sys from app.args import parse_args from app.print_table import print_table, generate_prefix_string from app.query_normalisation import is_ip_address, resolve_domain_name -from app.ip_info import ( # pragma: no cover +from app.ip_info import ( get_ip_information, get_autonomous_system_number, get_prefix_information, diff --git a/app/print_table.py b/app/print_table.py index a4626f9..cbb8a74 100644 --- a/app/print_table.py +++ b/app/print_table.py @@ -18,6 +18,6 @@ def generate_prefix_string(prefixes: list) -> Union[str, None]: return None -def print_table(table_data) -> None: +def print_table(table_data) -> None: # pragma: no cover """Print table generated by tabulate.""" print(tabulate(table_data)) diff --git a/sonar-project.properties b/sonar-project.properties index d470a33..4ffb8ad 100644 --- a/sonar-project.properties +++ b/sonar-project.properties @@ -6,3 +6,4 @@ sonar.python.xunit.reportPath=testresults.xml sonar.sources=app sonar.tests=tests sonar.exclusions=,.github/**,.gitignore,CODEOWNERS,CHANGELOG.md,LICENSE.md,README.md,renovate.json,requirements-dev.txt,requirements.txt +sonar.coverage.exclusions=app/_version.py,app/args.py,app/main.py diff --git a/tests/test_print_table.py b/tests/test_print_table.py new file mode 100644 index 0000000..7610486 --- /dev/null +++ b/tests/test_print_table.py @@ -0,0 +1,19 @@ +#!/usr/bin/env python3 + +"""MODULE: Provides test cases for app/print_table.py.""" + +from app.print_table import generate_prefix_string # pragma: no cover + + +def test_generate_prefix_string_small() -> None: + """TEST: Verifies if a small prefix list results in one line.""" + test_query = ["abc", "def"] + result = generate_prefix_string(prefixes=test_query) + assert result == "abc, def\n" + + +def test_generate_prefix_string_large() -> None: + """TEST: Verifies if a large prefix list results in multiple lines.""" + test_query = ["abc", "def", "ghi", "jkl", "mno", "pqr"] + result = generate_prefix_string(prefixes=test_query) + assert result == "abc, def, ghi, jkl\nmno, pqr\n"