From 963bad04050d3b0b885264c6b44970d29db18652 Mon Sep 17 00:00:00 2001
From: Will McGugan <willmcgugan@gmail.com>
Date: Thu, 17 Dec 2020 17:05:57 +0000
Subject: [PATCH] file proxy test

---
 tests/test_file_proxy.py | 27 +++++++++++++++++++++++++++
 1 file changed, 27 insertions(+)
 create mode 100644 tests/test_file_proxy.py

diff --git a/tests/test_file_proxy.py b/tests/test_file_proxy.py
new file mode 100644
index 00000000..10d458ee
--- /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"
\ No newline at end of file