def test_action_run_from_state_mesos(self): state_data = self.action_state_data state_data['executor'] = ExecutorTypes.mesos state_data['cpus'] = 2 state_data['mem'] = 200 state_data['constraints'] = [['pool', 'LIKE', 'default']] state_data['docker_image'] = 'fake-docker.com:400/image' state_data['docker_parameters'] = [{'key': 'test', 'value': 123}] state_data['env'] = {'TESTING': 'true'} state_data['extra_volumes'] = [{'path': '/tmp'}] action_run = ActionRunFactory.action_run_from_state( self.job_run, state_data, ) assert_equal(action_run.job_run_id, state_data['job_run_id']) assert_equal(action_run.cpus, state_data['cpus']) assert_equal(action_run.mem, state_data['mem']) assert_equal(action_run.constraints, state_data['constraints']) assert_equal(action_run.docker_image, state_data['docker_image']) assert_equal(action_run.docker_parameters, state_data['docker_parameters']) assert_equal(action_run.env, state_data['env']) assert_equal(action_run.extra_volumes, state_data['extra_volumes']) assert not action_run.is_cleanup assert_equal(action_run.__class__, MesosActionRun)
def test_action_run_from_state(self): state_data = self.action_state_data action_run = ActionRunFactory.action_run_from_state( self.job_run, state_data) assert_equal(action_run.job_run_id, state_data['job_run_id']) assert not action_run.is_cleanup