fsspec.implementations.smb: try harder to ensure session is registered (fsspec#1597)

Author: ap--

fsspec/implementations/smb.py

@@ -67,6 +67,7 @@ def __init__(
         timeout=60,
         encrypt=None,
         share_access=None,
+        register_session_retries=5,
         **kwargs,
     ):
         """
@@ -111,6 +112,7 @@ def __init__(
         self.encrypt = encrypt
         self.temppath = kwargs.pop("temppath", "")
         self.share_access = share_access
+        self.register_session_retries = register_session_retries
         self._connect()
 
     @property
@@ -120,7 +122,7 @@ def _port(self):
     def _connect(self):
         import time
 
-        for _ in range(5):
+        for _ in range(self.register_session_retries):
             try:
                 smbclient.register_session(
                     self.host,

fsspec/implementations/tests/test_smb.py

@@ -63,6 +63,7 @@ def smb_params(request):
             "port": request.param,
             "username": "testuser",
             "password": "testpass",
+            "register_session_retries": 100,  # max ~= 10 seconds
         }
     finally:
         import smbclient  # pylint: disable=import-outside-toplevel