def test_get_pr_number_fail(self):
mock = MagicMock()
mock_repo = MagicMock()
mock_org = MagicMock()
mock_commit = MagicMock()
mock_commit.commit.message = 'Not a pr title'
mock_repo.get_commit.return_value = mock_commit
mock_org.get_repo.return_value = mock_repo
mock.get_organization.return_value = mock_org
client = SourceControlClient(mock, mock, mock)
og_pr_number = client.get_pr_for_commit('blah', 'bleh', 'bluh')
self.assertFalse(og_pr_number)
def test_get_pr_number_no_match(self):
mock = MagicMock()
mock_repo = MagicMock()
mock_org = MagicMock()
mock_commit = MagicMock()
mock_commit.commit.message = 'New PR format on github very cool'
mock_repo.get_commit.return_value = mock_commit
mock_org.get_repo.return_value = mock_repo
mock.get_organization.return_value = mock_org
client = SourceControlClient(mock, mock, mock)
og_pr_number = client.get_pr_for_commit('blah', 'bleh', 'bluh')
self.assertFalse(og_pr_number)
def test_get_pr_number_old_format(self):
mock = MagicMock()
mock_repo = MagicMock()
mock_org = MagicMock()
mock_commit = MagicMock()
mock_commit.commit.message = 'Merge pull request #197 from test/master'
mock_repo.get_commit.return_value = mock_commit
mock_org.get_repo.return_value = mock_repo
mock.get_organization.return_value = mock_org
client = SourceControlClient(mock, mock, mock)
og_pr_number = client.get_pr_for_commit('blah', 'bleh', 'bluh')
self.assertEqual(og_pr_number, 197)
