Faking proc.poll.

rolobio · May 22, 2014 · badee0b · badee0b
1 parent aa74dca
commit badee0b
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/sshm/test/test_lib.py b/sshm/test/test_lib.py
@@ -195,6 +195,12 @@ def test_stdin(self):
         orig = lib.popen
         lib.popen = sub.popen
 
+        # Fake the process's poll to be None once
+        c = [1, None]
+        def none_once():
+            return c.pop()
+        proc.poll = none_once
+
         from io import BytesIO
         stdin_contents = b'foobar'
         stdin = BytesIO(stdin_contents)