Adding upstream version 9.11.0.upstream/9.11.0upstream

Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>

1 files changed, 27 insertions, 0 deletions

diff --git a/tests/test_file_proxy.py b/tests/test_file_proxy.py
new file mode 100644
index 0000000..2218d03
--- /dev/null
+++ b/tests/test_file_proxy.py
@@ -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"