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', ]