Improve code coverage: identify imports stream

1 files changed, 9 insertions, 0 deletions

diff --git a/tests/unit/test_main.py b/tests/unit/test_main.py
index 65f0dae4..dd2afdda 100644
--- a/tests/unit/test_main.py
+++ b/tests/unit/test_main.py
@@ -20,6 +20,10 @@ class UnseekableTextIOWrapper(TextIOWrapper):
         raise ValueError("underlying stream is not seekable")
 
 
+def as_stream(text: str) -> UnseekableTextIOWrapper:
+    return UnseekableTextIOWrapper(BytesIO(text.encode("utf8")))
+
+
 @given(
     file_name=st.text(),
     config=st.builds(Config),
@@ -1035,3 +1039,8 @@ def test_identify_imports_main(tmpdir, capsys):
 
     out, error = capsys.readouterr()
     assert out == file_imports.replace("\n", os.linesep)
+    assert not error
+
+    main.identify_imports_main(["-"], stdin=as_stream(file_content))
+    out, error = capsys.readouterr()
+    assert out == file_imports.replace("\n", os.linesep)