def setUp(self):
super(DatastoreHooksTest, self).setUp()
testing_common.SetIsInternalUser('*****@*****.**', True)
testing_common.SetIsInternalUser('*****@*****.**', False)
self._AddDataToDatastore()
datastore_hooks.InstallHooks()
self.PatchDatastoreHooksRequest()
def testGet_WithNonInternalUserAndWhitelistedIP(self): self._AddMockInternalData() self.UnsetCurrentUser() datastore_hooks.InstallHooks() testing_common.SetIpWhitelist(['123.45.67.89']) query = '/graph_csv?test_path=ChromiumPerf/win7/dromaeo/dom&num_points=3' expected = [['revision', 'value']] self._CheckGet(query, expected)
def testGet_InternalTestDataUnauthorized_DoesntShowData(self):
"""Tests a query for internal data for an unauthorized user."""
self._AddInternalSampleData()
self.UnsetCurrentUser()
datastore_hooks.InstallHooks()
response = self.testapp.get('/new_points')
# Only the header row is listed, not any others rows.
self.assertEqual(1, len(re.findall(r'<tr>', response.body)))
def setUp(self):
super(ListTestsTest, self).setUp()
app = webapp2.WSGIApplication([('/list_tests',
list_tests.ListTestsHandler)])
self.testapp = webtest.TestApp(app)
datastore_hooks.InstallHooks()
self.UnsetCurrentUser()
testing_common.SetInternalDomain('google.com')
def setUp(self):
super(ListTestsTest, self).setUp()
app = webapp2.WSGIApplication([('/list_tests',
list_tests.ListTestsHandler)])
self.testapp = webtest.TestApp(app)
datastore_hooks.InstallHooks()
self.UnsetCurrentUser()
testing_common.SetIsInternalUser('*****@*****.**', True)
testing_common.SetIsInternalUser('*****@*****.**', False)
def setUp(self):
super(DatastoreHooksTest, self).setUp()
testing_common.SetInternalDomain('google.com')
self._AddDataToDatastore()
datastore_hooks.InstallHooks()
get_request_patcher = mock.patch(
'webapp2.get_request', mock.MagicMock(return_value=FakeRequest()))
self.mock_get_request = get_request_patcher.start()
self.addCleanup(get_request_patcher.stop)
def setUp(self):
super(ListTestSuitesTest, self).setUp()
app = webapp2.WSGIApplication(
[('/update_test_suites',
update_test_suites.UpdateTestSuitesHandler)])
self.testapp = webtest.TestApp(app)
datastore_hooks.InstallHooks()
testing_common.SetIsInternalUser('*****@*****.**', True)
self.UnsetCurrentUser()
def testGet_InternalTestDataAuthorized_ShowsData(self):
"""Tests a query for internal data when the user should be authorized."""
self._AddInternalSampleData()
# The user doesn't need to be authorized as admin to view internal data,
# they only need to have an internal email address.
self.SetCurrentUser('*****@*****.**')
datastore_hooks.InstallHooks()
response = self.testapp.get('/new_points')
# 50 rows for xtest, 1 for the header.
self.assertEqual(51, len(re.findall(r'<tr>', response.body)))
def testGet_WhitelistedIPOnly(self):
self.PatchDatastoreHooksRequest('123.45.67.89')
self._AddMockInternalData()
self.UnsetCurrentUser()
datastore_hooks.InstallHooks()
testing_common.SetIpWhitelist(['123.45.67.89'])
query = '/graph_csv?test_path=ChromiumPerf/win7/dromaeo/dom&num_points=3'
expected = [
['revision', 'value'],
['47', '94.0'],
['48', '96.0'],
['49', '98.0']
]
self._CheckGet(query, expected, whitelisted_ip='123.45.67.89')
This module is loaded before others and can be used to set up the
App Engine environment. See:
https://cloud.google.com/appengine/docs/python/tools/appengineconfig
"""
import logging
import os
import sys
from google.appengine.ext import vendor
import dashboard
from dashboard import datastore_hooks
datastore_hooks.InstallHooks()
appstats_SHELL_OK = True
# Allows remote_api from the peng team to support the crosbolt dashboard.
remoteapi_CUSTOM_ENVIRONMENT_AUTHENTICATION = ('LOAS_PEER_USERNAME',
['chromeos-peng-performance'])
def _AddThirdPartyLibraries():
"""Registers the third party libraries with App Engine.
In order for third-party libraries to be available in the App Engine
runtime environment, they must be added with vendor.add. The directories
added this way must be inside the App Engine project directory.
"""
