def test_get_repos(self): def se_build(r): if r.slug == 'mylogin/foo': return False return True r1 = Mock(spec_set=Repo) type(r1).slug = 'mylogin/foo' r2 = Mock(spec_set=Repo) type(r2).slug = 'otherlogin/foo' r3 = Mock(spec_set=Repo) type(r3).slug = 'mylogin/bar' self.mock_travis.repos.return_value = [r1, r2, r3] with patch('%s.repo_build_in_last_day' % pb) as mock_build: mock_build.side_effect = se_build res = self.cls.get_repos() assert res == ['mylogin/foo'] assert self.mock_travis.mock_calls == [ call.repos(member='mylogin') ] assert mock_build.mock_calls == [ call(r1), call(r3) ]
def test_get_repos_date_check_false(self): r1 = Mock(spec_set=Repo) type(r1).slug = 'mylogin/foo' r2 = Mock(spec_set=Repo) type(r2).slug = 'otherlogin/foo' r3 = Mock(spec_set=Repo) type(r3).slug = 'mylogin/bar' self.mock_travis.repos.return_value = [r1, r2, r3] with patch('%s.repo_build_in_last_day' % pb) as mock_build: mock_build.return_value = True res = self.cls.get_repos(date_check=False) assert res == ['mylogin/bar', 'mylogin/foo'] assert self.mock_travis.mock_calls == [ call.repos(member='mylogin') ] assert mock_build.mock_calls == []