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 == []
