mirror of https://github.com/Textualize/rich.git
39 lines
918 B
Python
39 lines
918 B
Python
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"
|
|
|
|
|
|
def test_new_lines():
|
|
|
|
file = io.StringIO()
|
|
console = Console(file=file)
|
|
file_proxy = FileProxy(console, file)
|
|
file_proxy.write("-\n-")
|
|
assert file.getvalue() == "-\n"
|
|
file_proxy.flush()
|
|
assert file.getvalue() == "-\n-\n"
|