| Index: Tools/GardeningServer/alerts_test.py
|
| diff --git a/Tools/GardeningServer/alerts_test.py b/Tools/GardeningServer/alerts_test.py
|
| index 9933d738978d23d49799a5141702a20da6fdfaed..ed9f7779ef98e543944a5837e7c363fda91dd9db 100644
|
| --- a/Tools/GardeningServer/alerts_test.py
|
| +++ b/Tools/GardeningServer/alerts_test.py
|
| @@ -18,7 +18,6 @@
|
| def setUp(self):
|
| self.testbed = testbed.Testbed()
|
| self.testbed.activate()
|
| - self.testbed.init_user_stub()
|
| self.testbed.init_memcache_stub()
|
| self.testbed.init_datastore_v3_stub()
|
| self.testapp = webtest.TestApp(alerts.app)
|
| @@ -76,8 +75,6 @@
|
| alerts_query = alerts.AlertsJSON.query().order(alerts.AlertsJSON.date)
|
| stored_alerts = alerts_query.fetch(limit=3)
|
| self.assertEqual(2, len(stored_alerts))
|
| - self.assertEqual(stored_alerts[0].type, 'alerts')
|
| - self.assertEqual(stored_alerts[1].type, 'alerts')
|
| stored_alerts1 = json.loads(stored_alerts[0].json)
|
| stored_alerts2 = json.loads(stored_alerts[1].json)
|
| self.assertEqual(test_alerts1['alerts'], stored_alerts1['alerts'])
|
| @@ -94,67 +91,6 @@
|
| self.testapp.post('/alerts', {'content': json.dumps(test_alerts)})
|
| stored_alerts = alerts.AlertsJSON.query().fetch(limit=2)
|
| self.assertEqual(1, len(stored_alerts))
|
| -
|
| - def test_alerts_jsons_are_retrieved_from_history(self):
|
| - test_alert = {'alerts': ['hello', 'world', '1']}
|
| - alerts.AlertsJSON(json=json.dumps(test_alert), type='alerts').put()
|
| - response = self.testapp.get('/alerts-history')
|
| - self.assertEqual(response.status_int, 200)
|
| - self.assertEqual(response.content_type, 'application/json')
|
| - response_json = json.loads(response.normal_body)
|
| - self.assertEqual(len(response_json['history']), 1)
|
| - self.assertEqual(response_json['history'][0], test_alert)
|
| -
|
| - def test_internal_alerts_in_history_visible_to_internal_users_only(self):
|
| - test_alert = {'alerts': ['hello', 'world', '1']}
|
| - alerts.AlertsJSON(json=json.dumps(test_alert),
|
| - type='internal-alerts').put()
|
| -
|
| - # No signed-in user.
|
| - response = self.testapp.get('/alerts-history')
|
| - self.assertEqual(response.status_int, 200)
|
| - self.assertEqual(response.content_type, 'application/json')
|
| - response_json = json.loads(response.normal_body)
|
| - self.assertEqual(len(response_json['history']), 0)
|
| -
|
| - # Non-internal user.
|
| - self.testbed.setup_env(USER_EMAIL='test@example.com', USER_ID='1',
|
| - USER_IS_ADMIN='1', overwrite=True)
|
| - response = self.testapp.get('/alerts-history')
|
| - self.assertEqual(response.status_int, 200)
|
| - self.assertEqual(response.content_type, 'application/json')
|
| - response_json = json.loads(response.normal_body)
|
| - self.assertEqual(len(response_json['history']), 0)
|
| -
|
| - # Internal user.
|
| - self.testbed.setup_env(USER_EMAIL='test@google.com', USER_ID='2',
|
| - USER_IS_ADMIN='1', overwrite=True)
|
| - response = self.testapp.get('/alerts-history')
|
| - self.assertEqual(response.status_int, 200)
|
| - self.assertEqual(response.content_type, 'application/json')
|
| - response_json = json.loads(response.normal_body)
|
| - self.assertEqual(len(response_json['history']), 1)
|
| - self.assertEqual(response_json['history'][0], test_alert)
|
| -
|
| - def test_returned_alerts_from_history_are_paged(self):
|
| - for i in range(20):
|
| - test_alert = {'alerts': ['hello', 'world', i]}
|
| - alerts.AlertsJSON(json=json.dumps(test_alert), type='alerts').put()
|
| -
|
| - response = self.testapp.get('/alerts-history?limit=15')
|
| - self.assertEqual(response.status_int, 200)
|
| - self.assertEqual(response.content_type, 'application/json')
|
| - response_json = json.loads(response.normal_body)
|
| - self.assertEqual(len(response_json['history']), 15)
|
| - self.assertEqual(response_json['has_more'], True)
|
| -
|
| - url = '/alerts-history?limit=15&cursor=%s' % response_json['cursor']
|
| - response = self.testapp.get(url)
|
| - self.assertEqual(response.status_int, 200)
|
| - self.assertEqual(response.content_type, 'application/json')
|
| - response_json = json.loads(response.normal_body)
|
| - self.assertEqual(len(response_json['history']), 5)
|
| - self.assertEqual(response_json['has_more'], False)
|
|
|
| def test_large_number_of_alerts(self):
|
| # This generates ~2.5MB of JSON that compresses to ~750K. Real
|
|
|