Skip to content

Commit

Permalink
Add is_completed_checker function
Browse files Browse the repository at this point in the history
Signed-off-by: hoangtungdinh <[email protected]>
  • Loading branch information
hoangtungdinh committed Sep 11, 2024
1 parent 7c1825e commit 1b5ec1c
Show file tree
Hide file tree
Showing 2 changed files with 57 additions and 0 deletions.
12 changes: 12 additions & 0 deletions qc_baselib/result.py
Original file line number Diff line number Diff line change
Expand Up @@ -595,3 +595,15 @@ def all_checkers_completed_without_issue(self, check_id_set: Set[str]) -> bool:
result = result and checker_result

return result

def is_completed_checker(self, checker_id: str) -> bool:
"""
Check if the status of a checker is COMPLETED.
Return False if the checker is not found.
"""
for bundle in self._report_results.checker_bundles:
for checker in bundle.checkers:
if checker.checker_id == checker_id:
return checker.status == StatusType.COMPLETED

return False
45 changes: 45 additions & 0 deletions tests/test_result.py
Original file line number Diff line number Diff line change
Expand Up @@ -920,3 +920,48 @@ def test_registration_without_summary() -> None:
)

assert True


def test_is_completed_checker() -> None:
result_report = Result()

result_report.register_checker_bundle(
name="TestBundle",
build_date="2024-05-31",
description="Example checker bundle",
version="0.0.1",
)

assert result_report.is_completed_checker("TestChecker") == False

result_report.register_checker(
checker_bundle_name="TestBundle",
checker_id="TestChecker",
description="",
)

assert result_report.is_completed_checker("TestChecker") == False

result_report.set_checker_status(
checker_bundle_name="TestBundle",
checker_id="TestChecker",
status=StatusType.COMPLETED,
)

assert result_report.is_completed_checker("TestChecker") == True

result_report.set_checker_status(
checker_bundle_name="TestBundle",
checker_id="TestChecker",
status=StatusType.SKIPPED,
)

assert result_report.is_completed_checker("TestChecker") == False

result_report.set_checker_status(
checker_bundle_name="TestBundle",
checker_id="TestChecker",
status=StatusType.ERROR,
)

assert result_report.is_completed_checker("TestChecker") == False

0 comments on commit 1b5ec1c

Please sign in to comment.