def test_get_consumer_scoring_404(mock_get): mock_response = Mock(status=404) mock_get.side_effect = HTTPError(mock_response, 'not found') endpoint = 'http://example.com' income = '100' zipcode = '60614' age = '23' scoring.get_consumer_scoring(endpoint, income, zipcode, age) assert mock_get.called assert_raises(HTTPError)
def main(): args = parse_arguments() validate_arguments(args) options = set_query_defaults(args) result = get_consumer_scoring(options['endpoint'], options['income'], options['zipcode'], options['age']) print(RESULT_LINE) print(str(modify_propensity(result['propensity'])))
def test_get_consumer_scoring_200(mock_get): expected_string = '{"ping": "pong"}' mock_response = Mock(status=200, return_value=expected_string) expected_json = json.loads(expected_string) print(expected_json) endpoint = 'http://example.com' income = '100' zipcode = '60606' age = '23' mock_get.return_value = mock_response result = scoring.get_consumer_scoring(endpoint, income, zipcode, age) assert result assert mock_response.json.called assert mock_get.called