[sync] Serve fake OAuth2 tokens from python sync server

sync/tools/testserver/sync_testserver.py

 #!/usr/bin/env python
 # Copyright 2013 The Chromium Authors. All rights reserved.
 # Use of this source code is governed by a BSD-style license that can be
 # found in the LICENSE file.
 
 """This is a python sync server used for testing Chrome Sync.
 
 By default, it listens on an ephemeral port and xmpp_port and sends the port
 numbers back to the originating process over a pipe. The originating process can
 specify an explicit port and xmpp_port if necessary.
@@ skipping 133 matching lines @@
                     self.ChromiumSyncEnableNotificationsOpHandler,
                     self.ChromiumSyncSendNotificationOpHandler,
                     self.ChromiumSyncBirthdayErrorOpHandler,
                     self.ChromiumSyncTransientErrorOpHandler,
                     self.ChromiumSyncErrorOpHandler,
                     self.ChromiumSyncSyncTabFaviconsOpHandler,
                     self.ChromiumSyncCreateSyncedBookmarksOpHandler,
                     self.ChromiumSyncEnableKeystoreEncryptionOpHandler,
                     self.ChromiumSyncRotateKeystoreKeysOpHandler,
                     self.ChromiumSyncEnableManagedUserAcknowledgementHandler,
-                    self.ChromiumSyncEnablePreCommitGetUpdateAvoidanceHandler]
+                    self.ChromiumSyncEnablePreCommitGetUpdateAvoidanceHandler,
+                    self.GaiaOAuth2TokenHandler,
+                    self.GaiaSetOAuth2TokenResponseHandler]
 
     post_handlers = [self.ChromiumSyncCommandHandler,
-                     self.ChromiumSyncTimeHandler]
+                     self.ChromiumSyncTimeHandler,
+                     self.GaiaOAuth2TokenHandler,
+                     self.GaiaSetOAuth2TokenResponseHandler]
     testserver_base.BasePageHandler.__init__(self, request, client_address,
                                              sync_http_server, [], get_handlers,
                                              [], post_handlers, [])
 
 
   def ChromiumSyncTimeHandler(self):
     """Handle Chromium sync .../time requests.
 
     The syncer sometimes checks server reachability by examining /time.
     """
@@ skipping 266 matching lines @@
       return False
     result, raw_reply = (
         self.server._sync_handler.HandleEnablePreCommitGetUpdateAvoidance())
     self.send_response(result)
     self.send_header('Content-Type', 'text/html')
     self.send_header('Content-Length', len(raw_reply))
     self.end_headers()
     self.wfile.write(raw_reply)
     return True
 
+  def GaiaOAuth2TokenHandler(self):
+    test_name = "/o/oauth2/token"
+    if not self._ShouldHandleRequest(test_name):
+      return False
+    if self.headers.getheader('content-length'):
+      length = int(self.headers.getheader('content-length'))
+      _raw_request = self.rfile.read(length)
+    result, raw_reply = (
+        self.server._sync_handler.HandleGetOauth2Token())
+    self.send_response(result)
+    self.send_header('Content-Type', 'application/json')
+    self.send_header('Content-Length', len(raw_reply))
+    self.end_headers()
+    self.wfile.write(raw_reply)
+    return True
+
+  def GaiaSetOAuth2TokenResponseHandler(self):
+    test_name = "/setfakeoauth2token"
+    if not self._ShouldHandleRequest(test_name):
+      return False
+
+    # The index of 'query' is 4.
+    # See http://docs.python.org/2/library/urlparse.html
+    query = urlparse.urlparse(self.path)[4]
+    query_params = urlparse.parse_qs(query)
+
+    response_code = 0
+    request_token = ''
+    access_token = ''
+    expires_in = 0
+    token_type = ''
+
+    if 'response_code' in query_params:
+      response_code = query_params['response_code'][0]
+    if 'request_token' in query_params:
+      request_token = query_params['request_token'][0]
+    if 'access_token' in query_params:
+      access_token = query_params['access_token'][0]
+    if 'expires_in' in query_params:
+      expires_in = query_params['expires_in'][0]
+    if 'token_type' in query_params:
+      token_type = query_params['token_type'][0]
+
+    result, raw_reply = (
+        self.server._sync_handler.HandleSetOauth2Token(
+            response_code, request_token, access_token, expires_in, token_type))
+    self.send_response(result)
+    self.send_header('Content-Type', 'text/html')
+    self.send_header('Content-Length', len(raw_reply))
+    self.end_headers()
+    self.wfile.write(raw_reply)
+    return True
+
+
 class SyncServerRunner(testserver_base.TestServerRunner):
   """TestServerRunner for the net test servers."""
 
   def __init__(self):
     super(SyncServerRunner, self).__init__()
 
   def create_server(self, server_data):
     port = self.options.port
     host = self.options.host
     xmpp_port = self.options.xmpp_port
     server = SyncHTTPServer((host, port), xmpp_port, SyncPageHandler)
-    print 'Sync HTTP server started on port %d...' % server.server_port
-    print 'Sync XMPP server started on port %d...' % server.xmpp_port
+    print ('Sync HTTP server started at %s:%d/chromiumsync...' %
+           (host, server.server_port))
+    print ('Fake OAuth2 Token server started at %s:%d/o/oauth2/token...' %
+           (host, server.server_port))
+    print ('Sync XMPP server started at %s:%d...' %
+           (host, server.xmpp_port))
     server_data['port'] = server.server_port
     server_data['xmpp_port'] = server.xmpp_port
     return server
 
   def run_server(self):
     testserver_base.TestServerRunner.run_server(self)
 
   def add_options(self):
     testserver_base.TestServerRunner.add_options(self)
     self.option_parser.add_option('--xmpp-port', default='0', type='int',
                                   help='Port used by the XMPP server. If '
                                   'unspecified, the XMPP server will listen on '
                                   'an ephemeral port.')
     # Override the default logfile name used in testserver.py.
     self.option_parser.set_defaults(log_file='sync_testserver.log')
 
 if __name__ == '__main__':
   sys.exit(SyncServerRunner().main())