openstack/deb-python-oauth2client
Code Issues Proposed changes

Merge pull request #122 from cskertesz/master

Browse Source
This commit is contained in:
Nathaniel Manista
2015-02-12 10:30:23 -08:00
parent 28cc2e0392 927589bf06
commit c6b23180a7
2 changed files with 33 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
oauth2client/tools.py
View File

@@ -99,9 +99,9 @@ class ClientRedirectHandler(BaseHTTPServer.BaseHTTPRequestHandler):
query = self.path.split('?', 1)[-1] query = self.path.split('?', 1)[-1]
query = dict(urllib.parse.parse_qsl(query)) query = dict(urllib.parse.parse_qsl(query))
self.server.query_params = query self.server.query_params = query
self.wfile.write("<html><head><title>Authentication Status</title></head>") self.wfile.write(b"<html><head><title>Authentication Status</title></head>")
self.wfile.write("<body><p>The authentication flow has completed.</p>") self.wfile.write(b"<body><p>The authentication flow has completed.</p>")
self.wfile.write("</body></html>") self.wfile.write(b"</body></html>")
def log_message(self, format, *args): def log_message(self, format, *args):
"""Do not log messages to stdout while running as command line program.""" """Do not log messages to stdout while running as command line program."""

30
tests/test_tools.py Normal file
View File

@@ -0,0 +1,30 @@
"""Unit tests for oauth2client.tools."""
import unittest
from oauth2client import tools
from six.moves.urllib import request
import threading
class TestClientRedirectServer(unittest.TestCase):
"""Test the ClientRedirectServer and ClientRedirectHandler classes."""
def test_ClientRedirectServer(self):
# create a ClientRedirectServer and run it in a thread to listen
# for a mock GET request with the access token
# the server should return a 200 message and store the token
httpd = tools.ClientRedirectServer(('localhost', 0), tools.ClientRedirectHandler)
code = 'foo'
url = 'http://localhost:%i?code=%s' % (httpd.server_address[1], code)
t = threading.Thread(target = httpd.handle_request)
t.setDaemon(True)
t.start()
f = request.urlopen( url )
self.assertTrue(f.read())
t.join()
httpd.server_close()
self.assertEqual(httpd.query_params.get('code'),code)
if __name__ == '__main__':
unittest.main()