def test_teardown_build_runs_teardown(self):
job_config = JobConfig('name', 'setup', 'teardown', 'command', 'atomizer', 10, 10)
project_type = ProjectType()
project_type.job_config = MagicMock(return_value=job_config)
project_type.execute_command_in_project = MagicMock(return_value=('', 0))
project_type.teardown_build()
project_type.execute_command_in_project.assert_called_with('teardown', timeout=None)
def test_teardown_build_runs_teardown(self):
job_config = JobConfig("name", "setup", "teardown", "command", "atomizer", 10, 10)
project_type = ProjectType()
project_type.job_config = MagicMock(return_value=job_config)
project_type.execute_command_in_project = MagicMock(return_value=("", 0))
project_type.teardown_build()
project_type.execute_command_in_project.assert_called_with("teardown", timeout=None)
def test_teardown_build_runs_teardown(self):
job_config = JobConfig('name', 'setup', 'teardown', 'command', 'atomizer', 10)
project_type = ProjectType()
project_type.job_config = MagicMock(return_value=job_config)
project_type.execute_command_in_project = MagicMock(return_value=('', 0))
project_type.teardown_build()
project_type.execute_command_in_project.assert_called_with('teardown')
def test_teardown_build_executes_teardown_command(self, expected_timeout):
project_type = ProjectType()
mock_execute = MagicMock(return_value=('fake output', 0))
project_type.execute_command_in_project = mock_execute
project_type.job_config = MagicMock()
if expected_timeout:
project_type.teardown_build(timeout=expected_timeout)
else:
project_type.teardown_build()
mock_execute.assert_called_once_with(ANY, timeout=expected_timeout)
def test_teardown_build_executes_teardown_command(self, expected_timeout):
project_type = ProjectType()
mock_execute = MagicMock(return_value=("fake output", 0))
project_type.execute_command_in_project = mock_execute
project_type.job_config = MagicMock()
if expected_timeout:
project_type.teardown_build(timeout=expected_timeout)
else:
project_type.teardown_build()
mock_execute.assert_called_once_with(ANY, timeout=expected_timeout)
def test_job_config_uses_passed_in_config_instead_of_clusterrunner_yaml(self):
config_dict = {
'commands': ['shell command 1', 'shell command 2;'],
'atomizers': [{'TESTPATH': 'atomizer command'}],
'max_executors': 100,
'max_executors_per_slave': 2,
}
project_type = ProjectType(config=config_dict, job_name='some_job_name')
job_config = project_type.job_config()
self.assertEquals(job_config.name, 'some_job_name')
self.assertEquals(job_config.command, 'shell command 1 && shell command 2')
self.assertEquals(job_config.max_executors, 100)
self.assertEquals(job_config.max_executors_per_slave, 2)
