def test_get_org_pr_fail(self):
mock = MagicMock()
mock_repo = MagicMock()
mock_org = MagicMock()
mock_repo.get_pull.side_effect = GithubException(404, "Not Found")
mock_org.get_repo.return_value = mock_repo
mock.get_organization.return_value = mock_org
client = SourceControlClient(mock, mock, mock)
org_pr = client.get_organisation_pull_request("tribe-called-quest", "midnight-marauders", 123)
self.assertFalse(org_pr)
def test_get_org_pr(self):
mock = MagicMock()
mock_pr = MagicMock()
mock_repo = MagicMock()
mock_org = MagicMock()
mock_pr.title = "did a thing"
mock_repo.get_pull.return_value = mock_pr
mock_org.get_repo.return_value = mock_repo
mock.get_organization.return_value = mock_org
client = SourceControlClient(mock, mock, mock)
org_pr = client.get_organisation_pull_request("tribe-called-quest", "midnight-marauders", 123)
self.assertEqual(org_pr.title, "did a thing")
