Merge pull request googleapis#103 from craigcitro/master

Finish switching to mock from mox.

Author: craigcitro

tests/test_appengine.py

@@ -33,7 +33,7 @@
 
 import dev_appserver
 dev_appserver.fix_sys_path()
-import mox
+import mock
 import webapp2
 
 from google.appengine.api import apiproxy_stub
@@ -159,13 +159,9 @@ def test_raise_correct_type_of_exception(self):
       'memcache', memcache_stub.MemcacheServiceStub())
 
     scope = 'http://www.googleapis.com/scope'
-    try:
-      credentials = AppAssertionCredentials(scope)
-      http = httplib2.Http()
-      credentials.refresh(http)
-      self.fail('Should have raised an AccessTokenRefreshError')
-    except AccessTokenRefreshError:
-      pass
+    credentials = AppAssertionCredentials(scope)
+    http = httplib2.Http()
+    self.assertRaises(AccessTokenRefreshError, credentials.refresh, http)
 
   def test_get_access_token_on_refresh(self):
     app_identity_stub = self.AppIdentityStubImpl()
@@ -201,19 +197,19 @@ def test_get_access_token_on_refresh(self):
   def test_custom_service_account(self):
     scope = "http://www.googleapis.com/scope"
     account_id = "[email protected]"
-    m = mox.Mox()
-    m.StubOutWithMock(app_identity, 'get_access_token')
-    app_identity.get_access_token(
-        [scope], service_account_id=account_id).AndReturn(('a_token_456', None))
-    m.ReplayAll()
 
-    credentials = AppAssertionCredentials(scope, service_account_id=account_id)
-    http = httplib2.Http()
-    credentials.refresh(http)
-    m.VerifyAll()
-    m.UnsetStubs()
-    self.assertEqual('a_token_456', credentials.access_token)
-    self.assertEqual(scope, credentials.scope)
+    with mock.patch.object(app_identity, 'get_access_token',
+                           return_value=('a_token_456', None),
+                           autospec=True) as get_access_token:
+      credentials = AppAssertionCredentials(
+          scope, service_account_id=account_id)
+      http = httplib2.Http()
+      credentials.refresh(http)
+
+      self.assertEqual('a_token_456', credentials.access_token)
+      self.assertEqual(scope, credentials.scope)
+      get_access_token.assert_called_once_with(
+          [scope], service_account_id=account_id)
 
   def test_create_scoped_required_without_scopes(self):
     credentials = AppAssertionCredentials([])
@@ -538,7 +534,8 @@ def get(self, *args, **kwargs):
         'wsgi.url_scheme': 'http',
         'HTTP_HOST': 'localhost',
         })
-    users.get_current_user = user_mock()
+    self.current_user = user_mock()
+    users.get_current_user = self.current_user
     self.httplib2_orig = httplib2.Http
     httplib2.Http = Http2Mock
 
@@ -562,30 +559,28 @@ def test_required(self):
     self.assertEqual('code', q['response_type'][0])
     self.assertEqual(False, self.decorator.has_credentials())
 
-    m = mox.Mox()
-    m.StubOutWithMock(appengine, '_parse_state_value')
-    appengine._parse_state_value('foo_path:xsrfkey123',
-                       mox.IgnoreArg()).AndReturn('foo_path')
-    m.ReplayAll()
-
-    # Now simulate the callback to /oauth2callback.
-    response = self.app.get('/oauth2callback', {
-        'code': 'foo_access_code',
-        'state': 'foo_path:xsrfkey123',
-        })
-    parts = response.headers['Location'].split('?', 1)
-    self.assertEqual('http://localhost/foo_path', parts[0])
-    self.assertEqual(None, self.decorator.credentials)
-    if self.decorator._token_response_param:
-      response = urlparse.parse_qs(
-          parts[1])[self.decorator._token_response_param][0]
-      self.assertEqual(Http2Mock.content, json.loads(urllib.unquote(response)))
-    self.assertEqual(self.decorator.flow, self.decorator._tls.flow)
-    self.assertEqual(self.decorator.credentials,
-                     self.decorator._tls.credentials)
-
-    m.UnsetStubs()
-    m.VerifyAll()
+    with mock.patch.object(appengine, '_parse_state_value',
+                           return_value='foo_path',
+                           autospec=True) as parse_state_value:
+      # Now simulate the callback to /oauth2callback.
+      response = self.app.get('/oauth2callback', {
+          'code': 'foo_access_code',
+          'state': 'foo_path:xsrfkey123',
+          })
+      parts = response.headers['Location'].split('?', 1)
+      self.assertEqual('http://localhost/foo_path', parts[0])
+      self.assertEqual(None, self.decorator.credentials)
+      if self.decorator._token_response_param:
+        response_query = urlparse.parse_qs(parts[1])
+        response = response_query[self.decorator._token_response_param][0]
+        self.assertEqual(Http2Mock.content,
+                         json.loads(urllib.unquote(response)))
+      self.assertEqual(self.decorator.flow, self.decorator._tls.flow)
+      self.assertEqual(self.decorator.credentials,
+                       self.decorator._tls.credentials)
+
+      parse_state_value.assert_called_once_with(
+          'foo_path:xsrfkey123', self.current_user)
 
     # Now requesting the decorated path should work.
     response = self.app.get('/foo_path')
@@ -599,13 +594,9 @@ def test_required(self):
 
     # Raising an exception still clears the Credentials.
     self.should_raise = True
-    try:
-      response = self.app.get('/foo_path')
-      self.fail('Should have raised an exception.')
-    except Exception:
-      pass
-    self.assertEqual(None, self.decorator.credentials)
+    self.assertRaises(Exception, self.app.get, '/foo_path')
     self.should_raise = False
+    self.assertEqual(None, self.decorator.credentials)
 
     # Invalidate the stored Credentials.
     self.found_credentials.invalid = True
@@ -623,37 +614,34 @@ def test_storage_delete(self):
     response = self.app.get('/foo_path')
     self.assertTrue(response.status.startswith('302'))
 
-    m = mox.Mox()
-    m.StubOutWithMock(appengine, '_parse_state_value')
-    appengine._parse_state_value('foo_path:xsrfkey123',
-                       mox.IgnoreArg()).AndReturn('foo_path')
-    m.ReplayAll()
-
-    # Now simulate the callback to /oauth2callback.
-    response = self.app.get('/oauth2callback', {
-        'code': 'foo_access_code',
-        'state': 'foo_path:xsrfkey123',
-        })
-    self.assertEqual('http://localhost/foo_path', response.headers['Location'])
-    self.assertEqual(None, self.decorator.credentials)
-
-    # Now requesting the decorated path should work.
-    response = self.app.get('/foo_path')
+    with mock.patch.object(appengine, '_parse_state_value',
+                           return_value='foo_path',
+                           autospec=True) as parse_state_value:
+      # Now simulate the callback to /oauth2callback.
+      response = self.app.get('/oauth2callback', {
+          'code': 'foo_access_code',
+          'state': 'foo_path:xsrfkey123',
+      })
+      self.assertEqual('http://localhost/foo_path', response.headers['Location'])
+      self.assertEqual(None, self.decorator.credentials)
+
+      # Now requesting the decorated path should work.
+      response = self.app.get('/foo_path')
 
-    self.assertTrue(self.had_credentials)
+      self.assertTrue(self.had_credentials)
 
-    # Credentials should be cleared after each call.
-    self.assertEqual(None, self.decorator.credentials)
+      # Credentials should be cleared after each call.
+      self.assertEqual(None, self.decorator.credentials)
 
-    # Invalidate the stored Credentials.
-    self.found_credentials.store.delete()
+      # Invalidate the stored Credentials.
+      self.found_credentials.store.delete()
 
-    # Invalid Credentials should start the OAuth dance again.
-    response = self.app.get('/foo_path')
-    self.assertTrue(response.status.startswith('302'))
+      # Invalid Credentials should start the OAuth dance again.
+      response = self.app.get('/foo_path')
+      self.assertTrue(response.status.startswith('302'))
 
-    m.UnsetStubs()
-    m.VerifyAll()
+      parse_state_value.assert_called_once_with(
+          'foo_path:xsrfkey123', self.current_user)
 
   def test_aware(self):
     # An initial request to an oauth_aware decorated path should not redirect.
@@ -670,23 +658,20 @@ def test_aware(self):
                      q['state'][0].rsplit(':', 1)[0])
     self.assertEqual('code', q['response_type'][0])
 
-    m = mox.Mox()
-    m.StubOutWithMock(appengine, '_parse_state_value')
-    appengine._parse_state_value('bar_path:xsrfkey456',
-                       mox.IgnoreArg()).AndReturn('bar_path')
-    m.ReplayAll()
-
-    # Now simulate the callback to /oauth2callback.
-    url = self.decorator.authorize_url()
-    response = self.app.get('/oauth2callback', {
-        'code': 'foo_access_code',
-        'state': 'bar_path:xsrfkey456',
-        })
-    self.assertEqual('http://localhost/bar_path', response.headers['Location'])
-    self.assertEqual(False, self.decorator.has_credentials())
-
-    m.UnsetStubs()
-    m.VerifyAll()
+    with mock.patch.object(appengine, '_parse_state_value',
+                           return_value='bar_path',
+                           autospec=True) as parse_state_value:
+      # Now simulate the callback to /oauth2callback.
+      url = self.decorator.authorize_url()
+      response = self.app.get('/oauth2callback', {
+          'code': 'foo_access_code',
+          'state': 'bar_path:xsrfkey456',
+          })
+
+      self.assertEqual('http://localhost/bar_path', response.headers['Location'])
+      self.assertEqual(False, self.decorator.has_credentials())
+      parse_state_value.assert_called_once_with(
+          'bar_path:xsrfkey456', self.current_user)
 
     # Now requesting the decorated path will have credentials.
     response = self.app.get('/bar_path/2012/01')
@@ -703,13 +688,9 @@ def test_aware(self):
 
     # Raising an exception still clears the Credentials.
     self.should_raise = True
