def test_create_with_lock(self): command = CreateJobCommand() jobkey = AuroraJobKey("cluster", "role", "env", "job") mock_options = mock_verb_options(command) mock_options.jobspec = jobkey mock_options.config_file = "/tmp/whatever" fake_context = FakeAuroraCommandContext() fake_context.set_options(mock_options) mock_config = create_autospec(spec=AuroraConfig, spec_set=True, instance=True) mock_job_config = Mock() mock_job_config.has_cron_schedule.return_value = False mock_config.raw.return_value = mock_job_config fake_context.get_job_config = Mock(return_value=mock_config) mock_api = fake_context.get_api("test") mock_api.create_job.return_value = AuroraClientCommandTest.create_blank_response( ResponseCode.LOCK_ERROR, "Error.") with pytest.raises(Context.CommandError): command.execute(fake_context) mock_api.create_job.assert_called_once_with(mock_config) self.assert_lock_message(fake_context)
def test_create_with_lock(self): command = CreateJobCommand() jobkey = AuroraJobKey("cluster", "role", "env", "job") mock_options = mock_verb_options(command) mock_options.jobspec = jobkey mock_options.config_file = "/tmp/whatever" fake_context = FakeAuroraCommandContext() fake_context.set_options(mock_options) mock_config = create_autospec(spec=AuroraConfig, spec_set=True, instance=True) mock_job_config = Mock() mock_job_config.has_cron_schedule.return_value = False mock_config.raw.return_value = mock_job_config fake_context.get_job_config = Mock(return_value=mock_config) mock_api = fake_context.get_api("test") mock_api.create_job.return_value = AuroraClientCommandTest.create_blank_response( ResponseCode.JOB_UPDATING_ERROR, "Error.") with pytest.raises(Context.CommandError): command.execute(fake_context) mock_api.create_job.assert_called_once_with(mock_config) self.assert_lock_message(fake_context)