Add test for input case

Author: fcollonval

jupyter_server_nbmodel/handlers.py

@@ -375,6 +375,8 @@ async def post(self, kernel_id: str) -> None:
             client.stop_channels()
             del client
 
+        self.set_status(HTTPStatus.CREATED)
+
 
 class RequestHandler(ExtensionHandlerMixin, APIHandler):
     """Handler for /api/kernels/<kernel_id>/requests/<request_id>"""

jupyter_server_nbmodel/tests/test_handlers.py

@@ -19,8 +19,9 @@ async def _wait_request(fetch, endpoint: str):
 
     while (datetime.datetime.now() - start_time).total_seconds() < 0.9 * TEST_TIMEOUT:
         await asyncio.sleep(SLEEP)
-        response = await fetch(endpoint)
-        response.rethrow()
+        response = await fetch(endpoint, raise_error=False)
+        if response.code >= 400:
+            response.rethrow()
         if response.code != 202:
             return response
 
@@ -142,6 +143,54 @@ async def test_post_erroneous_execute(jp_fetch, pending_kernel_is_ready, snippet
     assert response2.code == 204
 
 
+@pytest.mark.timeout(TEST_TIMEOUT)
+async def test_post_input_execute(jp_fetch, pending_kernel_is_ready):
+    # Start the first kernel
+    r = await jp_fetch(
+        "api", "kernels", method="POST", body=json.dumps({"name": NATIVE_KERNEL_NAME})
+    )
+    kernel = json.loads(r.body.decode())
+    await pending_kernel_is_ready(kernel["id"])
+
+    response = await jp_fetch(
+        "api",
+        "kernels",
+        kernel["id"],
+        "execute",
+        method="POST",
+        body=json.dumps({"code": "input('Age:')"}),
+    )
+    assert response.code == 202
+    location = response.headers["Location"]
+
+    response2 = await _wait_request(jp_fetch, location)
+
+    assert response2.code == 300
+    payload = json.loads(response2.body)
+    assert "parent_header" in payload
+    assert payload["input_request"] == {"prompt": "Age:", "password": False}
+
+    response3 = await jp_fetch(
+        "api", "kernels", kernel["id"], "input", method="POST", body=json.dumps({"input": "42"})
+    )
+    assert response3.code == 201
+
+    response4 = await _wait_request(
+        jp_fetch,
+        location
+    )
+    assert response4.code == 200
+    payload2 = json.loads(response4.body)
+    assert payload2 == {
+        "status": "ok",
+        "execution_count": 1,
+        "outputs": '[{"output_type": "execute_result", "metadata": {}, "data": {"text/plain": "\'42\'"}, "execution_count": 1}]',
+    }
+
+    r2 = await jp_fetch("api", "kernels", kernel["id"], method="DELETE")
+    assert r2.code == 204
+
+
 # FIXME
 # @pytest.mark.timeout(TEST_TIMEOUT)
 # async def test_cancel_execute(jp_fetch, pending_kernel_is_ready):