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