import os
import pty

from nose.tools import assert_equal
from test_helper import Test

from stdout import Stdout
from pty_recorder import PtyRecorder


class FakeStdout(object):

    def __init__(self):
        self.data = []
        self.closed = False

    def write(self, data):
        self.data.append(data)


class TestPtyRecorder(Test):

    def setUp(self):
        self.real_os_write = os.write
        os.write = self.os_write

    def tearDown(self):
        os.write = self.real_os_write

    def os_write(self, fd, data):
        if fd != pty.STDOUT_FILENO:
            self.real_os_write(fd, data)

    def test_record_command_returns_stdout_instance(self):
        pty_recorder = PtyRecorder()

        output = pty_recorder.record_command('ls -l')

        assert_equal(Stdout, type(output))

    def test_record_command_writes_to_stdout(self):
        pty_recorder = PtyRecorder()
        output = FakeStdout()

        command = 'python -c "import sys; sys.stdout.write(\'foo\\n\'); ' \
                  'sys.stdout.flush(); sys.stdout.write(\'bar\')"'
        pty_recorder.record_command(command, output)

        assert_equal(['foo\r\n', 'bar'], output.data)