| Index: tools/telemetry/telemetry/internal/story_runner_unittest.py
|
| diff --git a/tools/telemetry/telemetry/internal/story_runner_unittest.py b/tools/telemetry/telemetry/internal/story_runner_unittest.py
|
| index b0d7bab670aaa5feec9fbfb2de4f2c8b1c9b296e..378c7e0cd712124ec9c34166a4d82b4085d8f6d3 100644
|
| --- a/tools/telemetry/telemetry/internal/story_runner_unittest.py
|
| +++ b/tools/telemetry/telemetry/internal/story_runner_unittest.py
|
| @@ -2,6 +2,7 @@
|
| # Use of this source code is governed by a BSD-style license that can be
|
| # found in the LICENSE file.
|
|
|
| +import os
|
| import StringIO
|
| import sys
|
| import unittest
|
| @@ -9,6 +10,7 @@ import unittest
|
| from catapult_base import cloud_storage
|
| from telemetry import benchmark
|
| from telemetry.core import exceptions
|
| +from telemetry.core import util
|
| from telemetry import decorators
|
| from telemetry.internal.results import results_options
|
| from telemetry.internal import story_runner
|
| @@ -448,6 +450,9 @@ class StoryRunnerTest(unittest.TestCase):
|
| def testUpdateAndCheckArchives(self):
|
| usr_stub = system_stub.Override(story_runner, ['cloud_storage'])
|
| wpr_stub = system_stub.Override(archive_info, ['cloud_storage'])
|
| + archive_data_dir = os.path.join(
|
| + util.GetTelemetryDir(),
|
| + 'telemetry', 'internal', 'testing', 'archive_files')
|
| try:
|
| story_set = story_module.StorySet()
|
| story_set.AddStory(page_module.Page(
|
| @@ -473,7 +478,7 @@ class StoryRunnerTest(unittest.TestCase):
|
| story_set.stories)
|
|
|
| story_set = story_module.StorySet(
|
| - archive_data_file='../../unittest_data/archive_files/test.json',
|
| + archive_data_file=os.path.join(archive_data_dir, 'test.json'),
|
| cloud_storage_bucket=cloud_storage.PUBLIC_BUCKET)
|
| story_set.AddStory(page_module.Page(
|
| 'http://www.testurl.com', story_set, story_set.base_dir))
|
| @@ -492,7 +497,8 @@ class StoryRunnerTest(unittest.TestCase):
|
| story_set.stories)
|
|
|
| story_set = story_module.StorySet(
|
| - archive_data_file='../../unittest_data/test_missing_wpr_file.json',
|
| + archive_data_file=
|
| + os.path.join(archive_data_dir, 'test_missing_wpr_file.json'),
|
| cloud_storage_bucket=cloud_storage.PUBLIC_BUCKET)
|
| story_set.AddStory(page_module.Page(
|
| 'http://www.testurl.com', story_set, story_set.base_dir))
|
|
|