Fix ExecResult compare: allow subclassing. Re-raise auth errors ASAP. (#99)

* Make tests faster.
* Bump to 3.1.1

(cherry picked from commit 2a64bbf)
Signed-off-by: Alexey Stepanov <penguinolog@gmail.com>

Author: penguinolog

exec_helpers/__init__.py

@@ -49,7 +49,7 @@
     "ExecResult",
 )
 
-__version__ = "2.9.0"
+__version__ = "2.9.1"
 __author__ = "Alexey Stepanov"
 __author_email__ = "penguinolog@gmail.com"
 __maintainers__ = {

exec_helpers/_ssh_client_base.py

@@ -49,6 +49,18 @@
 logging.getLogger("iso8601").setLevel(logging.WARNING)
 
 
+class RetryOnExceptions(tenacity.retry_if_exception):  # type: ignore
+    """Advanced retry on exceptions."""
+
+    def __init__(
+        self,
+        retry_on: typing.Union[typing.Type[BaseException], typing.Tuple[typing.Type[BaseException], ...]],
+        reraise: typing.Union[typing.Type[BaseException], typing.Tuple[typing.Type[BaseException], ...]],
+    ) -> None:
+        """Retry on exceptions, except several types."""
+        super(RetryOnExceptions, self).__init__(lambda e: isinstance(e, retry_on) and not isinstance(e, reraise))
+
+
 # noinspection PyTypeHints
 class SshExecuteAsyncResult(api.ExecuteAsyncResult):
     """Override original NamedTuple with proper typing."""
@@ -376,7 +388,7 @@ def _ssh(self) -> paramiko.SSHClient:
         return self.__ssh
 
     @tenacity.retry(  # type: ignore
-        retry=tenacity.retry_if_exception_type(paramiko.SSHException),
+        retry=RetryOnExceptions(retry_on=paramiko.SSHException, reraise=paramiko.AuthenticationException),
         stop=tenacity.stop_after_attempt(3),
         wait=tenacity.wait_fixed(3),
         reraise=True,

exec_helpers/exec_result.py

@@ -465,7 +465,17 @@ def __str__(self) -> str:
 
     def __eq__(self, other: typing.Any) -> bool:
         """Comparision."""
-        return hash(self) == hash(other)
+        return (
+            self.__class__ is other.__class__
+            or issubclass(self.__class__, other.__class__)
+            or issubclass(other.__class__, self.__class__)
+        ) and (
+            self.cmd == other.cmd
+            and self.stdin == other.stdin
+            and self.stdout == other.stdout
+            and self.stderr == other.stderr
+            and self.exit_code == other.exit_code
+        )
 
     def __ne__(self, other: typing.Any) -> bool:
         """Comparision."""

pytest.ini

@@ -1,3 +1,4 @@
 [pytest]
 addopts = -vvv -s -p no:django -p no:ipdb
 testpaths = test
+mock_use_standalone_module = true