fix lint

Signed-off-by: Tim Paine <[email protected]>

Author: timkpaine

csp_gateway/server/modules/controls/controls.py

@@ -22,7 +22,7 @@ class MountControls(GatewayModule):
     mount_heartbeat: bool = True
     mount_stats: bool = True
     mount_shutdown: bool = True
-    
+
     def connect(self, channels: GatewayChannels) -> None:
         self.subscribe(channels.get_channel("controls"))
         channels.add_send_channel("controls")
@@ -38,7 +38,7 @@ def rest(self, app: GatewayWebApp) -> None:
         if self.mount_shutdown:
             app.add_controls_api(field="shutdown")
             available.append("shutdown")
-            
+
         app.add_controls_available_channels(fields=set(available))
 
     @csp.node

csp_gateway/server/web/routes/controls.py

@@ -21,7 +21,6 @@
 
 
 def add_controls_routes(api_router: APIRouter, field: str) -> None:
-    
     if field == "heartbeat":
         # Add heartbeat channel
         @api_router.get(
@@ -50,6 +49,7 @@ async def heartbeat(request: Request) -> Controls:
             return prepare_response(data, is_list_model=False)
 
     elif field == "stats":
+
         @api_router.get(
             "/stats",
             responses=get_default_responses(),
@@ -83,6 +83,7 @@ async def stats(request: Request) -> Controls:
             return prepare_response(data, is_list_model=False)
 
     elif field == "shutdown":
+
         @api_router.post(
             "/shutdown",
             responses=get_default_responses(),
@@ -100,13 +101,12 @@ async def shutdown(request: Request, background_tasks: BackgroundTasks) -> Contr
 
             data = Controls(name="shutdown", status="ok")
             return prepare_response(data, is_list_model=False)
-        
+
     else:
         raise ValueError(f"Unsupported controls field: {field}. Supported fields are 'heartbeat', 'stats', and 'shutdown'.")
 
 
 def add_controls_available_channels(api_router: APIRouter, fields: Optional[Set[str]] = None) -> None:
-    
     @api_router.get(
         "/",
         responses=get_default_responses(),
@@ -117,4 +117,3 @@ async def get_controls(request: Request) -> List[str]:
         This endpoint will return a list of string values of all available channels under the `/controls` route.
         """
         return sorted(fields if fields else ("heartbeat", "stats", "shutdown"))
-

csp_gateway/server/web/routes/last.py

@@ -14,6 +14,7 @@
     "add_last_available_channels",
 )
 
+
 def add_last_routes(
     api_router: APIRouter,
     field: str,
@@ -151,7 +152,6 @@ async def get_last(request: Request) -> list_model:  # type: ignore[misc, valid-
 
 
 def add_last_available_channels(api_router: APIRouter, fields: Optional[Set[str]] = None) -> None:
-    
     @api_router.get(
         "/",
         responses=get_default_responses(),
@@ -162,4 +162,3 @@ async def get_last(request: Request) -> List[str]:
         This endpoint will return a list of string values of all available channels under the `/last` route.
         """
         return sorted(ChannelSelection().select_from(request.app.gateway.channels) if fields is None else fields)
-

csp_gateway/server/web/routes/lookup.py

@@ -13,6 +13,7 @@
     "add_lookup_available_channels",
 )
 
+
 def add_lookup_routes(
     api_router: APIRouter,
     field: str,
@@ -48,8 +49,8 @@ async def lookup(id: str, request: Request) -> List[model]:  # type: ignore[misc
         include_in_schema=False,
     )(lookup)
 
+
 def add_lookup_available_channels(api_router: APIRouter, fields: Optional[Set[str]] = None) -> None:
-    
     @api_router.get(
         "/",
         responses=get_default_responses(),
@@ -60,5 +61,3 @@ async def get_lookup(request: Request) -> List[str]:
         This endpoint will return a list of string values of all available channels under the `/lookup` route.
         """
         return sorted(ChannelSelection().select_from(request.app.gateway.channels) if fields is None else fields)
-
-

csp_gateway/server/web/routes/next.py

@@ -14,6 +14,7 @@
     "add_next_available_channels",
 )
 
+
 def add_next_routes(
     api_router: APIRouter,
     field: str,

csp_gateway/server/web/routes/send.py

@@ -17,6 +17,7 @@
     "add_send_available_channels",
 )
 
+
 def add_send_routes(
     api_router: APIRouter,
     field: str,

csp_gateway/server/web/routes/shared.py

@@ -8,6 +8,7 @@
     "get_next_tick",
 )
 
+
 def prepare_response(
     res: Any,
     is_list_model: bool = False,

csp_gateway/server/web/routes/state.py

@@ -14,6 +14,7 @@
     "add_state_available_channels",
 )
 
+
 def add_state_routes(
     api_router: APIRouter,
     field: str = "",

csp_gateway/tests/server/gateway/test_gateway_start_stop.py

@@ -12,7 +12,7 @@
 import requests
 from fastapi.testclient import TestClient
 
-from csp_gateway import Gateway, GatewaySettings, MountRestRoutes
+from csp_gateway import Gateway, GatewaySettings, MountControls, MountRestRoutes
 from csp_gateway.server.demo import ExampleGatewayChannels, ExampleModule
 from csp_gateway.testing import CspDieModule, LongStartModule
 from csp_gateway.tests.server.gateway.test_gateway import MyBuildFailureModule
@@ -119,10 +119,11 @@ def test_start_and_then_graph_start_error(caplog, free_port):
 def test_stop_with_shutdown(free_port):
     # instantiate gateway
     gateway = Gateway(
-        settings=GatewaySettings(API_KEY="12345", AUTHENTICATE=False, PORT=free_port),
+        settings=GatewaySettings(AUTHENTICATE=False, PORT=free_port),
         modules=[
             ExampleModule(),
             MountRestRoutes(force_mount_all=True),
+            MountControls(),
         ],
         channels=ExampleGatewayChannels(),
     )
@@ -145,6 +146,7 @@ def run_gateway(port_str):
         modules=[
             ExampleModule(),
             MountRestRoutes(force_mount_all=True),
+            MountControls(),
         ],
         channels=ExampleGatewayChannels(),
     )