[stream-refactor] fix su_test failure (issue #363)

6 years ago · 0ff5fb8fc4
parent 8769c3ce24
commit 0ff5fb8fc4
1 changed files with 3 additions and 4 deletions
--- a/mitogen/su.py
+++ b/mitogen/su.py
@ -102,10 +102,9 @@ class SetupBootstrapProtocol(mitogen.parent.BootstrapProtocol):
        self.password_sent = True
    def _on_password_incorrect(self, line, match):
-        if self.password_sent:
+        self.stream.conn._fail_connection(
-            self.stream.conn._fail_connection(
+            PasswordError(password_incorrect_msg)
-                PasswordError(password_incorrect_msg)
+        )
            )
 class Options(mitogen.parent.Options):