# 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')