Bläddra i källkod

HUE-2944 [proxy] Fix logging in test progress-line

Test server in proxy_test write logs in std.err instead of proper
logging.

Override `log_message` method of BaseHTTPRequestHandler.
Ivan Orlov 10 år sedan
förälder
incheckning
fe6e66a762
1 ändrade filer med 9 tillägg och 0 borttagningar
  1. 9 0
      apps/proxy/src/proxy/proxy_test.py

+ 9 - 0
apps/proxy/src/proxy/proxy_test.py

@@ -18,6 +18,7 @@
 # Tests for proxy app.
 
 import threading
+import logging
 import BaseHTTPServer
 import StringIO
 
@@ -28,6 +29,7 @@ from desktop.lib.django_test_util import make_logged_in_client
 from proxy.views import _rewrite_links
 import proxy.conf
 
+
 class Handler(BaseHTTPServer.BaseHTTPRequestHandler):
   """
   To avoid mocking out urllib, we setup a web server
@@ -52,6 +54,13 @@ class Handler(BaseHTTPServer.BaseHTTPRequestHandler):
     # number of bytes the test sends.
     self.wfile.write("Data: " + self.rfile.read(16))
 
+  def log_message(self, fmt, *args):
+    logging.debug("%s - - [%s] %s" %
+                  (self.address_string(),
+                   self.log_date_time_string(),
+                   fmt % args))
+
+
 def run_test_server():
   """
   Returns the server, and a method to close it out.