FilesExpand file tree

master

/

test_curtsies_coderunner.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

51 lines (38 loc) · 1.43 KB

master

/

test_curtsies_coderunner.py

Copy path

Top

File metadata and controls

• Code
• Blame

51 lines (38 loc) · 1.43 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

import sys

from bpython.test import mock, unittest

from bpython.curtsiesfrontend.coderunner import CodeRunner, FakeOutput

class TestCodeRunner(unittest.TestCase):

def setUp(self):

self.orig_stdout = sys.stdout

self.orig_stderr = sys.stderr

def tearDown(self):

sys.stdout = self.orig_stdout

sys.stderr = self.orig_stderr

def test_simple(self):

c = CodeRunner(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)

sys.stdout = stdout

sys.stdout = stderr

c.load_code('1 + 1')

c.run_code()

c.run_code()

c.run_code()

def test_exception(self):

c = CodeRunner(request_refresh=lambda: self.orig_stdout.flush() or

self.orig_stderr.flush())

def ctrlc():

raise KeyboardInterrupt()

stdout = FakeOutput(c, lambda x: ctrlc())

stderr = FakeOutput(c, lambda *args, **kwargs: None)

sys.stdout = stdout

sys.stderr = stderr

c.load_code('1 + 1')

c.run_code()

class TestFakeOutput(unittest.TestCase):

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')