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)