def test_pull_requests(self):
pr1 = pullrequest_factory("PR1")
pr2 = pullrequest_factory("PR2")
pr3 = pullrequest_factory("PR3")
req1 = Requirement.parse("django", 0)
req1.pull_request = pr1
req2 = Requirement.parse("pyramid", 0)
req2.pull_request = pr1
req3 = Requirement.parse("flask", 0)
req3.pull_request = pr2
req4 = Requirement.parse("elixir", 0)
req4.pull_request = pr3
reqs = RequirementsBundle()
req_file = RequirementFile(path="foo", content='')
req_file._requirements = [
req1, req2, req3, req4
]
reqs.append(req_file)
self.assertEqual(
[pr1, pr2, pr3],
[pr for pr in reqs.pull_requests()]
)
