In /graph_json, check internal_only on Test entity and skip test for all Row entities.

dashboard/dashboard/graph_json_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 json
 import unittest
 
 import mock
 import webapp2
 import webtest
 
 from google.appengine.ext import ndb
 
 from dashboard import graph_json
 from dashboard import testing_common
 from dashboard import utils
 from dashboard.models import anomaly
 from dashboard.models import graph_data
 
 
 class GraphJsonTest(testing_common.TestCase):
 
   def setUp(self):
     super(GraphJsonTest, self).setUp()
     app = webapp2.WSGIApplication(
         [('/graph_json', graph_json.GraphJsonHandler)])
     self.testapp = webtest.TestApp(app)
     testing_common.SetInternalDomain('google.com')
+    self.PatchDatastoreHooksRequest()
 
   # TODO(qyearsley): graph_json_test is very slow (it takes 60+ seconds
   # to run sometimes), and I have a hypothesis that most of the time is
   # spent in _AddTestColumns. Investigate this hypothesis and make changes
   # accordingly.
   def _AddTestColumns(self, start_rev=15000, end_rev=16500, step=3):
     """Adds a bunch of test data to the mock datastore.
 
     In particular, add Rows with revisions in the given range (but skipping
     some numbers, so the revisions are non-contiguous) under the dromaeo/dom
@@ skipping 756 matching lines @@
     rows = testing_common.AddRows('Master/b/my_suite', [345])
     # This row has no a_tracing_uri property, so there should be no
     # trace annotation returned by _PointInfoDict.
     point_info = graph_json._PointInfoDict(rows[0], test, {})
     self.assertFalse(hasattr(rows[0], 'a_tracing_uri'))
     self.assertNotIn('a_tracing_uri', point_info)
 
 
 if __name__ == '__main__':
   unittest.main()