def test_container_override_params(docker_image_built, tmpdir, mocker):
def validate_response(resp):
assert len(resp.docs) > 0
for doc in resp.docs:
assert doc.tags['greetings'] == 'overriden greetings'
mock = mocker.Mock()
abc_path = os.path.join(tmpdir, 'abc')
f = Flow().add(
name=random_name(),
uses=f'docker://{img_name}',
volumes=abc_path + ':' + '/mapped/here/abc',
uses_with={'greetings': 'overriden greetings'},
uses_metas={
'name': 'ext-mwu-encoder',
'workspace': '/mapped/here/abc',
},
)
with f:
responses = f.index(random_docs(10), return_results=True)
assert len(responses) > 0
assert os.path.exists(
os.path.join(abc_path, 'ext-mwu-encoder', '0', 'ext-mwu-encoder.bin'))
validate_response(responses[0])
def test_container_volume(docker_image_built, tmpdir):
abc_path = os.path.join(tmpdir, 'abc')
f = (Flow()
.add(name=random_name(), uses=img_name, volumes=abc_path,
uses_internal=os.path.join(cur_dir, '../mwu-encoder/mwu_encoder_upd.yml')))
with f:
f.index(random_docs(10))
out_file = os.path.join(abc_path, 'ext-mwu-encoder.bin')
assert os.path.exists(out_file)
def test_container_volume_arbitrary(docker_image_built, tmpdir):
abc_path = os.path.join(tmpdir, 'abc')
f = (Flow().add(name=random_name(),
uses=f'docker://{img_name}',
volumes=abc_path + ':' + '/mapped/here/abc',
uses_internal=os.path.join(
cur_dir,
'../../../mwu-encoder/mwu_encoder_volume_change.yml')))
with f:
f.index(random_docs(10))
assert os.path.exists(os.path.join(abc_path, 'ext-mwu-encoder.bin'))
def test_container_volume(self):
f = (Flow().add(name=random_name(),
uses=img_name,
volumes='./abc',
uses_internal=os.path.join(
cur_dir, 'mwu-encoder/mwu_encoder_upd.yml')))
with f:
f.index(random_docs(10))
out_file = 'ext-mwu-encoder.bin'
self.assertTrue(
os.path.exists(os.path.join(os.path.abspath('./abc'), out_file)))
self.add_tmpfile(out_file, os.path.abspath('./abc'))
def test_container_volume(docker_image_built, tmpdir):
tmpdir = Path(tmpdir)
abc_path = tmpdir / 'abc'
f = (Flow().add(name=random_name(),
uses=img_name,
volumes=str(abc_path),
uses_internal=str(cur_dir.parent / 'mwu-encoder' /
'mwu_encoder_upd.yml')))
with f:
f.index(random_docs(10))
out_file = abc_path / 'ext-mwu-encoder.bin'
assert out_file.exists()
def test_container_volume(docker_image_built, tmpdir):
abc_path = os.path.join(tmpdir, 'abc')
f = Flow().add(
name=random_name(),
uses=f'docker://{img_name}',
volumes=abc_path,
workspace='/abc',
uses_internal=os.path.join(cur_dir,
'../../../mwu-encoder/mwu_encoder_upd.yml'),
)
with f:
f.index(random_docs(10))
assert os.path.exists(
os.path.join(abc_path, 'ext-mwu-encoder', '0', 'ext-mwu-encoder.bin'))
def test_container_volume(docker_image_built, tmpdir):
abc_path = os.path.join(tmpdir, 'abc')
f = Flow().add(
name=random_name(),
uses=f'docker://{img_name}',
volumes=abc_path + ':' + '/mapped/here/abc',
uses_metas={
'name': 'ext-mwu-encoder',
'workspace': '/mapped/here/abc',
},
)
with f:
f.index(random_docs(10))
assert os.path.exists(
os.path.join(abc_path, 'ext-mwu-encoder', '0', 'ext-mwu-encoder.bin'))
