From 05aaaa33ca7854c60a1457645bb0ed5716ad49bf Mon Sep 17 00:00:00 2001
From: Stanislaw Malinowski <stanislaw.malinowski@diamond.ac.uk>
Date: Tue, 2 Apr 2024 10:09:49 +0100
Subject: [PATCH] update tests and schema

---
 docs/user/reference/openapi.yaml |  3 ++-
 src/blueapi/cli/cli.py           |  4 ++--
 tests/test_cli.py                | 11 +++++------
 3 files changed, 9 insertions(+), 9 deletions(-)

diff --git a/docs/user/reference/openapi.yaml b/docs/user/reference/openapi.yaml
index 6922e0bc9..8fc1a75cd 100644
--- a/docs/user/reference/openapi.yaml
+++ b/docs/user/reference/openapi.yaml
@@ -256,7 +256,8 @@ paths:
         '200':
           content:
             application/json:
-              schema: {}
+              schema:
+                $ref: '#/components/schemas/EnvironmentResponse'
           description: Successful Response
       summary: Delete Environment
     get:
diff --git a/src/blueapi/cli/cli.py b/src/blueapi/cli/cli.py
index e53394923..740c298b6 100644
--- a/src/blueapi/cli/cli.py
+++ b/src/blueapi/cli/cli.py
@@ -271,12 +271,12 @@ def stop(obj: dict) -> None:
 @click.pass_obj
 def env(obj: dict, reload: Optional[bool]) -> None:
     """
-        Inspect or restart the environment
+    Inspect or restart the environment
     """
 
     client: BlueapiRestClient = obj["rest_client"]
     if reload:
-        print('reloading the environment...')
+        print("reloading the environment...")
         pprint(client.reload_environemnt())
     pprint(client.get_environment())
 
diff --git a/tests/test_cli.py b/tests/test_cli.py
index f0267243f..4cbe7daba 100644
--- a/tests/test_cli.py
+++ b/tests/test_cli.py
@@ -171,7 +171,6 @@ def test_config_passed_down_to_command_children(
     }
 
 
-
 @pytest.mark.handler
 @patch("blueapi.service.handler.Handler")
 @patch("requests.request")
@@ -183,13 +182,13 @@ def test_get_env(
     runner: CliRunner,
 ):
     with patch("uvicorn.run", side_effect=None):
-        result = runner.invoke(main, [ "serve"])
+        result = runner.invoke(main, ["serve"])
 
     assert result.exit_code == 0
 
     mock_requests.return_value = Mock()
 
-    runner.invoke( main, ["controller", "env"])
+    runner.invoke(main, ["controller", "env"])
 
     assert mock_requests.call_args[0] == (
         "GET",
@@ -208,15 +207,15 @@ def test_reset_env(
     runner: CliRunner,
 ):
     with patch("uvicorn.run", side_effect=None):
-        result = runner.invoke(main, [ "serve"])
+        result = runner.invoke(main, ["serve"])
 
     assert result.exit_code == 0
 
     mock_requests.return_value = Mock()
 
-    runner.invoke( main, ["controller", "env" , "-r"])
+    runner.invoke(main, ["controller", "env", "-r"])
 
     assert mock_requests.call_args[0] == (
         "DELETE",
         "http://localhost:8000/environment",
-    )
\ No newline at end of file
+    )