9641c315ca
This is so we get 100% line coverage in our tests. Also adding unittest.main() lines to the files missing it: test__helpers.py test__pycrypto_crypt.py test_crypt.py test_devshell.py test_gce.py test_keyring_storage.py test_service_account.py test_util.py
32 lines
1.1 KiB
Python
32 lines
1.1 KiB
Python
"""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__': # pragma: NO COVER
|
|
unittest.main()
|