| # Copyright (c) 2012 The Chromium OS Authors. All rights reserved. |
| # Use of this source code is governed by a BSD-style license that can be |
| # found in the LICENSE file. |
| |
| """GoogleAuthServer unittest.""" |
| |
| import os, unittest, urllib |
| |
| import constants as chromeos_constants |
| from auth_server import GoogleAuthServer |
| |
| class test_auth_server(unittest.TestCase): |
| def setUp(self): |
| print "starting" |
| self._normal_port = 50020 |
| self._ssl_port = 50030 |
| creds_path = (os.path.dirname(os.path.realpath( __file__)) + |
| '/httpd_unittest_server') |
| |
| self._test_server = GoogleAuthServer(port=self._normal_port, |
| ssl_port=self._ssl_port, |
| cert_path=(creds_path+'.pem'), |
| key_path=(creds_path+'.key')) |
| self._test_server.run() |
| |
| |
| def tearDown(self): |
| print "tearing down" |
| self._test_server.stop() |
| |
| |
| def test_client_login(self): |
| args = urllib.urlencode({'Email': 'me@example.com', 'Passwd': 'fake'}) |
| try: |
| cl_resp = urllib.urlopen( |
| 'https://localhost:%s%s' % |
| (self._ssl_port, chromeos_constants.CLIENT_LOGIN_URL), |
| args).read() |
| except IOError: |
| pass |
| self._test_server.wait_for_client_login() |
| return cl_resp |
| |
| |
| def test_issue_token(self): |
| args = self.test_client_login().replace('\n','&') |
| try: |
| it_resp = urllib.urlopen('https://localhost:%s%s' % |
| (self._ssl_port, |
| chromeos_constants.ISSUE_AUTH_TOKEN_URL), |
| args).read() |
| except IOError: |
| pass |
| self._test_server.wait_for_issue_token() |
| return it_resp |
| |
| |
| def test_token_auth(self): |
| args = self.test_issue_token() |
| try: |
| ta_format = ('https://localhost:%s%s?auth=%s&' + |
| 'continue=https://localhost:%s/webhp') |
| urllib.urlopen( |
| ta_format % (self._ssl_port, |
| chromeos_constants.TOKEN_AUTH_URL, |
| args, |
| self._ssl_port)).read() |
| except IOError: |
| pass |
| self._test_server.wait_for_test_over() |
