blob: 2822877eb3a2256e152e924eb56a40d3d938d7d1 [file] [log] [blame]
// Copyright 2019 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 'dart:async';
import 'package:appengine/appengine.dart';
import 'package:cocoon_service/src/model/appengine/branch.dart';
import 'package:cocoon_service/src/model/appengine/key_helper.dart';
import 'package:cocoon_service/src/model/appengine/service_account_info.dart';
import 'package:cocoon_service/src/service/bigquery.dart';
import 'package:cocoon_service/src/service/config.dart';
import 'package:cocoon_service/src/service/github_service.dart';
import 'package:github/github.dart' as gh;
import 'package:googleapis/bigquery/v2.dart';
import 'package:graphql/client.dart';
import '../request_handling/fake_authentication.dart';
import 'fake_datastore.dart';
// ignore: must_be_immutable
class FakeConfig implements Config {
FakeConfig({
this.githubClient,
this.deviceLabServiceAccountValue,
this.maxTaskRetriesValue,
this.maxLuciTaskRetriesValue,
this.keyHelperValue,
this.oauthClientIdValue,
this.githubOAuthTokenValue,
this.mergeConflictPullRequestMessageValue = 'default mergeConflictPullRequestMessageValue',
this.missingTestsPullRequestMessageValue = 'default missingTestsPullRequestMessageValue',
this.wrongBaseBranchPullRequestMessageValue,
this.wrongHeadBranchPullRequestMessageValue,
this.releaseBranchPullRequestMessageValue,
this.webhookKeyValue,
this.loggingServiceValue,
this.tabledataResource,
this.githubService,
this.bigqueryService,
this.githubGraphQLClient,
this.cirrusGraphQLClient,
this.rollerAccountsValue,
this.flutterBuildValue,
this.flutterBuildDescriptionValue,
this.maxRecordsValue,
this.flutterGoldPendingValue,
this.flutterGoldSuccessValue,
this.flutterGoldChangesValue,
this.flutterGoldAlertConstantValue,
this.flutterGoldInitialAlertValue,
this.flutterGoldFollowUpAlertValue,
this.flutterGoldDraftChangeValue,
this.flutterGoldStalePRValue,
this.supportedBranchesValue,
this.supportedReposValue,
this.batchSizeValue,
FakeDatastoreDB? dbValue,
}) : dbValue = dbValue ?? FakeDatastoreDB();
gh.GitHub? githubClient;
GraphQLClient? githubGraphQLClient;
GraphQLClient? cirrusGraphQLClient;
TabledataResource? tabledataResource;
BigqueryService? bigqueryService;
GithubService? githubService;
FakeDatastoreDB dbValue;
ServiceAccountInfo? deviceLabServiceAccountValue;
int? maxTaskRetriesValue;
int? maxLuciTaskRetriesValue;
int? batchSizeValue;
FakeKeyHelper? keyHelperValue;
String? oauthClientIdValue;
String? githubOAuthTokenValue;
String mergeConflictPullRequestMessageValue;
String missingTestsPullRequestMessageValue;
String? wrongBaseBranchPullRequestMessageValue;
String? wrongHeadBranchPullRequestMessageValue;
String? releaseBranchPullRequestMessageValue;
String? webhookKeyValue;
String? flutterBuildValue;
String? flutterBuildDescriptionValue;
Logging? loggingServiceValue;
String? waitingForTreeToGoGreenLabelNameValue;
Set<String>? rollerAccountsValue;
int? maxRecordsValue;
String? flutterGoldPendingValue;
String? flutterGoldSuccessValue;
String? flutterGoldChangesValue;
String? flutterGoldAlertConstantValue;
String? flutterGoldInitialAlertValue;
String? flutterGoldFollowUpAlertValue;
String? flutterGoldDraftChangeValue;
String? flutterGoldStalePRValue;
List<String>? supportedBranchesValue;
String? overrideTreeStatusLabelValue;
Set<gh.RepositorySlug>? supportedReposValue;
@override
Future<gh.GitHub> createGitHubClient({gh.PullRequest? pullRequest, gh.RepositorySlug? slug}) async => githubClient!;
@override
gh.GitHub createGitHubClientWithToken(String token) => githubClient!;
@override
Future<GraphQLClient> createGitHubGraphQLClient() async => githubGraphQLClient!;
@override
Future<GraphQLClient> createCirrusGraphQLClient() async => cirrusGraphQLClient!;
@override
Future<TabledataResource> createTabledataResourceApi() async => tabledataResource!;
@override
Future<BigqueryService> createBigQueryService() async => bigqueryService!;
@override
Future<GithubService> createGithubService(gh.RepositorySlug slug) async => githubService!;
@override
GithubService createGithubServiceWithToken(String token) => githubService!;
@override
FakeDatastoreDB get db => dbValue;
@override
int get maxTaskRetries => maxTaskRetriesValue!;
/// Size of the shards to send to buildBucket when scheduling builds.
@override
int get schedulingShardSize => 5;
@override
int get batchSize => batchSizeValue ?? 5;
@override
int get maxLuciTaskRetries => maxLuciTaskRetriesValue!;
@override
int get maxRecords => maxRecordsValue!;
@override
String get flutterGoldPending => flutterGoldPendingValue!;
@override
String get flutterGoldSuccess => flutterGoldSuccessValue!;
@override
String get flutterGoldChanges => flutterGoldChangesValue!;
@override
String get flutterGoldDraftChange => flutterGoldDraftChangeValue!;
@override
String get flutterGoldStalePR => flutterGoldStalePRValue!;
@override
String flutterGoldInitialAlert(String url) => flutterGoldInitialAlertValue!;
@override
String flutterGoldFollowUpAlert(String url) => flutterGoldFollowUpAlertValue!;
@override
String flutterGoldAlertConstant(gh.RepositorySlug slug) => flutterGoldAlertConstantValue!;
@override
String flutterGoldCommentID(gh.PullRequest pr) => 'PR ${pr.number}, at ${pr.head!.sha}';
@override
int get commitNumber => 30;
@override
KeyHelper get keyHelper => keyHelperValue!;
@override
Future<String> get oauthClientId async => oauthClientIdValue!;
@override
Future<String> get githubOAuthToken async => githubOAuthTokenValue ?? 'token';
@override
String get mergeConflictPullRequestMessage => mergeConflictPullRequestMessageValue;
@override
String get missingTestsPullRequestMessage => missingTestsPullRequestMessageValue;
@override
String get wrongBaseBranchPullRequestMessage => wrongBaseBranchPullRequestMessageValue!;
@override
String wrongHeadBranchPullRequestMessage(String branch) => wrongHeadBranchPullRequestMessageValue!;
@override
String get releaseBranchPullRequestMessage => releaseBranchPullRequestMessageValue!;
@override
Future<String> get webhookKey async => webhookKeyValue!;
@override
String get flutterBuild => flutterBuildValue!;
@override
String get flutterBuildDescription =>
flutterBuildDescriptionValue ??
'Tree is currently broken. Please do not merge this '
'PR unless it contains a fix for the tree.';
@override
Logging get loggingService => loggingServiceValue!;
@override
String get waitingForTreeToGoGreenLabelName => waitingForTreeToGoGreenLabelNameValue!;
@override
Set<String> get rollerAccounts => rollerAccountsValue!;
@override
bool githubPresubmitSupportedRepo(gh.RepositorySlug slug) {
return <gh.RepositorySlug>[
Config.flutterSlug,
Config.engineSlug,
Config.cocoonSlug,
Config.packagesSlug,
Config.pluginsSlug,
].contains(slug);
}
@override
Future<String> generateGithubToken(gh.RepositorySlug slug) {
throw UnimplementedError();
}
@override
Future<String> generateJsonWebToken() {
throw UnimplementedError();
}
@override
Future<String> get githubAppId => throw UnimplementedError();
@override
Future<Map<String, dynamic>> get githubAppInstallations => throw UnimplementedError();
@override
Future<String> get githubPrivateKey => throw UnimplementedError();
@override
Future<String> get githubPublicKey => throw UnimplementedError();
@override
Future<GithubService> createDefaultGitHubService() async => githubService!;
@override
Future<String> get overrideTreeStatusLabel async => overrideTreeStatusLabelValue!;
@override
String get defaultRecipeBundleRef => 'refs/heads/main';
@override
Future<List<String>> get releaseAccounts async => <String>['dart-flutter-releaser'];
@override
Set<gh.RepositorySlug> get supportedRepos => supportedReposValue ?? <gh.RepositorySlug>{Config.flutterSlug};
@override
Future<Iterable<Branch>> getBranches(gh.RepositorySlug slug) async {
if (supportedBranchesValue == null) {
throw Exception('Test must set suportedBranchesValue to be able to use Config.getBranches');
}
return supportedBranchesValue!.map((String branch) => Branch(
key: db.emptyKey.append<String>(
Branch,
id: '${slug.fullName}/$branch',
),
));
}
}