darken

Author: meejah

src/_zkapauthorizer/recover.py

@@ -119,7 +119,11 @@ async def recover(
         :param cursor: A database cursor which can be used to populate the
             database with recovered state.
         """
-        if self._state.stage not in {RecoveryStages.inactive, RecoveryStages.download_failed, RecoveryStages.import_failed}:
+        if self._state.stage not in {
+            RecoveryStages.inactive,
+            RecoveryStages.download_failed,
+            RecoveryStages.import_failed,
+        }:
             return
 
         self._set_state(RecoveryState(stage=RecoveryStages.started))

src/_zkapauthorizer/tests/test_client_resource.py

@@ -1030,18 +1030,20 @@ async def recover():
                     lambda messages: list(
                         loads(args[0]) for (args, kwargs) in messages
                     ),
-                    Equals([
-                        {
-                            "stage": "started",
-                            "failure-reason": None,
-                        },
-                        # "our" downloader (above) doesn't set any downloading etc
-                        # state-updates
-                        {
-                            "stage": "download_failed",
-                            "failure-reason": "downloader fails",
-                        },
-                    ]),
+                    Equals(
+                        [
+                            {
+                                "stage": "started",
+                                "failure-reason": None,
+                            },
+                            # "our" downloader (above) doesn't set any downloading etc
+                            # state-updates
+                            {
+                                "stage": "download_failed",
+                                "failure-reason": "downloader fails",
+                            },
+                        ]
+                    ),
                 )
             ),
         )
@@ -1053,18 +1055,20 @@ async def recover():
                     lambda messages: list(
                         loads(args[0]) for (args, kwargs) in messages
                     ),
-                    Equals([
-                        {
-                            "stage": "started",
-                            "failure-reason": None,
-                        },
-                        # "our" downloader (above) doesn't set any downloading etc
-                        # state-updates
-                        {
-                            "stage": "succeeded",
-                            "failure-reason": None,
-                        },
-                    ]),
+                    Equals(
+                        [
+                            {
+                                "stage": "started",
+                                "failure-reason": None,
+                            },
+                            # "our" downloader (above) doesn't set any downloading etc
+                            # state-updates
+                            {
+                                "stage": "succeeded",
+                                "failure-reason": None,
+                            },
+                        ]
+                    ),
                 )
             ),
         )