| Index: appengine/findit/waterfall/flake/test/recursive_flake_pipeline_test.py
|
| diff --git a/appengine/findit/waterfall/flake/test/recursive_flake_pipeline_test.py b/appengine/findit/waterfall/flake/test/recursive_flake_pipeline_test.py
|
| index 617ce22052834b2797d3dbf229370f04126f0412..9b43ef2a6138de4110792aa6418f228ad76e2035 100644
|
| --- a/appengine/findit/waterfall/flake/test/recursive_flake_pipeline_test.py
|
| +++ b/appengine/findit/waterfall/flake/test/recursive_flake_pipeline_test.py
|
| @@ -60,7 +60,7 @@ class RecursiveFlakePipelineTest(wf_testcase.WaterfallTestCase):
|
| mocked_utc_now = datetime(2016, 9, 04, 20, 0, 0, 0)
|
| self.MockUTCNow(mocked_utc_now)
|
| self.MockUTCNowWithTimezone(mocked_utc_now)
|
| - with mock.patch('common.time_util.GetDatetimeInTimezone') as timezone_func:
|
| + with mock.patch('lib.time_util.GetDatetimeInTimezone') as timezone_func:
|
| timezone_func.side_effect = [mocked_pst_now, None]
|
| self.assertEqual(mocked_utc_now,
|
| recursive_flake_pipeline._GetETAToStartAnalysis(False))
|
| @@ -71,7 +71,7 @@ class RecursiveFlakePipelineTest(wf_testcase.WaterfallTestCase):
|
| mocked_utc_now = datetime(2016, 9, 20, 8, 0, 0, 0)
|
| self.MockUTCNow(mocked_utc_now)
|
| self.MockUTCNowWithTimezone(mocked_utc_now)
|
| - with mock.patch('common.time_util.GetDatetimeInTimezone') as timezone_func:
|
| + with mock.patch('lib.time_util.GetDatetimeInTimezone') as timezone_func:
|
| timezone_func.side_effect = [mocked_pst_now, None]
|
| self.assertEqual(mocked_utc_now,
|
| recursive_flake_pipeline._GetETAToStartAnalysis(False))
|
| @@ -86,7 +86,7 @@ class RecursiveFlakePipelineTest(wf_testcase.WaterfallTestCase):
|
| mocked_utc_eta = datetime(2016, 9, 21, 1, 0, 0, 0) # Without delay.
|
| self.MockUTCNow(mocked_utc_now)
|
| self.MockUTCNowWithTimezone(mocked_utc_now)
|
| - with mock.patch('common.time_util.GetDatetimeInTimezone') as (
|
| + with mock.patch('lib.time_util.GetDatetimeInTimezone') as (
|
| timezone_func), mock.patch('random.randint') as random_func:
|
| timezone_func.side_effect = [mocked_pst_now, mocked_utc_eta]
|
| random_func.side_effect = [seconds_delay, None]
|
|
|