auto_submit/test/requests/check_pull_request_test.dart

// Copyright 2022 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.

// ignore_for_file: constant_identifier_names
import 'dart:async';
import 'dart:convert';

import 'package:auto_submit/configuration/repository_configuration.dart';
import 'package:auto_submit/requests/check_pull_request.dart';
import 'package:auto_submit/requests/graphql_queries.dart';
import 'package:auto_submit/service/config.dart';
import 'package:auto_submit/service/validation_service.dart';
import 'package:cocoon_common/cocoon_common.dart';
import 'package:cocoon_common_test/cocoon_common_test.dart';
import 'package:cocoon_server/bigquery.dart';
import 'package:cocoon_server/logging.dart';
import 'package:cocoon_server_test/bigquery_testing.dart';
import 'package:cocoon_server_test/mocks.dart';
import 'package:cocoon_server_test/test_logging.dart';
import 'package:github/github.dart';
import 'package:googleapis/bigquery/v2.dart';
import 'package:graphql/client.dart' hide Request, Response;
import 'package:mockito/mockito.dart';
import 'package:test/test.dart';

import '../configuration/repository_configuration_data.dart';
import '../src/request_handling/fake_authentication.dart';
import '../src/request_handling/fake_pubsub.dart';
import '../src/service/fake_config.dart';
import '../src/service/fake_github_service.dart';
import '../src/service/fake_graphql_client.dart';
import '../utilities/mocks.dart';
import '../utilities/utils.dart' hide createQueryResult;
import './github_webhook_test_data.dart';

const String oid = '6dcb09b5b57875f334f61aebed695e2e4193db5e';

