def test_get_scorecard_data_no_results(self, mock_requests): mock_response = mock.Mock() mock_response.ok = True mock_response.json.return_value = self.no_data_dict mock_requests.return_value = mock_response test_data = update_colleges.get_scorecard_data('example.com') self.assertEqual(mock_requests.call_count, 1) self.assertIs(test_data, None)
def test_get_scorecard_data_throttled(self, mock_requests): mock_response = mock.Mock() mock_response.ok = False mock_response.reason = "API limit reached" mock_response.status_code = 429 mock_requests.return_value = mock_response test_data = update_colleges.get_scorecard_data('example.com') self.assertEqual(mock_requests.call_count, 1) self.assertIs(test_data, None)
def test_get_scorecard_data(self, mock_requests): mock_response = mock.Mock() mock_response.json.return_value = self.mock_results mock_response.ok = True mock_requests.return_value = mock_response data = update_colleges.get_scorecard_data('example.com') self.assertEqual(mock_requests.call_count, 1) self.assertTrue(mock_requests.called_with('example.com')) self.assertEqual(type(data), dict)
def test_get_scorecard_data_ssl_error(self, mock_requests): test_data = update_colleges.get_scorecard_data('example.com') self.assertEqual(mock_requests.call_count, 1) self.assertIsNone(test_data)