file proxy test

This commit is contained in:
Will McGugan 2020-12-17 17:05:57 +00:00
parent 7747025280
commit 963bad0405
1 changed files with 27 additions and 0 deletions

27
tests/test_file_proxy.py Normal file
View File

@ -0,0 +1,27 @@
import io
import sys
import pytest
from rich.console import Console
from rich.file_proxy import FileProxy
def test_empty_bytes():
console = Console()
file_proxy = FileProxy(console, sys.stdout)
# File should raise TypeError when writing bytes
with pytest.raises(TypeError):
file_proxy.write(b"") # type: ignore
with pytest.raises(TypeError):
file_proxy.write(b"foo") # type: ignore
def test_flush():
file = io.StringIO()
console = Console(file=file)
file_proxy = FileProxy(console, file)
file_proxy.write("foo")
assert file.getvalue() == ""
file_proxy.flush()
assert file.getvalue() == "foo\n"