-    try:
-      response = self.app.get('/bar_path/2012/01')
-      self.fail('Should have raised an exception.')
-    except Exception:
-      pass
-    self.assertEqual(None, self.decorator.credentials)
+    self.assertRaises(Exception, self.app.get, '/bar_path/2012/01')
     self.should_raise = False
+    self.assertEqual(None, self.decorator.credentials)
 
 
   def test_error_in_step2(self):

tests/test_gce.py

@@ -20,13 +20,11 @@
 
 __author__ = '[email protected] (Joe Gregorio)'
 
-import httplib2
-try:
-  from mox3 import mox
-except ImportError:
-  import mox
 import unittest
 
+import httplib2
+import mock
+
 from oauth2client.client import AccessTokenRefreshError
 from oauth2client.client import Credentials
 from oauth2client.client import save_to_well_known_file
@@ -36,56 +34,29 @@
 class AssertionCredentialsTests(unittest.TestCase):
 
   def test_good_refresh(self):
-    m = mox.Mox()
-
-    httplib2_response = m.CreateMock(object)
-    httplib2_response.status = 200
-
-    httplib2_request = m.CreateMock(object)
-    httplib2_request.__call__(
-        ('http://metadata.google.internal/0.1/meta-data/service-accounts/'
-         'default/acquire'
-         '?scope=http%3A%2F%2Fexample.com%2Fa%20http%3A%2F%2Fexample.com%2Fb'
-        )).AndReturn((httplib2_response, '{"accessToken": "this-is-a-token"}'))
-
-    m.ReplayAll()
+    http = mock.MagicMock()
+    http.request = mock.MagicMock(
+        return_value=(mock.Mock(status=200),
+                      '{"accessToken": "this-is-a-token"}'))
 
     c = AppAssertionCredentials(scope=['http://example.com/a',
                                        'http://example.com/b'])
-
-    c._refresh(httplib2_request)
-
+    self.assertEquals(None, c.access_token)
+    c.refresh(http)
     self.assertEquals('this-is-a-token', c.access_token)
 
-    m.UnsetStubs()
-    m.VerifyAll()
+    http.request.assert_called_exactly_once_with(
+        'http://metadata.google.internal/0.1/meta-data/service-accounts/'
+        'default/acquire'
+        '?scope=http%3A%2F%2Fexample.com%2Fa%20http%3A%2F%2Fexample.com%2Fb')
 
   def test_fail_refresh(self):
-    m = mox.Mox()
-
-    httplib2_response = m.CreateMock(object)
-    httplib2_response.status = 400
-
-    httplib2_request = m.CreateMock(object)
-    httplib2_request.__call__(
-        ('http://metadata.google.internal/0.1/meta-data/service-accounts/'
-         'default/acquire'
-         '?scope=http%3A%2F%2Fexample.com%2Fa%20http%3A%2F%2Fexample.com%2Fb'
-        )).AndReturn((httplib2_response, '{"accessToken": "this-is-a-token"}'))
-
-    m.ReplayAll()
+    http = mock.MagicMock()
+    http.request = mock.MagicMock(return_value=(mock.Mock(status=400), '{}'))
 
     c = AppAssertionCredentials(scope=['http://example.com/a',
                                        'http://example.com/b'])
-
-    try:
-      c._refresh(httplib2_request)
-      self.fail('Should have raised exception on 400')
-    except AccessTokenRefreshError:
-      pass
-
-    m.UnsetStubs()
-    m.VerifyAll()
+    self.assertRaises(AccessTokenRefreshError, c.refresh, http)
 
   def test_to_from_json(self):
     c = AppAssertionCredentials(scope=['http://example.com/a',
@@ -111,30 +82,19 @@ def test_create_scoped(self):
     self.assertEqual('dummy_scope', new_credentials.scope)
 
   def test_get_access_token(self):
-    m = mox.Mox()
-
-    httplib2_response = m.CreateMock(object)
-    httplib2_response.status = 200
-
-    httplib2_request = m.CreateMock(object)
-    httplib2_request.__call__(
-        ('http://metadata.google.internal/0.1/meta-data/service-accounts/'
-         'default/acquire?scope=dummy_scope'
-        )).AndReturn((httplib2_response, '{"accessToken": "this-is-a-token"}'))
-
-    m.ReplayAll()
+    http = mock.MagicMock()
+    http.request = mock.MagicMock(
+        return_value=(mock.Mock(status=200),
+                      '{"accessToken": "this-is-a-token"}'))
 
     credentials = AppAssertionCredentials(['dummy_scope'])
-
-    http = httplib2.Http()
-    http.request = httplib2_request
-
     token = credentials.get_access_token(http=http)
     self.assertEqual('this-is-a-token', token.access_token)
     self.assertEqual(None, token.expires_in)
 
-    m.UnsetStubs()
-    m.VerifyAll()
+    http.request.assert_called_exactly_once_with(
+        'http://metadata.google.internal/0.1/meta-data/service-accounts/'
+        'default/acquire?scope=dummy_scope')
 
   def test_save_to_well_known_file(self):
     credentials = AppAssertionCredentials([])