def test_list_root(get_ticket_from_issue, github_lib): get_ticket_from_issue.side_effect = lambda x: x + 100 issue_list = github_lib.return_value.issues.list issue_list.side_effect = MultiReturn([[1, 2], [3, 4]]).side_effect github = GitHub("MockUser/MockRepo") ret = github.list_root() assert_called_once(github_lib) assert_equal(2, issue_list.call_count) expected_kwargs = [{}, {"state": "closed"}] for args, kwargs in issue_list.call_args_list: assert_equal((), args) assert_true(kwargs in expected_kwargs) expected_kwargs.remove(kwargs) assert_equal(4, get_ticket_from_issue.call_count) get_ticket_from_issue_args = [call[0][0] for call in get_ticket_from_issue.call_args_list] assert_equal(set([1, 2, 3, 4]), set(get_ticket_from_issue_args)) assert_equal(set([101, 102, 103, 104]), set(ret))
def test_list_root(get_ticket_from_issue, github_lib): get_ticket_from_issue.side_effect = lambda x: x + 100 issue_list = github_lib.return_value.issues.list issue_list.side_effect = MultiReturn([[1, 2], [3, 4]]).side_effect github = GitHub('MockUser/MockRepo') ret = github.list_root() assert_called_once(github_lib) assert_equal(2, issue_list.call_count) expected_kwargs = [{}, {'state': 'closed'}] for args, kwargs in issue_list.call_args_list: assert_equal((), args) assert_true(kwargs in expected_kwargs) expected_kwargs.remove(kwargs) assert_equal(4, get_ticket_from_issue.call_count) get_ticket_from_issue_args = [ call[0][0] for call in get_ticket_from_issue.call_args_list ] assert_equal(set([1, 2, 3, 4]), set(get_ticket_from_issue_args)) assert_equal(set([101, 102, 103, 104]), set(ret))