Class FakeGitHub

dev/test_merge_arrow_pr.py:55–98

class FakeGitHub:

    def __init__(self, issues=None, project_versions=None, state='open'):
        self._issues = issues
        self._project_versions = project_versions
        self._state = state
        self._transitions = []

    @property
    def issue(self):
        return self._issues[fake_issue_id].fields._asdict()

    @property
    def current_versions(self):
        return [
            v.title for v in self._project_versions if not v.state == 'closed'
        ]

    @property
    def current_fix_versions(self):
        return 'JS-0.4.0'

    @property
    def state(self):
        return self._state

    def get_issue_data(self, issue_id):
        return self._issues.get(issue_id).fields._asdict()

    def get_milestones(self):
        return [v._asdict() for v in self._project_versions]

    def assign_milestone(self, issue_id, milestone):
        self._transitions.append(
            {'action': 'assign_milestone', 'issue_id': issue_id,
             'milestone': milestone})

    def close_issue(self, issue_id, comment):
        self._transitions.append(
            {'action': 'close_issue', 'issue_id': issue_id, 'comment': comment})

    @property
    def captured_transitions(self):
        return self._transitions


class FakeCLI: