formatting

Author: edenhaus

deebot_client/command.py

@@ -21,9 +21,7 @@
     from types import MappingProxyType
 
     from .authentication import Authenticator
-    from .const import (
-        DataType,
-    )
+    from .const import DataType
     from .event_bus import EventBus
     from .models import ApiDeviceInfo
 

tests/commands/__init__.py

@@ -57,13 +57,10 @@ async def assert_command(
 ) -> None:
     command_result = command_result or CommandResult.success()
     event_bus = Mock(spec_set=EventBus)
-    authenticator = Mock(
-        spec_set=Authenticator,
-    )
+    authenticator = Mock(spec_set=Authenticator)
     authenticator.authenticate = AsyncMock(
         return_value=Credentials("token", "user_id", 9999)
     )
-
     if isinstance(json_api_response, tuple):
         mock = AsyncMock(side_effect=json_api_response)
     else:

tests/conftest.py

@@ -12,7 +12,6 @@
 from deebot_client.authentication import (
     Authenticator,
     RestConfiguration,
-    UserAuthenticator,
     create_rest_config as create_config_rest,
 )
 from deebot_client.event_bus import EventBus
@@ -59,8 +58,8 @@ def rest_config(
 
 
 @pytest.fixture
-def authenticator() -> UserAuthenticator:
-    authenticator = Mock(spec_set=UserAuthenticator)
+def authenticator() -> Authenticator:
+    authenticator = Mock(spec_set=Authenticator)
     authenticator.authenticate.return_value = Credentials("token", "user_id", 9999)
     authenticator.execute_command_request.return_value = (
         authenticator.post_authenticated.return_value