def test_command_local_grade_with_extra_args(run_container, utils, common):
args = argparse.Namespace()
args.dir = '/tmp'
args.imageId = 'myimageId'
args.args = ['extra', 'args']
common.mk_submission_volume_str.return_value = 'foo'
docker_mock = MagicMock()
docker_mock.create_container.return_value = {
"Id": "myContainerInstanceId",
}
docker_mock.inspect_image.return_value = {
"Config": {
"Entrypoint": ["command"],
}
}
utils.docker_client.return_value = docker_mock
grade.command_grade_local(args)
docker_mock.create_container.assert_called_with(
image='myimageId',
entrypoint=['command', 'extra', 'args'],
user='******',
host_config=docker.utils.create_host_config(
binds=['foo', ],
network_mode='none',
mem_limit='1g',
memswap_limit='1g',
),
)
run_container.assert_called_with(
docker_mock,
docker_mock.create_container.return_value,
args)
def test_command_local_grade_simple(docker_utils, run_container, utils,
common):
args = argparse.Namespace()
args.dir = '/tmp'
args.imageId = 'myimageId'
common.mk_submission_volume_str.return_value = 'foo'
docker_mock = MagicMock()
docker_mock.create_container.return_value = {
"Id": "myContainerInstanceId",
}
utils.docker_client.return_value = docker_mock
h_config = {'foo': 'bar'} # just some unique value
docker_utils.create_host_config.return_value = h_config
grade.command_grade_local(args)
docker_utils.create_host_config.assert_called_with(
binds=[
'foo',
],
network_mode='none',
mem_limit='1g',
memswap_limit='1g',
)
docker_mock.create_container.assert_called_with(
image='myimageId',
user='******',
host_config=h_config,
)
run_container.assert_called_with(docker_mock,
docker_mock.create_container.return_value,
args)
def test_command_local_grade_with_extra_args(run_container, utils, common):
args = argparse.Namespace()
args.dir = '/tmp'
args.imageId = 'myimageId'
args.args = ['extra', 'args']
common.mk_submission_volume_str.return_value = 'foo'
docker_mock = MagicMock()
docker_mock.create_container.return_value = {
"Id": "myContainerInstanceId",
}
docker_mock.inspect_image.return_value = {
"Config": {
"Entrypoint": ["command"],
}
}
utils.docker_client.return_value = docker_mock
grade.command_grade_local(args)
docker_mock.create_container.assert_called_with(
image='myimageId',
entrypoint=['command', 'extra', 'args'],
user='******',
host_config=docker.utils.create_host_config(
binds=[
'foo',
],
network_mode='none',
mem_limit='1g',
memswap_limit='1g',
),
)
run_container.assert_called_with(docker_mock,
docker_mock.create_container.return_value,
args)
def test_command_local_grade_simple(run_container, utils, common):
args = argparse.Namespace()
args.dir = '/tmp'
args.containerId = 'myContainerId'
common.mk_submission_volume_str.return_value = 'foo'
docker_mock = MagicMock()
docker_mock.create_container.return_value = {
"Id": "myContainerInstanceId",
}
utils.docker_client.return_value = docker_mock
grade.command_grade_local(args)
docker_mock.create_container.assert_called_with(
image='myContainerId',
user='******',
host_config=docker.utils.create_host_config(
binds=['foo', ],
network_mode='none',
mem_limit='1g',
memswap_limit='1g',
),
)
run_container.assert_called_with(
docker_mock,
docker_mock.create_container.return_value,
args)