def test_get_open_reviews_502_retry(self): """Tests that we retry if we get a 502 response from the server.""" with mock.patch('requests.get') as mock_get: mock_get.side_effect = (FakeResponse("bad proxy gateway", status_code=502), FakeResponse("[]\n")) self.assertEqual(graph.get_open_reviews('1353131'), []) # We call twice because we retried once. self.assertEqual(2, mock_get.call_count)
def test_get_open_reviews_502_retry(self): """Tests that we retry if we get a 502 response from the server.""" with mock.patch('requests.get') as mock_get: mock_get.side_effect = ( FakeResponse("bad proxy gateway", status_code=502), FakeResponse("[]\n")) self.assertEqual(graph.get_open_reviews('1353131'), []) # We call twice because we retried once. self.assertEqual(2, mock_get.call_count)
def test_get_open_reviews(self): with mock.patch('requests.get') as mock_get: with open('elastic_recheck/tests/unit/samples/' 'gerrit-bug-query.json') as f: mock_get.return_value = FakeResponse(f.read()) self.assertEqual(graph.get_open_reviews('1288393'), [113009])
def test_get_open_reviews_empty(self): with mock.patch('requests.get') as mock_get: mock_get.return_value = FakeResponse("[]\n") self.assertEqual(graph.get_open_reviews('1353131'), [])