class Test(unittest.TestCase): def setUp(self): self.os=Mock() self.file=Mock(write=MagicMock()) self.sys=Mock(stdout=self.file) self.screen_printer = ScreenPrinter(self.os, self.sys) def test_print_lines(self): lines = ['a', 'b'] self.screen_printer.print_lines(lines) self.file.write.assert_has_calls([call('a\n'), call('b\n')]) def test_clear(self): self.screen_printer.clear() self.assertTrue(self.os.wasClearCalled) self.os.clear.assert_called()
def setUp(self): self.mock_telnet = MockTelnet() self.mock_os = MockOs() self.mock_file = MockFile() self.mock_file.char = 'q' self.mock_time = MockTime() self.mock_sys = MockSys(self.mock_file) self.mock_termios = MockTermios() self.mock_fcntl = MockFcntl() self.beanstalkd = Beanstalkd(self.mock_telnet, '', '') self.stats = BeanstalkdStats(self.beanstalkd) self.char_reader = CharReader(self.mock_os, self.mock_sys, self.mock_termios, self.mock_fcntl) self.screen_printer = ScreenPrinter(self.mock_os, self.mock_sys) self.console = Console(Clock(self.mock_time), self.char_reader, self.screen_printer, self.stats)
def setUp(self): self.os=Mock() self.file=Mock(write=MagicMock()) self.sys=Mock(stdout=self.file) self.screen_printer = ScreenPrinter(self.os, self.sys)