def test_build_run_for_mesos_action(self):
action = MagicMock(
name='theaction',
command="doit",
executor=ExecutorTypes.mesos,
cpus=10,
mem=500,
constraints=[['pool', 'LIKE', 'default']],
docker_image='fake-docker.com:400/image',
docker_parameters=[{
'key': 'test',
'value': 123
}],
env={'TESTING': 'true'},
extra_volumes=[{
'path': '/tmp'
}],
)
action_run = ActionRunFactory.build_run_for_action(
self.job_run,
action,
self.action_runner,
)
assert_equal(action_run.__class__, MesosActionRun)
assert_equal(action_run.cpus, action.cpus)
assert_equal(action_run.mem, action.mem)
assert_equal(action_run.constraints, action.constraints)
assert_equal(action_run.docker_image, action.docker_image)
assert_equal(action_run.docker_parameters, action.docker_parameters)
assert_equal(action_run.env, action.env)
assert_equal(action_run.extra_volumes, action.extra_volumes)
def test_build_run_for_action_with_node(self):
action = Turtle(name='theaction', is_cleanup=True, command="doit")
action_run = ActionRunFactory.build_run_for_action(self.job_run, action)
assert_equal(action_run.job_run_id, self.job_run.id)
assert_equal(action_run.node, action.node_pool.next.returns[0])
assert action_run.is_cleanup
assert_equal(action_run.action_name, action.name)
assert_equal(action_run.command, action.command)
def test_build_run_for_action_with_node(self):
action = Turtle(name='theaction', is_cleanup=True, command="doit")
action_run = ActionRunFactory.build_run_for_action(
self.job_run, action, self.action_runner)
assert_equal(action_run.job_run_id, self.job_run.id)
assert_equal(action_run.node, action.node_pool.next.returns[0])
assert action_run.is_cleanup
assert_equal(action_run.action_name, action.name)
assert_equal(action_run.command, action.command)
def test_build_run_for_action(self):
action = Turtle(
name='theaction', node_pool=None, is_cleanup=False, command="doit")
action_run = ActionRunFactory.build_run_for_action(self.job_run, action)
assert_equal(action_run.job_run_id, self.job_run.id)
assert_equal(action_run.node, self.job_run.node)
assert_equal(action_run.action_name, action.name)
assert not action_run.is_cleanup
assert_equal(action_run.command, action.command)
def test_build_run_for_ssh_action(self):
action = MagicMock(
name='theaction',
command="doit",
executor=ExecutorTypes.ssh,
)
action_run = ActionRunFactory.build_run_for_action(
self.job_run,
action,
self.action_runner,
)
assert_equal(action_run.__class__, SSHActionRun)
def test_build_run_for_action(self):
action = Turtle(name='theaction',
node_pool=None,
is_cleanup=False,
command="doit")
action_run = ActionRunFactory.build_run_for_action(
self.job_run, action, self.action_runner)
assert_equal(action_run.job_run_id, self.job_run.id)
assert_equal(action_run.node, self.job_run.node)
assert_equal(action_run.action_name, action.name)
assert not action_run.is_cleanup
assert_equal(action_run.command, action.command)
def test_build_run_for_action_with_node(self):
action = MagicMock(name='theaction', is_cleanup=True, command="doit")
action.node_pool = mock.create_autospec(node.NodePool)
action_run = ActionRunFactory.build_run_for_action(
self.job_run,
action,
self.action_runner,
)
assert_equal(action_run.job_run_id, self.job_run.id)
assert_equal(action_run.node, action.node_pool.next())
assert action_run.is_cleanup
assert_equal(action_run.action_name, action.name)
assert_equal(action_run.command, action.command)
