| OLD | NEW |
| 1 class MockWPTGitHub(object): | 1 class MockWPTGitHub(object): |
| 2 | 2 |
| 3 def __init__(self, pull_requests, unsuccessful_merge=False): | 3 def __init__(self, pull_requests, unsuccessful_merge=False): |
| 4 self.pull_requests = pull_requests | 4 self.pull_requests = pull_requests |
| 5 self.unsuccessful_merge = unsuccessful_merge | 5 self.unsuccessful_merge = unsuccessful_merge |
| 6 self.calls = [] | 6 self.calls = [] |
| 7 self.pull_requests_created = [] | 7 self.pull_requests_created = [] |
| 8 | 8 |
| 9 def in_flight_pull_requests(self): | 9 def in_flight_pull_requests(self): |
| 10 self.calls.append('in_flight_pull_requests') | 10 self.calls.append('in_flight_pull_requests') |
| 11 return self.pull_requests | 11 return self.pull_requests |
| 12 | 12 |
| 13 def merge_pull_request(self, number): | 13 def merge_pull_request(self, number): |
| 14 self.calls.append('merge_pull_request') | 14 self.calls.append('merge_pull_request') |
| 15 if self.unsuccessful_merge: | 15 if self.unsuccessful_merge: |
| 16 raise Exception('PR could not be merged: %d' % number) | 16 raise Exception('PR could not be merged: %d' % number) |
| 17 | 17 |
| 18 def create_pr(self, local_branch_name, desc_title, body): | 18 def create_pr(self, remote_branch_name, desc_title, body): |
| 19 self.calls.append('create_pr') | 19 self.calls.append('create_pr') |
| 20 | 20 |
| 21 assert local_branch_name | 21 assert remote_branch_name |
| 22 assert desc_title | 22 assert desc_title |
| 23 assert body | 23 assert body |
| 24 | 24 |
| 25 self.pull_requests_created.append((local_branch_name, desc_title, body)) | 25 self.pull_requests_created.append((remote_branch_name, desc_title, body)
) |
| 26 | 26 |
| 27 return {} | 27 return {} |
| OLD | NEW |