chromium/tools/perf/page_sets/companion/companion_stories.py

# Copyright 2023 The Chromium Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

from telemetry import story
from page_sets.companion import basic_companion_story


class CompanionStorySet(story.StorySet):
  COMPANION_STORIES = [
      basic_companion_story.CompanionStoryBasicOpen,
      basic_companion_story.CompanionStoryBasicOpenLoggedOut,
      basic_companion_story.CompanionStorySRP,
      basic_companion_story.CompanionStoryScreenshot
  ]

  def __init__(self):
    super(CompanionStorySet,
          self).__init__(cloud_storage_bucket=story.PARTNER_BUCKET)

    for cls in self.COMPANION_STORIES:
      self.AddStory(cls(self))