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..865bbb26139ee544f92d8a49370909dcb940e144 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_testcase |
class _MockedHttpClient(RetryHttpClient): |
@@ -30,7 +28,7 @@ class _MockedHttpClient(RetryHttpClient): |
pass |
-class MonitorAlertsTest(testing.AppengineTestCase): |
+class MonitorAlertsTest(wf_testcase.WaterfallTestCase): |
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)) |