| Index: appengine/findit/waterfall/flake/test/initialize_flake_pipeline_test.py
|
| diff --git a/appengine/findit/waterfall/flake/test/initialize_flake_pipeline_test.py b/appengine/findit/waterfall/flake/test/initialize_flake_pipeline_test.py
|
| index 5afe7ed0300ec533e34ce27e25808963bd8be403..3b8c8f669463108785e1349a6d4a889d5aa00d82 100644
|
| --- a/appengine/findit/waterfall/flake/test/initialize_flake_pipeline_test.py
|
| +++ b/appengine/findit/waterfall/flake/test/initialize_flake_pipeline_test.py
|
| @@ -22,7 +22,7 @@ class InitializeFlakePipelineTest(wf_testcase.WaterfallTestCase):
|
| analysis = MasterFlakeAnalysis.Create(
|
| master_name, builder_name, build_number, step_name, test_name)
|
| analysis.status = status
|
| - analysis.put()
|
| + analysis.Save()
|
|
|
| def testAnalysisIsNotNeededWhenNoneExistsAndNotAllowedToSchedule(self):
|
| master_name = 'm'
|
| @@ -118,8 +118,8 @@ class InitializeFlakePipelineTest(wf_testcase.WaterfallTestCase):
|
| step_name, test_name, allow_new_analysis=True,
|
| queue_name=constants.DEFAULT_QUEUE)
|
|
|
| - analysis = MasterFlakeAnalysis.Get(master_name, builder_name,
|
| - build_number, step_name, test_name)
|
| + analysis = MasterFlakeAnalysis.GetVersion(
|
| + master_name, builder_name, build_number, step_name, test_name)
|
| self.assertIsNotNone(analysis)
|
| mocked_pipeline.assert_has_calls(
|
| [mock.call().start(queue_name=constants.DEFAULT_QUEUE)])
|
|
|