def test_get_docker_run_cmd(
mock_getegid,
mock_geteuid,
):
mock_geteuid.return_value = 1234
mock_getegid.return_value = 100
container_name = 'fake_name'
volumes = ['v1:v1:rw', 'v2:v2:rw']
env = {'k1': 'v1', 'k2': 'v2'}
docker_img = 'fake-registry/fake-service'
docker_cmd = 'pyspark'
actual = get_docker_run_cmd(
container_name,
volumes,
env,
docker_img,
docker_cmd,
)
assert actual[5:] == [
'--user=1234:100',
'--name=fake_name',
'--env', 'k1=v1', '--env', 'k2=v2',
'--volume=v1:v1:rw', '--volume=v2:v2:rw',
'fake-registry/fake-service',
'sh', '-c', 'pyspark', {},
]
def test_get_docker_run_cmd(mock_getegid, mock_geteuid):
mock_geteuid.return_value = 1234
mock_getegid.return_value = 100
container_name = "fake_name"
volumes = ["v1:v1:rw", "v2:v2:rw"]
env = {"k1": "v1", "k2": "v2"}
docker_img = "fake-registry/fake-service"
docker_cmd = "pyspark"
nvidia = False
actual = get_docker_run_cmd(container_name, volumes, env, docker_img,
docker_cmd, nvidia)
assert actual[5:] == [
"--user=1234:100",
"--name=fake_name",
"--env",
"k1=v1",
"--env",
"k2=v2",
"--volume=v1:v1:rw",
"--volume=v2:v2:rw",
"fake-registry/fake-service",
"sh",
"-c",
"pyspark",
{},
]
def test_get_docker_run_cmd():
container_name = 'fake_name'
volumes = ['v1:v1:rw', 'v2:v2:rw']
env = {'k1': 'v1', 'k2': 'v2'}
docker_img = 'fake-registry/fake-service'
docker_cmd = 'pyspark'
actual = get_docker_run_cmd(
container_name,
volumes,
env,
docker_img,
docker_cmd,
)
assert actual[6:] == [
'--name=fake_name',
'--env',
'k1=v1',
'--env',
'k2=v2',
'--volume=v1:v1:rw',
'--volume=v2:v2:rw',
'fake-registry/fake-service',
'pyspark',
]