58 lines
1.8 KiB
Python
58 lines
1.8 KiB
Python
# Copyright Jonathan Hartley 2013. BSD 3-Clause license, see LICENSE file.
|
|
import sys
|
|
from unittest import TestCase, main
|
|
|
|
from ..ansitowin32 import StreamWrapper, AnsiToWin32
|
|
from .utils import pycharm, replace_by, replace_original_by, StreamTTY, StreamNonTTY
|
|
|
|
|
|
def is_a_tty(stream):
|
|
return StreamWrapper(stream, None).isatty()
|
|
|
|
class IsattyTest(TestCase):
|
|
|
|
def test_TTY(self):
|
|
tty = StreamTTY()
|
|
self.assertTrue(is_a_tty(tty))
|
|
with pycharm():
|
|
self.assertTrue(is_a_tty(tty))
|
|
|
|
def test_nonTTY(self):
|
|
non_tty = StreamNonTTY()
|
|
self.assertFalse(is_a_tty(non_tty))
|
|
with pycharm():
|
|
self.assertFalse(is_a_tty(non_tty))
|
|
|
|
def test_withPycharm(self):
|
|
with pycharm():
|
|
self.assertTrue(is_a_tty(sys.stderr))
|
|
self.assertTrue(is_a_tty(sys.stdout))
|
|
|
|
def test_withPycharmTTYOverride(self):
|
|
tty = StreamTTY()
|
|
with pycharm(), replace_by(tty):
|
|
self.assertTrue(is_a_tty(tty))
|
|
|
|
def test_withPycharmNonTTYOverride(self):
|
|
non_tty = StreamNonTTY()
|
|
with pycharm(), replace_by(non_tty):
|
|
self.assertFalse(is_a_tty(non_tty))
|
|
|
|
def test_withPycharmNoneOverride(self):
|
|
with pycharm():
|
|
with replace_by(None), replace_original_by(None):
|
|
self.assertFalse(is_a_tty(None))
|
|
self.assertFalse(is_a_tty(StreamNonTTY()))
|
|
self.assertTrue(is_a_tty(StreamTTY()))
|
|
|
|
def test_withPycharmStreamWrapped(self):
|
|
with pycharm():
|
|
self.assertTrue(AnsiToWin32(StreamTTY()).stream.isatty())
|
|
self.assertFalse(AnsiToWin32(StreamNonTTY()).stream.isatty())
|
|
self.assertTrue(AnsiToWin32(sys.stdout).stream.isatty())
|
|
self.assertTrue(AnsiToWin32(sys.stderr).stream.isatty())
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|