Add BuildBucket v2 migration test support files (Part 2)

app_dart/test/src/service/fake_luci_build_service.dart

@@ -4,13 +4,15 @@
 
 import 'package:cocoon_service/src/foundation/github_checks_util.dart';
 import 'package:cocoon_service/src/request_handling/pubsub.dart';
+import 'package:cocoon_service/src/service/build_bucket_v2_client.dart';
 import 'package:cocoon_service/src/service/buildbucket.dart';
 import 'package:cocoon_service/src/service/cache_service.dart';
 import 'package:cocoon_service/src/service/gerrit_service.dart';
 import 'package:cocoon_service/src/service/luci_build_service.dart';
 
 import '../request_handling/fake_pubsub.dart';
 import '../utilities/mocks.dart';
+import 'fake_build_bucket_v2_client.dart';
 import 'fake_buildbucket.dart';
 import 'fake_gerrit_service.dart';
 
@@ -19,12 +21,14 @@ class FakeLuciBuildService extends LuciBuildService {
   FakeLuciBuildService({
     required super.config,
     BuildBucketClient? buildbucket,
+    BuildBucketV2Client? buildBucketV2Client,
     GithubChecksUtil? githubChecksUtil,
     GerritService? gerritService,
     PubSub? pubsub,
   }) : super(
           cache: CacheService(inMemory: true),
           buildBucketClient: buildbucket ?? FakeBuildBucketClient(),
+          buildBucketV2Client: buildBucketV2Client ?? FakeBuildBucketV2Client(),
           githubChecksUtil: githubChecksUtil ?? MockGithubChecksUtil(),
           gerritService: gerritService ?? FakeGerritService(),
           pubsub: pubsub ?? FakePubSub(),

app_dart/test/src/service/fake_luci_build_service_v2.dart

@@ -0,0 +1,32 @@
+// Copyright 2021 The Flutter Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+import 'package:cocoon_service/src/foundation/github_checks_util.dart';
+import 'package:cocoon_service/src/request_handling/pubsub.dart';
+import 'package:cocoon_service/src/service/build_bucket_v2_client.dart';
+import 'package:cocoon_service/src/service/cache_service.dart';
+import 'package:cocoon_service/src/service/gerrit_service.dart';
+import 'package:cocoon_service/src/service/luci_build_service_v2.dart';
+
+import '../request_handling/fake_pubsub.dart';
+import '../utilities/mocks.dart';
+import 'fake_build_bucket_v2_client.dart';
+import 'fake_gerrit_service.dart';
+
+/// Fake [LuciBuildService] for use in tests.
+class FakeLuciBuildServiceV2 extends LuciBuildServiceV2 {
+  FakeLuciBuildServiceV2({
+    required super.config,
+    BuildBucketV2Client? buildBucketV2Client,
+    GithubChecksUtil? githubChecksUtil,
+    GerritService? gerritService,
+    PubSub? pubsub,
+  }) : super(
+          cache: CacheService(inMemory: true),
+          buildBucketV2Client: buildBucketV2Client ?? FakeBuildBucketV2Client(),
+          githubChecksUtil: githubChecksUtil ?? MockGithubChecksUtil(),
+          gerritService: gerritService ?? FakeGerritService(),
+          pubsub: pubsub ?? FakePubSub(),
+        );
+}

app_dart/test/src/service/fake_scheduler_v2.dart

@@ -0,0 +1,307 @@
+// Copyright 2021 The Flutter Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+import 'package:cocoon_service/src/foundation/github_checks_util.dart';
+import 'package:cocoon_service/src/model/appengine/commit.dart';
+import 'package:cocoon_service/src/model/ci_yaml/ci_yaml.dart';
+import 'package:cocoon_service/src/model/proto/protos.dart' as pb;
+import 'package:cocoon_service/src/service/build_bucket_v2_client.dart';
+import 'package:cocoon_service/src/service/cache_service.dart';
+import 'package:cocoon_service/src/service/config.dart';
+import 'package:cocoon_service/src/service/github_checks_service_v2.dart';
+import 'package:cocoon_service/src/service/luci_build_service_v2.dart';
+import 'package:cocoon_service/src/service/scheduler.dart';
+import 'package:cocoon_service/src/service/scheduler_v2.dart';
+import 'package:github/github.dart';
+import 'package:retry/retry.dart';
+
+import '../utilities/entity_generators.dart';
+import 'fake_luci_build_service_v2.dart';
+
+/// Fake for [Scheduler] to use for tests that rely on it.
+class FakeSchedulerV2 extends SchedulerV2 {
+  FakeSchedulerV2({
+    this.ciYaml,
+    LuciBuildServiceV2? luciBuildService,
+    BuildBucketV2Client? buildbucket,
+    required super.config,
+    GithubChecksUtil? githubChecksUtil,
+  }) : super(
+          cache: CacheService(inMemory: true),
+          githubChecksService: GithubChecksServiceV2(
+            config,
+            githubChecksUtil: githubChecksUtil,
+          ),
+          luciBuildService: luciBuildService ??
+              FakeLuciBuildServiceV2(
+                config: config,
+                buildBucketV2Client: buildbucket,
+                githubChecksUtil: githubChecksUtil,
+              ),
+        );
+
+  final CiYaml _defaultConfig = emptyConfig;
+
+  /// [CiYaml] value to be injected on [getCiYaml].
+  CiYaml? ciYaml;
+
+  @override
+  Future<CiYaml> getCiYaml(
+    Commit commit, {
+    CiYaml? totCiYaml,
+    RetryOptions? retryOptions,
+    bool validate = false,
+  }) async =>
+      ciYaml ?? _defaultConfig;
+
+  @override
+  Future<Commit> generateTotCommit({required String branch, required RepositorySlug slug}) async {
+    return generateCommit(1);
+  }
+
+  int cancelPreSubmitTargetsCallCnt = 0;
+
+  int get cancelPreSubmitTargetsCallCount => cancelPreSubmitTargetsCallCnt;
+
+  void resetCancelPreSubmitTargetsCallCount() => cancelPreSubmitTargetsCallCnt = 0;
+
+  @override
+  Future<void> cancelPreSubmitTargets({
+    required PullRequest pullRequest,
+    String reason = 'Newer commit available',
+  }) async {
+    await super.cancelPreSubmitTargets(pullRequest: pullRequest);
+    cancelPreSubmitTargetsCallCnt++;
+  }
+
+  int triggerPresubmitTargetsCnt = 0;
+
+  int get triggerPresubmitTargetsCallCount => triggerPresubmitTargetsCnt;
+
+  void resetTriggerPresubmitTargetsCallCount() => triggerPresubmitTargetsCnt = 0;
+
+  @override
+  Future<void> triggerPresubmitTargets({
+    required PullRequest pullRequest,
+    String reason = 'Newer commit available',
+    List<String>? builderTriggerList,
+  }) async {
+    await super.triggerPresubmitTargets(pullRequest: pullRequest);
+    triggerPresubmitTargetsCnt++;
+  }
+
+  int addPullRequestCallCnt = 0;
+
+  int get addPullRequestCallCount {
+    return addPullRequestCallCnt;
+  }
+
+  void resetAddPullRequestCallCount() {
+    addPullRequestCallCnt = 0;
+  }
+
+  @override
+  Future<void> addPullRequest(
+    PullRequest pr,
+  ) async {
+    await super.addPullRequest(pr);
+    addPullRequestCallCnt++;
+  }
+}
+
+final CiYaml emptyConfig = CiYaml(
+  slug: Config.flutterSlug,
+  branch: Config.defaultBranch(Config.flutterSlug),
+  config: pb.SchedulerConfig(
+    enabledBranches: <String>[
+      Config.defaultBranch(Config.flutterSlug),
+    ],
+    targets: <pb.Target>[
+      pb.Target(
+        name: 'Linux A',
+        scheduler: pb.SchedulerSystem.luci,
+      ),
+    ],
+  ),
+);
+
+CiYaml exampleConfig = CiYaml(
+  slug: Config.flutterSlug,
+  branch: Config.defaultBranch(Config.flutterSlug),
+  config: pb.SchedulerConfig(
+    enabledBranches: <String>[
+      Config.defaultBranch(Config.flutterSlug),
+    ],
+    targets: <pb.Target>[
+      pb.Target(
+        name: 'Linux A',
+        scheduler: pb.SchedulerSystem.luci,
+      ),
+      pb.Target(
+        name: 'Mac A',
+        scheduler: pb.SchedulerSystem.luci,
+      ),
+      pb.Target(
+        name: 'Windows A',
+        scheduler: pb.SchedulerSystem.luci,
+      ),
+      pb.Target(
+        name: 'Google Internal Roll',
+        presubmit: false,
+        postsubmit: true,
+        scheduler: pb.SchedulerSystem.google_internal,
+      ),
+    ],
+  ),
+);
+
+CiYaml exampleBackfillConfig = CiYaml(
+  slug: Config.flutterSlug,
+  branch: Config.defaultBranch(Config.flutterSlug),
+  config: pb.SchedulerConfig(
+    enabledBranches: <String>[
+      Config.defaultBranch(Config.flutterSlug),
+    ],
+    targets: <pb.Target>[
+      pb.Target(
+        name: 'Linux A',
+        scheduler: pb.SchedulerSystem.luci,
+        postsubmit: true,
+        properties: {'backfill': 'true'},
+      ),
+      pb.Target(
+        name: 'Mac A',
+        scheduler: pb.SchedulerSystem.luci,
+        postsubmit: true,
+      ),
+      pb.Target(
+        name: 'Windows A',
+        scheduler: pb.SchedulerSystem.luci,
+        postsubmit: true,
+        properties: {'backfill': 'false'},
+      ),
+    ],
+  ),
+);
+
+CiYaml examplePresubmitRescheduleConfig = CiYaml(
+  slug: Config.flutterSlug,
+  branch: Config.defaultBranch(Config.flutterSlug),
+  config: pb.SchedulerConfig(
+    enabledBranches: <String>[
+      Config.defaultBranch(Config.flutterSlug),
+    ],
+    targets: <pb.Target>[
+      pb.Target(
+        name: 'Linux A',
+      ),
+      pb.Target(
+        name: 'Linux B',
+        postsubmit: true,
+        properties: {'presubmit_retry': '1'},
+      ),
+    ],
+  ),
+);
+
+final CiYaml batchPolicyConfig = CiYaml(
+  slug: Config.flutterSlug,
+  branch: Config.defaultBranch(Config.flutterSlug),
+  config: pb.SchedulerConfig(
+    enabledBranches: <String>[
+      Config.defaultBranch(Config.flutterSlug),
+    ],
+    targets: <pb.Target>[
+      pb.Target(
+        name: 'Linux_android A',
+      ),
+      pb.Target(
+        name: 'Linux_android B',
+      ),
+      pb.Target(
+        name: 'Linux_android C',
+      ),
+    ],
+  ),
+);
+
+final CiYaml unsupportedPostsubmitCheckrunConfig = CiYaml(
+  slug: Config.flutterSlug,
+  branch: Config.defaultBranch(Config.flutterSlug),
+  config: pb.SchedulerConfig(
+    enabledBranches: <String>[
+      Config.defaultBranch(Config.flutterSlug),
+    ],
+    targets: <pb.Target>[
+      pb.Target(
+        name: 'Linux flutter',
+      ),
+    ],
+  ),
+);
+
+final CiYaml nonBringupPackagesConfig = CiYaml(
+  slug: Config.packagesSlug,
+  branch: Config.defaultBranch(Config.packagesSlug),
+  config: pb.SchedulerConfig(
+    enabledBranches: <String>[
+      Config.defaultBranch(Config.packagesSlug),
+    ],
+    targets: <pb.Target>[
+      pb.Target(
+        name: 'Linux nonbringup',
+      ),
+    ],
+  ),
+);
+
+final CiYaml bringupPackagesConfig = CiYaml(
+  slug: Config.packagesSlug,
+  branch: Config.defaultBranch(Config.packagesSlug),
+  config: pb.SchedulerConfig(
+    enabledBranches: <String>[
+      Config.defaultBranch(Config.packagesSlug),
+    ],
+    targets: <pb.Target>[
+      pb.Target(
+        name: 'Linux bringup',
+        bringup: true,
+      ),
+    ],
+  ),
+);
+
+final CiYaml totCiYaml = CiYaml(
+  slug: Config.flutterSlug,
+  branch: Config.defaultBranch(Config.flutterSlug),
+  config: pb.SchedulerConfig(
+    enabledBranches: <String>[
+      Config.defaultBranch(Config.flutterSlug),
+    ],
+    targets: <pb.Target>[
+      pb.Target(
+        name: 'Linux_android B',
+      ),
+      pb.Target(
+        name: 'Linux_android C',
+      ),
+    ],
+  ),
+);
+
+final CiYaml notInToTConfig = CiYaml(
+  slug: Config.flutterSlug,
+  branch: Config.defaultBranch(Config.flutterSlug),
+  config: pb.SchedulerConfig(
+    enabledBranches: <String>[
+      Config.defaultBranch(Config.flutterSlug),
+    ],
+    targets: <pb.Target>[
+      pb.Target(
+        name: 'Linux_android A',
+      ),
+    ],
+  ),
+  totConfig: totCiYaml,
+);