| # Copyright 2017 The Chromium Authors. All rights reserved. |
| # Use of this source code is governed by a BSD-style license that can be |
| # found in the LICENSE file. |
| |
| |
| class MockWPTGitHub(object): |
| |
| def __init__(self, pull_requests, unsuccessful_merge=False): |
| self.pull_requests = pull_requests |
| self.unsuccessful_merge = unsuccessful_merge |
| self.calls = [] |
| self.pull_requests_created = [] |
| |
| def in_flight_pull_requests(self): |
| self.calls.append('in_flight_pull_requests') |
| return self.pull_requests |
| |
| def merge_pull_request(self, number): |
| self.calls.append('merge_pull_request') |
| if self.unsuccessful_merge: |
| raise Exception('PR could not be merged: %d' % number) |
| |
| def create_pr(self, remote_branch_name, desc_title, body): |
| self.calls.append('create_pr') |
| |
| assert remote_branch_name |
| assert desc_title |
| assert body |
| |
| self.pull_requests_created.append((remote_branch_name, desc_title, body)) |
| |
| return {} |
| |
| def delete_remote_branch(self, _): |
| self.calls.append('delete_remote_branch') |