def test_run_remote_registry(runner, registry):
img = DockerImage(IMAGE_NAME,
repository=REPOSITORY_NAME,
docker_registry=registry)
host = LocalTargetHost()
instance = DockerServiceInstance(CONTAINER_NAME, img, host)
if is_container_running(CONTAINER_NAME):
stop_container(CONTAINER_NAME, host)
runner = runner(host, img, CONTAINER_NAME)
runner.run(instance)
assert is_container_running(CONTAINER_NAME, host)
def test_run_default_registry(runner):
img_registry = DefaultDockerRegistry()
img = DockerImage(IMAGE_NAME, docker_registry=img_registry)
host = LocalTargetHost()
instance = DockerServiceInstance(CONTAINER_NAME, img, host, {80: 8080})
if is_container_running(CONTAINER_NAME):
stop_container(CONTAINER_NAME, host)
runner = runner(host, img, CONTAINER_NAME)
runner.run(instance, detach=True)
assert is_container_running(CONTAINER_NAME, host)
def test_build_and_run_service(ebnt, container_name):
reg, data = train_model()
task = ebnt.get_or_create_task("Test Project", "Test Task")
model = task.create_and_push_model(reg, data, "Test Model")
ebnt.build_and_run_service(container_name, model, detach=True)
assert is_container_running(container_name)
def test_create_service_from_model(service_name):
reg, data = train_model()
with use_local_installation():
create_service_from_model(model_name='test_model',
model_object=reg,
model_input=data,
service_name=service_name,
run_service=True)
assert is_container_running(service_name)
def test_run_local_fail_inside_container_attached(runner, registry):
img = DockerImage(BROKEN_IMAGE_NAME,
repository=REPOSITORY_NAME,
docker_registry=registry)
host = LocalTargetHost()
instance = DockerServiceInstance(CONTAINER_NAME, img, host, {80: 8080})
if is_container_running(CONTAINER_NAME):
stop_container(CONTAINER_NAME, host)
with pytest.raises(DockerRunnerException):
runner = runner(host, img, CONTAINER_NAME)
runner.run(instance, detach=False, rm=True)
def test_run_local_image_name_that_will_never_exist(runner):
img_registry = DefaultDockerRegistry()
img = DockerImage('ebonite_image_name_that_will_never_exist',
docker_registry=img_registry)
host = LocalTargetHost()
instance = DockerServiceInstance(CONTAINER_NAME, img, host, {80: 8080})
if is_container_running(CONTAINER_NAME):
stop_container(CONTAINER_NAME, host)
with pytest.raises(HTTPError):
runner = runner(host, img, CONTAINER_NAME)
runner.run(instance)
def test_run_docker_img(container_name):
run_docker_img(container_name, 'mike0sv/ebaklya', detach=True)
assert is_container_running(container_name)
def test_build_model_docker(model, server, img_name, container_name):
with use_local_installation():
build_model_flask_docker(img_name, model, force_overwrite=True)
run_docker_img(container_name, img_name, detach=True)
assert is_container_running(container_name)