void main() {
  useTestLoggerPerTest();

  group('Check CheckPullRequest', () {
    late CheckPullRequest checkPullRequest;
    late FakeConfig config;
    late FakeCronAuthProvider auth;
    late FakeGraphQLClient githubGraphQLClient;
    late FakeGithubService githubService;
    late MockJobsResource jobsResource;
    late BigqueryService bigqueryService;
    late MockPullRequestsService pullRequests;
    final gitHub = MockGitHub();
    late FakePubSub pubsub;
    late PullRequestHelper flutterRequest;
    late PullRequestHelper cocoonRequest;
    late List<QueryOptions> expectedOptions;
    late QueryOptions flutterOption;
    late QueryOptions cocoonOption;
    const testSubscription = 'test-sub';
    const testTopic = 'test-topic';
    const rollorAuthor = 'engine-flutter-autoroll';
    const cocoonRepo = 'cocoon';
    const noAutosubmitLabel = 'no_autosubmit';

    setUp(() {
      // To keep pre-MQ workflow tested until we're completely moved to MQ-based
      // workflow, pretend that flutter/flutter does not yet use MQ. Then all
      // the existing tests will continue testing the non-MQ workflow.
      mqEnabledRepos = const <String>[];

      githubGraphQLClient = FakeGraphQLClient();
      auth = FakeCronAuthProvider();
      pubsub = FakePubSub();
      expectedOptions = <QueryOptions>[];
      githubService = FakeGithubService();

      githubGraphQLClient.mutateResultForOptions = (MutationOptions options) {
        final operation = options.asRequest.operation;
        final mutationNameRegExp = RegExp(r'"mutation (\w+)');
        final mutationName = mutationNameRegExp
            .firstMatch(operation.toString())!
            .group(1);

        if (mutationName == 'EnqueueFlutterPullRequest') {
          return QueryResult(
            data: {'clientMutationId': 1},
            options: QueryOptions(document: options.document),
            source: QueryResultSource.network,
          );
        }

        return createFakeQueryResult();
      };

      githubGraphQLClient.queryResultForOptions = (QueryOptions options) {
        final operation = options.asRequest.operation;
        final queryNameRegExp = RegExp(r'"query (\w+)');
        final queryName = queryNameRegExp
            .firstMatch(operation.toString())!
            .group(1);

        if (queryName == 'LabeledPullRequestWithReviews') {
          expect(options.variables['sOwner'], 'flutter');
          final repoName = options.variables['sName'] as String?;
          if (repoName == 'flutter') {
            return createQueryResult(flutterRequest);
          } else if (repoName == 'cocoon') {
            return createQueryResult(cocoonRequest);
          } else {
            fail('unexpected repo $repoName');
          }
        } else if (queryName == 'FindPullRequestNodeId') {
          return createFakeQueryResult(
            data: <String, dynamic>{
              'repository': <String, dynamic>{
                'pullRequest': {'id': 'PR_blahblah'},
              },
            },
          );
        } else {
          throw UnsupportedError('Unexpected query "$queryName"');
        }
      };

      final findPullRequestsWithReviewsQueryFlutter =
          FindPullRequestsWithReviewsQuery(
            repositoryOwner: 'flutter',
            repositoryName: 'flutter',
            pullRequestNumber: 0,
          );

      flutterOption = QueryOptions(
        document: findPullRequestsWithReviewsQueryFlutter.documentNode,
        fetchPolicy: FetchPolicy.noCache,
        variables: findPullRequestsWithReviewsQueryFlutter.variables,
      );

      final findPullRequestsWithReviewsQueryCocoon =
          FindPullRequestsWithReviewsQuery(
            repositoryOwner: 'flutter',
            repositoryName: 'cocoon',
            pullRequestNumber: 1,
          );

      cocoonOption = QueryOptions(
        document: findPullRequestsWithReviewsQueryCocoon.documentNode,
        fetchPolicy: FetchPolicy.noCache,
        variables: findPullRequestsWithReviewsQueryCocoon.variables,
      );

      githubService.checkRunsData = checkRunsMock;
      githubService.compareTwoCommitsData = compareTwoCommitsMock;
      githubService.successMergeData = successMergeMock;
      githubService.createCommentData = createCommentMock;
      githubService.commitData = commitMock;
      jobsResource = MockJobsResource();
      bigqueryService = BigqueryService.forTesting(jobsResource);
      config = FakeConfig(
        githubService: githubService,
        githubGraphQLClient: githubGraphQLClient,
        githubClient: gitHub,
      );
      config.bigqueryService = bigqueryService;
      config.repositoryConfigurationMock = RepositoryConfiguration.fromYaml(
        sampleConfigNoOverride,
      );
      pullRequests = MockPullRequestsService();
      when(gitHub.pullRequests).thenReturn(pullRequests);
      when(
        // ignore: discarded_futures
        pullRequests.get(any, any),
      ).thenAnswer((_) async => PullRequest(number: 123, state: 'open'));

      // ignore: discarded_futures
      when(jobsResource.query(captureAny, any)).thenAnswer((
        Invocation invocation,
      ) {
        return Future<QueryResponse>.value(
          QueryResponse.fromJson(
            jsonDecode(insertDeleteUpdateSuccessResponse)
                as Map<dynamic, dynamic>,
          ),
        );
      });
    });

    void verifyQueries(List<QueryOptions> expectedOptions) {
      githubGraphQLClient.verifyQueries(expectedOptions);
    }

    test('Multiple identical messages are processed once', () async {
      final pullRequest1 = generatePullRequest(
        prNumber: 0,
        repoName: cocoonRepo,
      );
      // 'member' is in the review nodes and 'author1' is the pr author.
      githubService.isTeamMemberMockMap['member'] = true;
      githubService.isTeamMemberMockMap['author1'] = true;
      githubService.pullRequestData = pullRequest1;
      for (var i = 0; i < 2; i++) {
        unawaited(pubsub.publish('auto-submit-queue-sub', pullRequest1));
      }

      checkPullRequest = CheckPullRequest(
        config: config,
        pubsub: pubsub,
        cronAuthProvider: auth,
      );
      cocoonRequest = PullRequestHelper(prNumber: 0, lastCommitHash: oid);

      final expectedMergeRequestMap = <int, RepositorySlug>{};
      expectedMergeRequestMap[0] = RepositorySlug('flutter', cocoonRepo);

      await checkPullRequest.get();

      githubService.verifyMergePullRequests(expectedMergeRequestMap);

      expect(0, pubsub.messagesQueue.length);
    });

    test(
      'Failing messages are logged by do not fail the entire request handler',
      () async {
        final pullRequest = generatePullRequest(
          prNumber: 0,
          repoName: cocoonRepo,
        );
        // 'member' is in the review nodes and 'author1' is the pr author.
        githubService.isTeamMemberMockMap['member'] = true;
        githubService.isTeamMemberMockMap['author1'] = true;
        githubService.pullRequestData = null;
        unawaited(pubsub.publish('auto-submit-queue-sub', pullRequest));

        checkPullRequest = CheckPullRequest(
          config: config,
          pubsub: pubsub,
          cronAuthProvider: auth,
        );

        // Calling get should not fail despite the failing message inside.
        await checkPullRequest.get();

        expect(
          log,
          bufferedLoggerOf(
            contains(
              logThat(
                message: stringContainsInOrder([
                  'CheckPullRequest(flutter/cocoon/0): failed to process message.',
                  'Pull request: https://github.com/flutter/cocoon/0',
                ]),
                severity: equals(Severity.error),
                error: isA<Error>().having(
                  (e) => e.toString(),
                  'toString()',
                  contains('Unexpected invocation of getPullRequest method'),
                ),
              ),
            ),
          ),
        );

        // The message failed, so it shouldn't be acked.
        expect(pubsub.messagesQueue, hasLength(1));
      },
    );

    test('Closed PRs are not processed', () async {
      final pullRequest1 = generatePullRequest(
        prNumber: 0,
        repoName: cocoonRepo,
        state: 'close',
      );
      githubService.pullRequestData = pullRequest1;
      when(
        pullRequests.get(any, any),
      ).thenAnswer((_) async => PullRequest(number: 0, state: 'close'));
      for (var i = 0; i < 2; i++) {
        unawaited(pubsub.publish('auto-submit-queue-sub', pullRequest1));
      }

      checkPullRequest = CheckPullRequest(
        config: config,
        pubsub: pubsub,
        cronAuthProvider: auth,
      );
      cocoonRequest = PullRequestHelper(prNumber: 0, lastCommitHash: oid);
      await checkPullRequest.get();

      githubGraphQLClient.verifyMutations(<MutationOptions>[]);
      expect(0, pubsub.messagesQueue.length);
    });

    test('Merge exception is handled correctly', () async {
      final pullRequest1 = generatePullRequest(prNumber: 0);
      final pullRequest2 = generatePullRequest(
        prNumber: 1,
        repoName: cocoonRepo,
      );

      githubService.pullRequestData = pullRequest1;
      // 'member' is in the review nodes and 'author1' is the pr author.
      githubService.isTeamMemberMockMap['member'] = true;
      githubService.isTeamMemberMockMap['author1'] = true;

      final pullRequests = <PullRequest>[pullRequest1, pullRequest2];
      for (var pr in pullRequests) {
        unawaited(pubsub.publish(testTopic, pr));
      }

      checkPullRequest = CheckPullRequest(
        config: config,
        pubsub: pubsub,
        cronAuthProvider: auth,
      );
      flutterRequest = PullRequestHelper(prNumber: 0, lastCommitHash: oid);
      cocoonRequest = PullRequestHelper(prNumber: 1, lastCommitHash: oid);

      githubService.useMergeRequestMockList = true;
      githubService.pullRequestMergeMockList.add(
        PullRequestMerge(
          merged: false,
          message: 'Unable to merge pull request',
        ),
      );
      githubService.pullRequestMergeMockList.add(
        PullRequestMerge(
          merged: true,
          sha: 'sha',
          message: 'Pull request merged successfully',
        ),
      );

      final expectedMergeRequestMap = <int, RepositorySlug>{};
      expectedMergeRequestMap[0] = RepositorySlug('flutter', 'flutter');
      expectedMergeRequestMap[1] = RepositorySlug('flutter', cocoonRepo);

      // this is the test.
      await checkPullRequest.get();
      // every failure is now acknowledged from the queue.
      expect(pubsub.messagesQueue.length, 0);

      expect(
        log,
        bufferedLoggerOf(
          contains(
            logThat(
              message: contains('Failed to merge'),
              severity: equals(Severity.error),
            ),
          ),
        ),
      );
      pubsub.messagesQueue.clear();
    });

    test('Merges PR with successful status and checks', () async {
      final pullRequest1 = generatePullRequest(prNumber: 0);
      final pullRequest2 = generatePullRequest(
        prNumber: 1,
        repoName: cocoonRepo,
      );

      githubService.usePullRequestList = true;
      githubService.pullRequestMockList = [pullRequest1, pullRequest2];

      // 'octocat' is the pr author from generatePullRequest calls.
      // 'member' is in the review nodes and 'author1' is the pr author.
      githubService.isTeamMemberMockMap['member'] = true;
      githubService.isTeamMemberMockMap['author1'] = true;
      final pullRequests = <PullRequest>[pullRequest1, pullRequest2];
      for (var pr in pullRequests) {
        unawaited(pubsub.publish(testTopic, pr));
      }

      checkPullRequest = CheckPullRequest(
        config: config,
        pubsub: pubsub,
        cronAuthProvider: auth,
      );

      flutterRequest = PullRequestHelper(prNumber: 0, lastCommitHash: oid);
      cocoonRequest = PullRequestHelper(prNumber: 1, lastCommitHash: oid);

      await checkPullRequest.get();
      expectedOptions.add(flutterOption);
      expectedOptions.add(cocoonOption);
      verifyQueries(expectedOptions);

      githubService.useMergeRequestMockList = true;
      githubService.pullRequestMergeMockList.add(
        PullRequestMerge(
          merged: true,
          sha: 'sha1',
          message: 'Pull request merged successfully',
        ),
      );
      githubService.pullRequestMergeMockList.add(
        PullRequestMerge(
          merged: true,
          sha: 'sha2',
          message: 'Pull request merged successfully',
        ),
      );

      final expectedMergeRequestMap = <int, RepositorySlug>{};
      expectedMergeRequestMap[0] = RepositorySlug('flutter', 'flutter');
      expectedMergeRequestMap[1] = RepositorySlug('flutter', cocoonRepo);

      githubService.verifyMergePullRequests(expectedMergeRequestMap);

      assert(pubsub.messagesQueue.isEmpty);
    });

    test('Merges unapproved PR from autoroller', () async {
      final pullRequest = generatePullRequest(
        prNumber: 0,
        author: rollorAuthor,
      );
      githubService.pullRequestData = pullRequest;
      unawaited(pubsub.publish(testTopic, pullRequest));

      checkPullRequest = CheckPullRequest(
        config: config,
        pubsub: pubsub,
        cronAuthProvider: auth,
        approverProvider: (Config config) => MockApproverService(),
      );

      flutterRequest = PullRequestHelper(
        prNumber: 0,
        author: 'dependabot',
        reviews: const <PullRequestReviewHelper>[],
        lastCommitHash: oid,
      );

      await checkPullRequest.get();
      expectedOptions.add(flutterOption);
      verifyQueries(expectedOptions);

      final expectedMergeRequestMap = <int, RepositorySlug>{};
      expectedMergeRequestMap[0] = RepositorySlug('flutter', 'flutter');

      githubService.mergeRequestMock = PullRequestMerge(
        merged: true,
        sha: 'sha1',
        message: 'Pull request merged successfully',
      );

      githubService.verifyMergePullRequests(expectedMergeRequestMap);

      assert(pubsub.messagesQueue.isEmpty);
    });

    test('Merges a clean revert PR with in progress tests', () async {
      final pullRequest = generatePullRequest(prNumber: 0);
      githubService.pullRequestData = pullRequest;
      // 'member' is in the review nodes and 'author1' is the pr author.
      githubService.isTeamMemberMockMap['member'] = true;
      githubService.isTeamMemberMockMap['author1'] = true;
      unawaited(pubsub.publish(testTopic, pullRequest));
      checkPullRequest = CheckPullRequest(
        config: config,
        pubsub: pubsub,
        cronAuthProvider: auth,
      );

      flutterRequest = PullRequestHelper(
        prNumber: 0,
        lastCommitHash: oid,
        lastCommitStatuses: const <StatusHelper>[
          StatusHelper.flutterBuildSuccess,
        ],
        lastCommitMessage:
            'Revert "This is a test PR" This reverts commit abc.',
      );

      await checkPullRequest.get();
      expectedOptions.add(flutterOption);
      verifyQueries(expectedOptions);

      final expectedMergeRequestMap = <int, RepositorySlug>{};
      expectedMergeRequestMap[0] = RepositorySlug('flutter', 'flutter');

      githubService.mergeRequestMock = PullRequestMerge(
        merged: true,
        sha: 'sha1',
        message: 'Pull request merged successfully',
      );

      githubService.verifyMergePullRequests(expectedMergeRequestMap);

      assert(pubsub.messagesQueue.isEmpty);
    });

    test(
      'Merges PR with successful checks on repo without tree status',
      () async {
        final pullRequest = generatePullRequest(
          prNumber: 1,
          repoName: cocoonRepo,
        );
        githubService.pullRequestData = pullRequest;
        // 'member' is in the review nodes and 'author1' is the pr author.
        githubService.isTeamMemberMockMap['member'] = true;
        githubService.isTeamMemberMockMap['author1'] = true;
        unawaited(pubsub.publish(testTopic, pullRequest));

        checkPullRequest = CheckPullRequest(
          config: config,
          pubsub: pubsub,
          cronAuthProvider: auth,
        );

        cocoonRequest = PullRequestHelper(
          lastCommitHash: oid,
          lastCommitStatuses: const <StatusHelper>[],
        );

        await checkPullRequest.get();
        expectedOptions.add(cocoonOption);
        verifyQueries(expectedOptions);

        final expectedMergeRequestMap = <int, RepositorySlug>{};
        expectedMergeRequestMap[1] = RepositorySlug('flutter', cocoonRepo);

        githubService.mergeRequestMock = PullRequestMerge(
          merged: true,
          sha: 'sha1',
          message: 'Pull request merged successfully',
        );

        githubService.verifyMergePullRequests(expectedMergeRequestMap);

        assert(pubsub.messagesQueue.isEmpty);
      },
    );

    test('Merges PR with neutral status checkrun', () async {
      final pullRequest1 = generatePullRequest(prNumber: 0);
      final pullRequest2 = generatePullRequest(
        prNumber: 1,
        repoName: cocoonRepo,
      );
      githubService.pullRequestData = pullRequest1;
      // 'member' is in the review nodes and 'author1' is the pr author.
      githubService.isTeamMemberMockMap['member'] = true;
      githubService.isTeamMemberMockMap['author1'] = true;
      final pullRequests = <PullRequest>[pullRequest1, pullRequest2];
      for (var pr in pullRequests) {
        unawaited(pubsub.publish(testTopic, pr));
      }
      githubService.checkRunsData = neutralCheckRunsMock;
      checkPullRequest = CheckPullRequest(
        config: config,
        pubsub: pubsub,
        cronAuthProvider: auth,
      );
      flutterRequest = PullRequestHelper(prNumber: 0, lastCommitHash: oid);
      cocoonRequest = PullRequestHelper(prNumber: 1, lastCommitHash: oid);

      await checkPullRequest.get();
      expectedOptions.add(flutterOption);
      expectedOptions.add(cocoonOption);
      verifyQueries(expectedOptions);
      assert(pubsub.messagesQueue.isEmpty);
    });

    test('Removes the label for the PR with failed tests', () async {
      final pullRequest1 = generatePullRequest(prNumber: 0);
      final pullRequest2 = generatePullRequest(
        prNumber: 1,
        repoName: cocoonRepo,
      );
      githubService.pullRequestData = pullRequest1;
      // 'member' is in the review nodes and 'author1' is the pr author.
      githubService.isTeamMemberMockMap['member'] = true;
      githubService.isTeamMemberMockMap['author1'] = true;
      final pullRequests = <PullRequest>[pullRequest1, pullRequest2];
      for (var pr in pullRequests) {
        unawaited(pubsub.publish(testTopic, pr));
      }
      githubService.checkRunsData = failedCheckRunsMock;
      checkPullRequest = CheckPullRequest(
        config: config,
        pubsub: pubsub,
        cronAuthProvider: auth,
      );
      flutterRequest = PullRequestHelper(prNumber: 0, lastCommitHash: oid);
      cocoonRequest = PullRequestHelper(prNumber: 1, lastCommitHash: oid);

      await checkPullRequest.get();
      expectedOptions.add(flutterOption);
      expectedOptions.add(cocoonOption);
      verifyQueries(expectedOptions);
      assert(pubsub.messagesQueue.isEmpty);
    });

    test('Removes the label for the PR with failed status', () async {
      final pullRequest = generatePullRequest(prNumber: 0);
      githubService.pullRequestData = pullRequest;
      // 'member' is in the review nodes and 'author1' is the pr author.
      githubService.isTeamMemberMockMap['member'] = true;
      githubService.isTeamMemberMockMap['author1'] = true;
      unawaited(pubsub.publish(testTopic, pullRequest));

      checkPullRequest = CheckPullRequest(
        config: config,
        pubsub: pubsub,
        cronAuthProvider: auth,
      );

      flutterRequest = PullRequestHelper(
        lastCommitHash: oid,
        lastCommitStatuses: const <StatusHelper>[
          StatusHelper.flutterBuildSuccess,
          StatusHelper.otherStatusFailure,
        ],
      );

      await checkPullRequest.get();
      expectedOptions.add(flutterOption);
      verifyQueries(expectedOptions);
      assert(pubsub.messagesQueue.isEmpty);
    });

    test(
      'Removes the label if non member does not have at least 2 member reviews',
      () async {
        final pullRequest = generatePullRequest(prNumber: 0);
        githubService.pullRequestData = pullRequest;
        // 'octocat' is the pr author from generatePullRequest calls.
        githubService.isTeamMemberMockMap['octocat'] = false;
        unawaited(pubsub.publish(testTopic, pullRequest));

        checkPullRequest = CheckPullRequest(
          config: config,
          pubsub: pubsub,
          cronAuthProvider: auth,
        );

        flutterRequest = PullRequestHelper(
          authorAssociation: '',
          lastCommitHash: oid,
          lastCommitStatuses: const <StatusHelper>[
            StatusHelper.flutterBuildSuccess,
          ],
        );

        await checkPullRequest.get();
        expectedOptions.add(flutterOption);
        verifyQueries(expectedOptions);
        assert(pubsub.messagesQueue.isEmpty);
      },
    );

    test(
      'Removes the label for the PR with null checks and statuses',
      () async {
        final pullRequest = generatePullRequest(prNumber: 0);
        githubService.pullRequestData = pullRequest;
        // 'octocat' is the pr author from generatePullRequest calls.
        githubService.isTeamMemberMockMap['octocat'] = true;
        unawaited(pubsub.publish(testTopic, pullRequest));

        githubService.checkRunsData = emptyCheckRunsMock;
        checkPullRequest = CheckPullRequest(
          config: config,
          pubsub: pubsub,
          cronAuthProvider: auth,
        );

        flutterRequest = PullRequestHelper(
          lastCommitHash: oid,
          lastCommitStatuses: const <StatusHelper>[],
        );

        await checkPullRequest.get();
        expectedOptions.add(flutterOption);
        verifyQueries(expectedOptions);
        assert(pubsub.messagesQueue.isEmpty);
      },
    );

    test('Does not merge PR with in progress checks', () async {
      final pullRequest1 = generatePullRequest(prNumber: 0);
      final pullRequest2 = generatePullRequest(
        prNumber: 1,
        repoName: cocoonRepo,
      );
      githubService.pullRequestData = pullRequest1;
      // 'member' is in the review nodes and 'author1' is the pr author.
      githubService.isTeamMemberMockMap['member'] = true;
      githubService.isTeamMemberMockMap['author1'] = true;
      final pullRequests = <PullRequest>[pullRequest1, pullRequest2];
      for (var pr in pullRequests) {
        unawaited(pubsub.publish(testTopic, pr));
      }
      githubService.checkRunsData = inProgressCheckRunsMock;
      checkPullRequest = CheckPullRequest(
        config: config,
        pubsub: pubsub,
        cronAuthProvider: auth,
      );
      flutterRequest = PullRequestHelper(prNumber: 0);
      cocoonRequest = PullRequestHelper(prNumber: 1);
      await checkPullRequest.get();
      expectedOptions.add(flutterOption);
      expectedOptions.add(cocoonOption);
      verifyQueries(expectedOptions);
      expect(pubsub.messagesQueue.length, 2);
      pubsub.messagesQueue.clear();
    });

    test('Does not merge PR if no autosubmit label any more', () async {
      final pullRequest1 = generatePullRequest(
        prNumber: 0,
        autosubmitLabel: noAutosubmitLabel,
      );
      final pullRequest2 = generatePullRequest(
        prNumber: 1,
        autosubmitLabel: noAutosubmitLabel,
        repoName: cocoonRepo,
      );
      githubService.pullRequestData = pullRequest1;
      final pullRequests = <PullRequest>[pullRequest1, pullRequest2];
      for (var pr in pullRequests) {
        unawaited(pubsub.publish(testTopic, pr));
      }
      checkPullRequest = CheckPullRequest(
        config: config,
        pubsub: pubsub,
        cronAuthProvider: auth,
      );
      flutterRequest = PullRequestHelper(prNumber: 0);
      cocoonRequest = PullRequestHelper(prNumber: 1);
      await checkPullRequest.get();
      assert(pubsub.messagesQueue.isEmpty);
    });

    test('Self review is disallowed', () async {
      final pullRequest = generatePullRequest(
        prNumber: 0,
        author: 'some_rando',
      );
      githubService.pullRequestData = pullRequest;
      // 'octocat' is the pr author from generatePullRequest calls.
      githubService.isTeamMemberMockMap['some_rando'] = true;
      unawaited(pubsub.publish(testTopic, pullRequest));
      checkPullRequest = CheckPullRequest(
        config: config,
        pubsub: pubsub,
        cronAuthProvider: auth,
      );
      flutterRequest = PullRequestHelper(
        author: 'some_rando',
        lastCommitHash: oid,
        authorAssociation: 'MEMBER',
        reviews: <PullRequestReviewHelper>[
          const PullRequestReviewHelper(
            authorName: 'some_rando',
            state: ReviewState.APPROVED,
            memberType: MemberType.MEMBER,
          ),
        ],
        lastCommitStatuses: const <StatusHelper>[
          StatusHelper.flutterBuildSuccess,
        ],
      );
      await checkPullRequest.get();
      expectedOptions.add(flutterOption);
      verifyQueries(expectedOptions);
      // Re-add to queue to poll for reviews
      assert(pubsub.messagesQueue.isNotEmpty);
    });

    test('All messages are pulled', () async {
      for (var i = 0; i < 3; i++) {
        final pullRequest = generatePullRequest(
          prNumber: i,
          repoName: cocoonRepo,
        );
        unawaited(pubsub.publish('auto-submit-queue-sub', pullRequest));
      }

      checkPullRequest = CheckPullRequest(
        config: config,
        pubsub: pubsub,
        cronAuthProvider: auth,
      );
      cocoonRequest = PullRequestHelper(prNumber: 0, lastCommitHash: oid);
      final messages = await checkPullRequest.pullMessages(
        testSubscription,
        5,
        5,
      );
      expect(messages.length, 3);
    });
  });
}

QueryResult createQueryResult(PullRequestHelper pullRequest) {
  return createFakeQueryResult(
    data: <String, dynamic>{
      'repository': <String, dynamic>{
        'pullRequest': pullRequest.toEntry().cast<String, dynamic>(),
      },
    },
  );
}