def test_clear(): """Should clear jobs container""" arc = AutoRegressionConfiguration() job1 = MagicMock() job1.name = "test" arc.add_job(job1) job2 = MagicMock() job2.name = "test2" arc.add_job(job2) assert arc.get_num_jobs() == 2 arc.clear() assert arc.get_num_jobs() == 0
def test_autogression_configuration__add_job(): """Should add job to container""" arc = AutoRegressionConfiguration() job1 = MagicMock() job1.name = "test" arc.add_job(job1) job2 = MagicMock() job2.name = "test2" arc.add_job(job2) assert arc.get_num_jobs() == 2
def test_remove_job(): """Should remove job if job exists in container""" arc = AutoRegressionConfiguration() job1 = MagicMock() job1.name = "Job1" arc.add_job(job1) job2 = MagicMock() job2.name = "Job2" arc.add_job(job2) arc.remove_job(job1) assert arc.get_num_jobs() == 1
def test_reconfigure_jobs(): """Should reconfigure with a list of jobs""" job1 = MagicMock() job1.name = "Job1" arc = AutoRegressionConfiguration() arc.add_job(job1) job2 = MagicMock() job2.name = "Job2" job3 = MagicMock() job3.name = "Job3" arc.reconfigure_jobs(jobs=[job2, job3]) assert arc.get_num_jobs() == 2 with pytest.raises(InvalidParameter): arc.get_job("Job1") assert arc.get_job("Job2") == job2