Expose EncryptionMethod.NONE results

Author: espipj

sagemcom_api/client.py

@@ -579,13 +579,15 @@ def __ensure_legacy_api(self):
                 "Use helper methods supported for REST firmware instead."
             )
 
-    async def get_encryption_method(self):
+    async def get_encryption_method(self) -> EncryptionMethod:
         """Determine which encryption method to use for authentication and set it directly."""
         if self.api_mode == ApiMode.REST:
-            return None
+            return EncryptionMethod.NONE
 
         for encryption_method in EncryptionMethod:
             try:
+                if encryption_method == EncryptionMethod.NONE:
+                    continue
                 self.authentication_method = encryption_method
                 self._password_hash = self.__generate_hash(
                     self.password, encryption_method
@@ -605,7 +607,7 @@ async def get_encryption_method(self):
             ):
                 pass
 
-        return None
+        return EncryptionMethod.NONE
 
     @backoff.on_exception(
         backoff.expo,

sagemcom_api/enums.py

@@ -18,6 +18,7 @@ class EncryptionMethod(StrEnum):
     MD5 = "MD5"
     MD5_NONCE = "MD5_NONCE"
     SHA512 = "SHA512"
+    NONE = "NONE"
 
 
 @unique

tests/unit/test_client_basic.py

@@ -187,6 +187,22 @@ async def test_login_auto_fallbacks_to_rest_when_legacy_503():
     assert mock_session.request.call_count == 1
 
 
+@pytest.mark.asyncio
+async def test_get_encryption_method_rest_returns_none():
+    """REST mode should immediately signal that no encryption method is needed."""
+    client = SagemcomClient(
+        host="192.168.1.1",
+        username="admin",
+        password="admin",
+        api_mode=ApiMode.REST,
+    )
+
+    result = await client.get_encryption_method()
+
+    assert result == EncryptionMethod.NONE
+    await client.close()
+
+
 @pytest.mark.asyncio
 async def test_get_hosts_rest_mode():
     """get_hosts should parse wifi and ethernet devices on REST firmware."""