Switch to pytest for container testing

test/test_notebook.py

@@ -0,0 +1,52 @@
+# Copyright (c) Jupyter Development Team.
+# Distributed under the terms of the Modified BSD License.
+import os
+import time
+
+import docker
+import pytest
+import requests
+
+
+@pytest.fixture(scope='session')
+def docker_client():
+    """Docker client to use"""
+    return docker.from_env()
+
+
+@pytest.fixture(scope='session')
+def image_name():
+    """Image name to test"""
+    return os.getenv('TEST_IMAGE', 'jupyter/base-notebook')
+
+
+@pytest.fixture(scope='function')
+def nb_container(docker_client, image_name):
+    """Notebook container to test"""
+    container = docker_client.containers.run(
+        image_name,
+        detach=True,
+        auto_remove=True,
+        ports={
+            '8888/tcp': 8888
+        }
+    )
+    yield container
+    container.kill()
+
+
+def test_server_liveliness(nb_container):
+    """Notebook server should eventually respond with HTTP 200 OK."""
+    for i in range(10):
+        try:
+            resp = requests.get('http://localhost:8888')
+        except requests.exceptions.ConnectionError:
+            # Wait a bit and try again. Just because the docker container
+            # is running doesn't mean the notebook server is ready to accept
+            # connections inside it.
+            time.sleep(i)
+        else:
+            assert resp.status_code == 200
+            break
+    else:
+        assert False, 'could not connect to server'