Index: appengine/findit/handlers/test/handlers_util_test.py |
diff --git a/appengine/findit/handlers/test/handlers_util_test.py b/appengine/findit/handlers/test/handlers_util_test.py |
index f0c6230339b779a0fb896fee14c9f4bc7ebdde75..b0def529254819b7c53ae2a7838acd976cc84f9d 100644 |
--- a/appengine/findit/handlers/test/handlers_util_test.py |
+++ b/appengine/findit/handlers/test/handlers_util_test.py |
@@ -2,18 +2,15 @@ |
# Use of this source code is governed by a BSD-style license that can be |
# found in the LICENSE file. |
-from testing_utils import testing |
- |
from handlers import handlers_util |
-from model.wf_analysis import WfAnalysis |
from model import wf_analysis_status |
+from model.wf_analysis import WfAnalysis |
from model.wf_swarming_task import WfSwarmingTask |
from model.wf_try_job import WfTryJob |
-from waterfall import buildbot |
-from waterfall import waterfall_config |
+from waterfall.test import wf_testcase |
-class HandlersUtilResultTest(testing.AppengineTestCase): |
+class HandlersUtilResultTest(wf_testcase.WaterfallTestCase): |
def setUp(self): |
super(HandlersUtilResultTest, self).setUp() |
@@ -21,11 +18,6 @@ class HandlersUtilResultTest(testing.AppengineTestCase): |
self.builder_name = 'b' |
self.build_number = 121 |
- def MockedGetSwarmingSettings(): |
- return {'server_host': 'chromium-swarm.appspot.com'} |
- self.mock( |
- waterfall_config, 'GetSwarmingSettings', MockedGetSwarmingSettings) |
- |
def testGenerateSwarmingTasksDataNoAnalysis(self): |
data = handlers_util.GenerateSwarmingTasksData( |
self.master_name, self.builder_name, self.build_number) |