def ConstructStage(self):
     sync_stage = sync_stages.ManifestVersionedSyncStage(
         self._run, self.buildstore)
     sync_stage.pool = mock.MagicMock()
     return completion_stages.PublishUprevChangesStage(self._run,
                                                       self.buildstore,
                                                       sync_stage,
                                                       success=True)
 def ConstructStage(self):
     """Returns a CanaryCompletionStage object."""
     sync_stage = sync_stages.ManifestVersionedSyncStage(
         self._run, self.buildstore)
     return completion_stages.CanaryCompletionStage(self._run,
                                                    self.buildstore,
                                                    sync_stage,
                                                    success=True)
    def _Prepare(self, bot_id=None, **kwargs):  # pylint: disable=arguments-differ
        super(ManifestVersionedSyncStageTest, self)._Prepare(bot_id, **kwargs)

        self._run.config['manifest_version'] = self.manifest_version_url
        self.sync_stage = sync_stages.ManifestVersionedSyncStage(
            self._run, self.buildstore)
        self.sync_stage.manifest_manager = self.manager
        self._run.attrs.manifest_manager = self.manager
  def ConstructStage(self):
    sync_stage = sync_stages.ManifestVersionedSyncStage(self._run)

    scheduled_slaves_list = {
        ('build_1', 'buildbucket_id_1', 1),
        ('build_2', 'buildbucket_id_2', 2)
    }
    self._run.attrs.metadata.ExtendKeyListWithList(
        constants.METADATA_SCHEDULED_IMPORTANT_SLAVES, scheduled_slaves_list)
    return completion_stages.MasterSlaveSyncCompletionStage(
        self._run, sync_stage, success=True)
 def ConstructStage(self):
     sync_stage = sync_stages.ManifestVersionedSyncStage(
         self._run, self.buildstore)
     return completion_stages.MasterSlaveSyncCompletionStage(
         self._run, self.buildstore, sync_stage, success=True)