def fake_extract_json_data(review_url, cc, checkout, _, dummy): if any(hostname in review_url for hostname in code_review_parse.KNOWN_RIETVELD_INSTANCES): return code_review_parse._extract_json_data_from_rietveld(review_url) elif any(hostname in review_url for hostname in code_review_parse.KNOWN_GERRIT_INSTANCES): return code_review_parse._extract_json_data_from_gerrit(review_url, cc, checkout)
def test_extract_json_data_from_gerrit(self, mock_gerrit_util, _): mock_cc = mock.MagicMock() checkout = None _ = code_review_parse._extract_json_data_from_gerrit( self.g_not_lgtm, mock_cc, checkout) mock_gerrit_util.assert_called_with( 'chromium-review.googlesource.com', '288240') _ = code_review_parse._extract_json_data_from_gerrit( self.g_committed, mock_cc, checkout) mock_gerrit_util.assert_called_with( 'chromium-review.googlesource.com', None) _ = code_review_parse._extract_json_data_from_gerrit( self.g_not_lgtm, mock_cc, checkout) mock_gerrit_util.assert_called_with( 'chromium-review.googlesource.com', '288240')