mirror of https://github.com/Textualize/rich.git
file proxy test
This commit is contained in:
parent
7747025280
commit
963bad0405
|
@ -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"
|
Loading…
Reference in New Issue