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