Index: appengine/findit/waterfall/test/analyze_build_failure_pipeline_test.py |
diff --git a/appengine/findit/waterfall/test/analyze_build_failure_pipeline_test.py b/appengine/findit/waterfall/test/analyze_build_failure_pipeline_test.py |
index 0978af793332859474c42c160f554a172cc28169..f728c3e71897235841aab9e3a26d348951f25469 100644 |
--- a/appengine/findit/waterfall/test/analyze_build_failure_pipeline_test.py |
+++ b/appengine/findit/waterfall/test/analyze_build_failure_pipeline_test.py |
@@ -4,19 +4,17 @@ |
import os |
-from testing_utils import testing |
- |
from common import chromium_deps |
-from model.wf_analysis import WfAnalysis |
from model import wf_analysis_status |
+from model.test import configured_test_case |
+from model.wf_analysis import WfAnalysis |
from pipeline_wrapper import pipeline_handlers |
from waterfall import buildbot |
-from waterfall.analyze_build_failure_pipeline import AnalyzeBuildFailurePipeline |
from waterfall import lock_util |
-from waterfall import waterfall_config |
+from waterfall.analyze_build_failure_pipeline import AnalyzeBuildFailurePipeline |
-class AnalyzeBuildFailurePipelineTest(testing.AppengineTestCase): |
+class AnalyzeBuildFailurePipelineTest(configured_test_case.ConfiguredTestCase): |
app_module = pipeline_handlers._APP |
def _MockChangeLog( |
@@ -104,13 +102,6 @@ class AnalyzeBuildFailurePipelineTest(testing.AppengineTestCase): |
self._Setup(master_name, builder_name, build_number) |
- def MockStepIsSupportedForMaster(*_): |
- return True |
- |
- self.mock(waterfall_config, |
- 'StepIsSupportedForMaster', |
- MockStepIsSupportedForMaster) |
- |
root_pipeline = AnalyzeBuildFailurePipeline(master_name, |
builder_name, |
build_number, |