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 setUp(self):
super(PostBisectResultsTest, self).setUp()
app = webapp2.WSGIApplication([
('/post_bisect_results',
post_bisect_results.PostBisectResultsHandler)
])
self.testapp = webtest.TestApp(app)
testing_common.SetIpWhitelist([_WHITELISTED_IP])
self._AddRietveldConfig()
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')
def testGet_NoTestPathGiven_GivesError(self):
testing_common.SetIpWhitelist(['123.45.67.89'])
self.testapp.get(
'/graph_csv',
extra_environ={'REMOTE_ADDR': '123.45.67.89'},
status=400)
def testPost_IPInWhiteList_Authorized(self):
testing_common.SetIpWhitelist(['123.45.67.89', '98.76.54.32'])
self.testapp.post('/whitelist_test',
{'data': json.dumps([_SAMPLE_POINT])},
extra_environ={'REMOTE_ADDR': '123.45.67.89'})