def test_config_with_docker_image():
image_name = 'some-image'
image_tag = 'some-tag'
job = convert_pystachio_to_thrift(
HELLO_WORLD(container=Mesos(image=DockerImage(name=image_name, tag=image_tag))))
assert job.taskConfig.container.mesos.image.appc is None
assert job.taskConfig.container.mesos.image.docker.name == image_name
assert job.taskConfig.container.mesos.image.docker.tag == image_tag
def test_config_with_appc_image():
image_name = 'some-image'
image_id = 'some-image-id'
job = convert_pystachio_to_thrift(
HELLO_WORLD(container=Mesos(image=AppcImage(name=image_name, image_id=image_id))))
assert job.taskConfig.container.mesos.image.docker is None
assert job.taskConfig.container.mesos.image.appc.name == image_name
assert job.taskConfig.container.mesos.image.appc.imageId == image_id
def test_config_with_volumes():
image_name = 'some-image'
image_tag = 'some-tag'
host_path = '/etc/secrets/role/'
container_path = '/etc/secrets/'
volume = Volume(host_path=host_path, container_path=container_path, mode=Mode('RO'))
container = Mesos(image=DockerImage(name=image_name, tag=image_tag), volumes=[volume])
job = convert_pystachio_to_thrift(HELLO_WORLD(container=container))
assert len(job.taskConfig.container.mesos.volumes) == 1
thrift_volume = job.taskConfig.container.mesos.volumes[0]
assert thrift_volume.hostPath == host_path
assert thrift_volume.containerPath == container_path
assert thrift_volume.mode == ThriftMode.RO