""" Test cases for PtyProcess.wait method. """
import time
import unittest
from ptyprocess import PtyProcess


class TestWaitAfterTermination(unittest.TestCase):
    """Various test cases for PtyProcess.wait()"""

    def test_wait_true_shortproc(self):
        """Ensure correct (True) wait status for short-lived processes."""
        child = PtyProcess.spawn(['true'])
        # Wait so we're reasonable sure /bin/true has terminated
        time.sleep(0.2)
        self.assertEqual(child.wait(), 0)

    def test_wait_false_shortproc(self):
        """Ensure correct (False) wait status for short-lived processes."""
        child = PtyProcess.spawn(['false'])
        # Wait so we're reasonable sure /bin/false has terminated
        time.sleep(0.2)
        self.assertNotEqual(child.wait(), 0)

    def test_wait_twice_longproc(self):
        """Ensure correct wait status when called twice."""
        # previous versions of ptyprocess raises PtyProcessError when
        # wait was called more than once with "Cannot wait for dead child
        # process.".  No longer true since v0.5.
        child = PtyProcess.spawn(['sleep', '1'])
        # this call to wait() will block for 1s
        for count in range(2):
            self.assertEqual(child.wait(), 0, count)