Stdin ssh flush

Changelog.rst

@@ -1,6 +1,14 @@
 Change Log
 ============
 
+2.12.1
++++++++
+
+Fixes
+------
+
+* Calling ``HostOutput.stdin.flush`` with a ``pssh.clients.ssh`` client would raise exception - #.
+
 2.12.0
 +++++++
 

pssh/clients/base/single.py

@@ -68,6 +68,8 @@ def write(self, data):
 
     def flush(self):
         """Flush pending data written to stdin."""
+        if not hasattr(self._channel, "flush"):
+            return
         return self._client._eagain(self._channel.flush)
 
 

tests/ssh/test_single_client.py

@@ -326,3 +326,11 @@ def _disc():
         client._disconnect_eagain = _disc
         client._connect_init_session_retry(0)
         client.disconnect()
+
+    def test_stdin(self):
+        host_out = self.client.run_command('read line; echo $line')
+        host_out.stdin.write('a line\n')
+        host_out.stdin.flush()
+        self.client.wait_finished(host_out)
+        stdout = list(host_out.stdout)
+        self.assertListEqual(stdout, ['a line'])