mirror of https://github.com/mongodb/mongo
63 lines
2.7 KiB
Python
63 lines
2.7 KiB
Python
"""Unit tests for gen_task_validation.py"""
|
|
import unittest
|
|
from unittest.mock import MagicMock
|
|
|
|
import buildscripts.task_generation.gen_task_validation as under_test
|
|
|
|
# pylint: disable=missing-docstring,invalid-name,unused-argument,no-self-use,protected-access
|
|
|
|
|
|
class TestShouldTasksBeGenerated(unittest.TestCase):
|
|
def test_during_first_execution(self):
|
|
task_id = "task_id"
|
|
mock_evg_api = MagicMock()
|
|
mock_evg_api.task_by_id.return_value.execution = 0
|
|
validate_service = under_test.GenTaskValidationService(mock_evg_api)
|
|
|
|
self.assertTrue(validate_service.should_task_be_generated(task_id))
|
|
mock_evg_api.task_by_id.assert_called_with(task_id, fetch_all_executions=True)
|
|
|
|
def test_after_successful_execution(self):
|
|
task_id = "task_id"
|
|
mock_evg_api = MagicMock()
|
|
task = mock_evg_api.task_by_id.return_value
|
|
task.execution = 1
|
|
task.get_execution.return_value.is_success.return_value = True
|
|
validate_service = under_test.GenTaskValidationService(mock_evg_api)
|
|
|
|
self.assertFalse(validate_service.should_task_be_generated(task_id))
|
|
mock_evg_api.task_by_id.assert_called_with(task_id, fetch_all_executions=True)
|
|
|
|
def test_after_multiple_successful_execution(self):
|
|
task_id = "task_id"
|
|
mock_evg_api = MagicMock()
|
|
task = mock_evg_api.task_by_id.return_value
|
|
task.execution = 5
|
|
task.get_execution.return_value.is_success.return_value = True
|
|
validate_service = under_test.GenTaskValidationService(mock_evg_api)
|
|
|
|
self.assertFalse(validate_service.should_task_be_generated(task_id))
|
|
mock_evg_api.task_by_id.assert_called_with(task_id, fetch_all_executions=True)
|
|
|
|
def test_after_failed_execution(self):
|
|
mock_evg_api = MagicMock()
|
|
task_id = "task_id"
|
|
task = mock_evg_api.task_by_id.return_value
|
|
task.execution = 1
|
|
task.get_execution.return_value.is_success.return_value = False
|
|
validate_service = under_test.GenTaskValidationService(mock_evg_api)
|
|
|
|
self.assertTrue(validate_service.should_task_be_generated(task_id))
|
|
mock_evg_api.task_by_id.assert_called_with(task_id, fetch_all_executions=True)
|
|
|
|
def test_after_multiple_failed_execution(self):
|
|
mock_evg_api = MagicMock()
|
|
task_id = "task_id"
|
|
task = mock_evg_api.task_by_id.return_value
|
|
task.execution = 5
|
|
task.get_execution.return_value.is_success.return_value = False
|
|
validate_service = under_test.GenTaskValidationService(mock_evg_api)
|
|
|
|
self.assertTrue(validate_service.should_task_be_generated(task_id))
|
|
mock_evg_api.task_by_id.assert_called_with(task_id, fetch_all_executions=True)
|