| Index: appengine/findit/handlers/test/monitor_alerts_test.py
|
| diff --git a/appengine/findit/handlers/test/monitor_alerts_test.py b/appengine/findit/handlers/test/monitor_alerts_test.py
|
| index 8b56e4e15501a11256cad231ea09aef211935bd9..7b2fa172f361d5d3114d72c79f54b965e87537c3 100644
|
| --- a/appengine/findit/handlers/test/monitor_alerts_test.py
|
| +++ b/appengine/findit/handlers/test/monitor_alerts_test.py
|
| @@ -6,12 +6,10 @@ import textwrap
|
|
|
| import webapp2
|
|
|
| -from testing_utils import testing
|
| -
|
| from common.http_client_appengine import RetryHttpClient
|
| from handlers import monitor_alerts
|
| from waterfall import build_failure_analysis_pipelines
|
| -from waterfall import waterfall_config
|
| +from waterfall.test import wf_configured_test_case
|
|
|
|
|
| class _MockedHttpClient(RetryHttpClient):
|
| @@ -30,7 +28,7 @@ class _MockedHttpClient(RetryHttpClient):
|
| pass
|
|
|
|
|
| -class MonitorAlertsTest(testing.AppengineTestCase):
|
| +class MonitorAlertsTest(wf_configured_test_case.WaterfallConfiguredTestCase):
|
| app_module = webapp2.WSGIApplication([
|
| ('/monitor-alerts', monitor_alerts.MonitorAlerts),
|
| ], debug=True)
|
| @@ -38,11 +36,6 @@ class MonitorAlertsTest(testing.AppengineTestCase):
|
| def setUp(self):
|
| super(MonitorAlertsTest, self).setUp()
|
|
|
| - def MockMasterIsSupported(master_name):
|
| - return master_name != 'm0'
|
| -
|
| - self.mock(waterfall_config, 'MasterIsSupported', MockMasterIsSupported)
|
| -
|
| def testGetLatestBuildFailuresWhenFailedToPullAlerts(self):
|
| http_client = _MockedHttpClient(404, 'Not Found')
|
| self.assertEqual([], monitor_alerts._GetLatestBuildFailures(http_client))
|
|
|