Add request-level caching for privileged queries.

dashboard/dashboard/models/internal_only_model_test.py

 # Copyright 2015 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.
 
 import unittest
 
 import mock
 
 from google.appengine.ext import ndb
 
@@ skipping 15 matching lines @@
   def CreateCallback(self):
     pass
 
 
 class InternalOnlyModelTest(testing_common.TestCase):
 
   def setUp(self):
     super(InternalOnlyModelTest, self).setUp()
     testing_common.SetIsInternalUser('x@google.com', True)
     testing_common.SetIsInternalUser('x@foo.com', False)
+    self.PatchDatastoreHooksRequest()
 
   def testInternalOnlyModel_InternalUser_EntityFetched(self):
     key = InternalOnlyModelExample(internal_only=True).put()
     self.SetCurrentUser('x@google.com')
     self.assertEqual(key, key.get().key)
 
   def testInternalOnlyModel_ExternalUserInternalEntity_AssertionRaised(self):
     key = InternalOnlyModelExample(internal_only=True).put()
     self.SetCurrentUser('x@foo.com')
     with self.assertRaises(AssertionError):
@@ skipping 30 matching lines @@
       key.get()
 
   def testCreateHookEntity_ExternalUserExternalEntity_EntityFetched(self):
     key = CreateHookModelExample(internal_only=False).put()
     self.SetCurrentUser('x@foo.com')
     self.assertEqual(key, key.get().key)
 
 
 if __name__ == '__main__':
   unittest.main()