def test_simple_docker(self):
t = DockerRunTask(
command="env",
image="ubuntu:latest",
override={DockerEngineConfig.environment: {"UNIT": "TEST"}},
)
t.dbnd_run()
def test_aws_batch_hello_world_task(self):
t = DockerRunTask(
image=None,
command="not in use",
override={
AwsBatchConfig.job_definition: "hello-world",
AwsBatchConfig.job_queue: "first-run-job-queue",
AwsBatchConfig.max_retries: 5,
},
task_env=CloudType.aws,
)
t.dbnd_run()
def submit_to_engine_task(self, env, task_name, args, interactive=True):
from dbnd_docker.docker.docker_task import DockerRunTask
submit_task = DockerRunTask(
task_name=task_name,
command=subprocess.list2cmdline(args),
image=self.full_image,
docker_engine=self,
task_is_system=True,
)
return submit_task
def submit_to_engine_task(self, env, task_name, args, interactive=True):
docker_engine = self
if not interactive:
docker_engine = docker_engine.clone(auto_remove=True, detach_run=True)
return DockerRunTask(
task_name=task_name,
command=subprocess.list2cmdline(args),
image=self.full_image,
docker_engine=docker_engine,
task_is_system=True,
)