mirror of
https://github.com/saymrwulf/pytorch.git
synced 2026-05-14 20:57:59 +00:00
19bdc4f007
### Description Renamed the file the retrieves the latest green commit for viable/strict promotion. The previous name, `print_latest_commit` did not relate to its actual function. ### Testing All of the test cases in `test_fetch_latest_green_commit` pass, nothing else should be affected Pull Request resolved: https://github.com/pytorch/pytorch/pull/82414 Approved by: https://github.com/clee2000
101 lines
4.9 KiB
Python
101 lines
4.9 KiB
Python
from unittest import TestCase, main, mock
|
|
from typing import Any, List, Dict
|
|
from fetch_latest_green_commit import isGreen, WorkflowCheck
|
|
|
|
workflowNames = [
|
|
"pull",
|
|
"trunk",
|
|
"Lint",
|
|
"linux-binary-libtorch-pre-cxx11",
|
|
"android-tests",
|
|
"windows-binary-wheel",
|
|
"periodic",
|
|
"docker-release-builds",
|
|
"nightly",
|
|
"pr-labels",
|
|
"Close stale pull requests",
|
|
"Update S3 HTML indices for download.pytorch.org",
|
|
"Create Release"
|
|
]
|
|
|
|
def set_workflow_job_status(workflow: List[Dict[str, Any]], name: str, status: str) -> List[Dict[str, Any]]:
|
|
for check in workflow:
|
|
if check['workflowName'] == name:
|
|
check['conclusion'] = status
|
|
return workflow
|
|
|
|
class TestChecks:
|
|
def make_test_checks(self) -> List[Dict[str, Any]]:
|
|
workflow_checks = []
|
|
for i in range(len(workflowNames)):
|
|
workflow_checks.append(WorkflowCheck(
|
|
workflowName=workflowNames[i],
|
|
name="test/job",
|
|
jobName="job",
|
|
conclusion="success",
|
|
)._asdict())
|
|
return workflow_checks
|
|
|
|
class TestPrintCommits(TestCase):
|
|
@mock.patch('fetch_latest_green_commit.get_commit_results', return_value=TestChecks().make_test_checks())
|
|
def test_all_successful(self, mock_get_commit_results: Any) -> None:
|
|
"Test with workflows are successful"
|
|
workflow_checks = mock_get_commit_results()
|
|
self.assertTrue(isGreen("sha", workflow_checks)[0])
|
|
|
|
@mock.patch('fetch_latest_green_commit.get_commit_results', return_value=TestChecks().make_test_checks())
|
|
def test_necessary_successful(self, mock_get_commit_results: Any) -> None:
|
|
"Test with necessary workflows are successful"
|
|
workflow_checks = mock_get_commit_results()
|
|
workflow_checks = set_workflow_job_status(workflow_checks, workflowNames[8], "failed")
|
|
workflow_checks = set_workflow_job_status(workflow_checks, workflowNames[9], "failed")
|
|
workflow_checks = set_workflow_job_status(workflow_checks, workflowNames[10], "failed")
|
|
workflow_checks = set_workflow_job_status(workflow_checks, workflowNames[11], "failed")
|
|
workflow_checks = set_workflow_job_status(workflow_checks, workflowNames[12], "failed")
|
|
self.assertTrue(isGreen("sha", workflow_checks)[0])
|
|
|
|
@mock.patch('fetch_latest_green_commit.get_commit_results', return_value=TestChecks().make_test_checks())
|
|
def test_necessary_skipped(self, mock_get_commit_results: Any) -> None:
|
|
"Test with necessary job (ex: pull) skipped"
|
|
workflow_checks = mock_get_commit_results()
|
|
workflow_checks = set_workflow_job_status(workflow_checks, "pull", "skipped")
|
|
result = isGreen("sha", workflow_checks)
|
|
self.assertTrue(result[0])
|
|
|
|
@mock.patch('fetch_latest_green_commit.get_commit_results', return_value=TestChecks().make_test_checks())
|
|
def test_skippable_skipped(self, mock_get_commit_results: Any) -> None:
|
|
"Test with skippable jobs (periodic and docker-release-builds skipped"
|
|
workflow_checks = mock_get_commit_results()
|
|
workflow_checks = set_workflow_job_status(workflow_checks, "periodic", "skipped")
|
|
workflow_checks = set_workflow_job_status(workflow_checks, "docker-release-builds", "skipped")
|
|
self.assertTrue(isGreen("sha", workflow_checks))
|
|
|
|
@mock.patch('fetch_latest_green_commit.get_commit_results', return_value=TestChecks().make_test_checks())
|
|
def test_necessary_failed(self, mock_get_commit_results: Any) -> None:
|
|
"Test with necessary job (ex: Lint) failed"
|
|
workflow_checks = mock_get_commit_results()
|
|
workflow_checks = set_workflow_job_status(workflow_checks, "Lint", "failed")
|
|
result = isGreen("sha", workflow_checks)
|
|
self.assertFalse(result[0])
|
|
self.assertEqual(result[1], "Lint checks were not successful")
|
|
|
|
@mock.patch('fetch_latest_green_commit.get_commit_results', return_value=TestChecks().make_test_checks())
|
|
def test_skippable_failed(self, mock_get_commit_results: Any) -> None:
|
|
"Test with skippable job (ex: docker-release-builds) failing"
|
|
workflow_checks = mock_get_commit_results()
|
|
workflow_checks = set_workflow_job_status(workflow_checks, "periodic", "skipped")
|
|
workflow_checks = set_workflow_job_status(workflow_checks, "docker-release-builds", "failed")
|
|
result = isGreen("sha", workflow_checks)
|
|
self.assertFalse(result[0])
|
|
self.assertEqual(result[1], "docker-release-builds checks were not successful")
|
|
|
|
@mock.patch('fetch_latest_green_commit.get_commit_results', return_value={})
|
|
def test_no_workflows(self, mock_get_commit_results: Any) -> None:
|
|
"Test with missing workflows"
|
|
workflow_checks = mock_get_commit_results()
|
|
result = isGreen("sha", workflow_checks)
|
|
self.assertFalse(result[0])
|
|
self.assertEqual(result[1], "missing required workflows: pull, trunk, lint, linux-binary, windows-binary")
|
|
|
|
if __name__ == "__main__":
|
|
main()
|