def test_log_plan(self): logger = Logger() logger.plan_log = StringIO() logger.log_plan(["plan"]) self.assertEqual("['plan']\n", logger.plan_log.getvalue())
def test_log_plan_log_not_open_yet(self, string_io_open): logger = Logger() logger.plan_log_file_name = None logger.plan_log = None logger.log_plan(["plan"]) self.assertEqual("['plan']\n", string_io_open().getvalue())
def test_close_when_already_closed(self): logger = Logger() logger.log = Mock() logger.log.closed = True logger.plan_log = Mock() logger.plan_log.closed = True logger.close() self.assertEqual(False, logger.log.write.called) self.assertEqual(False, logger.log.close.called) self.assertEqual(False, logger.plan_log.close.called)
def test_close(self): logger = Logger() logger.log = Mock() logger.log.closed = False logger.plan_log = Mock() logger.plan_log.closed = False logger.close() logger.log.write.assert_called_once_with("}\n") logger.log.close.assert_called_once_with() logger.plan_log.close.assert_called_once_with()
def test_close_re_raise_io_error(self): logger = Logger() logger.log = Mock() logger.log.closed = False logger.log.write.side_effect = IOError logger.plan_log = Mock() logger.plan_log.closed = False self.assertRaises(IOError, logger.close) logger.log.write.assert_called_once_with("}\n") self.assertEqual(False, logger.log.close.called) logger.plan_log.close.assert_called_once_with()