Update tests for new FakeOutput

thomasballinger · thomasballinger · commit 525e360f08a5 · 2020-07-04T23:01:47.000-06:00
diff --git a/bpython/test/test_curtsies_coderunner.py b/bpython/test/test_curtsies_coderunner.py
@@ -20,8 +20,8 @@ def test_simple(self):
             request_refresh=lambda: self.orig_stdout.flush()
             or self.orig_stderr.flush()
         )
-        stdout = FakeOutput(c, lambda *args, **kwargs: None)
-        stderr = FakeOutput(c, lambda *args, **kwargs: None)
+        stdout = FakeOutput(c, lambda *args, **kwargs: None, None)
+        stderr = FakeOutput(c, lambda *args, **kwargs: None, None)
         sys.stdout = stdout
         sys.stdout = stderr
         c.load_code("1 + 1")
@@ -38,8 +38,8 @@ def test_exception(self):
         def ctrlc():
             raise KeyboardInterrupt()
 
-        stdout = FakeOutput(c, lambda x: ctrlc())
-        stderr = FakeOutput(c, lambda *args, **kwargs: None)
+        stdout = FakeOutput(c, lambda x: ctrlc(), None)
+        stderr = FakeOutput(c, lambda *args, **kwargs: None, None)
         sys.stdout = stdout
         sys.stderr = stderr
         c.load_code("1 + 1")
@@ -51,5 +51,5 @@ def assert_unicode(self, s):
         self.assertIsInstance(s, type(u""))
 
     def test_bytes(self):
-        out = FakeOutput(mock.Mock(), self.assert_unicode)
-        out.write("native string type")
+        out = FakeOutput(mock.Mock(), self.assert_unicode, None)
+        out.write('native string type')
