]> git.ipfire.org Git - thirdparty/tornado.git/commitdiff
Change variable names and streamline test method
authorPing <yuanta11@gmail.com>
Tue, 11 Aug 2015 05:41:44 +0000 (13:41 +0800)
committerPing <yuanta11@gmail.com>
Sun, 6 Sep 2015 06:57:02 +0000 (14:57 +0800)
tornado/log.py
tornado/test/log_test.py

index b72c5ba85e515ced52c3d277f70138f07738ef1b..9b147bf0a04c3a78a82a90774b0a13f2562f0e93 100644 (file)
@@ -190,16 +190,16 @@ def enable_pretty_logging(options=None, logger=None):
         logger = logging.getLogger()
     logger.setLevel(getattr(logging, options.logging.upper()))
     if options.log_file_prefix:
-        if options.rotating_mode == 'sized':
+        if options.log_rotate_mode == 'size':
             channel = logging.handlers.RotatingFileHandler(
                 filename=options.log_file_prefix,
                 maxBytes=options.log_file_max_size,
                 backupCount=options.log_file_num_backups)
-        elif options.rotating_mode == 'timed':
+        elif options.log_rotate_mode == 'time':
             channel = logging.handlers.TimedRotatingFileHandler(
                 filename=options.log_file_prefix,
-                when=options.log_file_rotating_when,
-                interval=options.log_file_rotating_interval,
+                when=options.log_rotate_when,
+                interval=options.log_rotate_interval,
                 backupCount=options.log_file_num_backups)
         else:
             channel = logging.NullHandler()
@@ -244,13 +244,13 @@ def define_logging_options(options=None):
     options.define("log_file_num_backups", type=int, default=10,
                    help="number of log files to keep")
 
-    options.define("log_file_rotating_when", type=str, default='midnight',
+    options.define("log_rotate_when", type=str, default='midnight',
                    help=("specify the type of TimedRotatingFileHandler interval "
                          "other options:('S', 'M', 'H', 'D', 'W0'-'W6')"))
-    options.define("log_file_rotating_interval", type=int, default=1,
+    options.define("log_rotate_interval", type=int, default=1,
                    help="The interval value of timed rotating")
 
-    options.define("rotating_mode", type=str, default='sized',
-                   help="The mode of rotating files(timed or sized)")
+    options.define("log_rotate_mode", type=str, default='size',
+                   help="The mode of rotating files(time or size)")
 
     options.add_parse_callback(lambda: enable_pretty_logging(options))
index 58582522f9ba8bee36765dde58b2ccfcb5144219..33f2cbc25894a4c38d12df6bda6e4d9eb5d89855 100644 (file)
@@ -165,25 +165,16 @@ class EnablePrettyLoggingTest(unittest.TestCase):
         tmpdir = tempfile.mkdtemp()
         try:
             self.options.log_file_prefix = tmpdir + '/test_log'
-            self.options.rotating_mode = 'timed'
-            self.options.log_file_rotating_when = 'S'
-            self.options.log_file_rotating_interval = 1
+            self.options.log_rotate_mode = 'time'
             enable_pretty_logging(options=self.options, logger=self.logger)
-            self.logger.error('hello1 from TimedRotatingFileHandler')
-            time.sleep(1.01)
-            self.logger.error('hello2 from TimedRotatingFileHandler')
+            self.logger.error('hello')
+            self.logger.handlers[0].flush()
             filenames = glob.glob(tmpdir + '/test_log*')
-            self.assertEqual(2, len(filenames))
-            new_file = tmpdir + '/test_log'
-            old_file = glob.glob(tmpdir + '/test_log.*')[0]
-            with open(old_file) as f:
-                self.assertRegexpMatches(
-                    f.read(),
-                    r'^\[E [^]]*\] hello1 from TimedRotatingFileHandler$')
-            with open(new_file) as f:
+            self.assertEqual(1, len(filenames))
+            with open(filenames[0]) as f:
                 self.assertRegexpMatches(
                     f.read(),
-                    r'^\[E [^]]*\] hello2 from TimedRotatingFileHandler$')
+                    r'^\[E [^]]*\] hello$')
         finally:
             for handler in self.logger.handlers:
                 handler.flush()