def test_get(): """Test to make sure get rest call behaves as expected""" params = {'param_one': 'value_one', 'param_two': 'value_two'} def request_callback(_, uri, headers): """Callback function that asserts get request has proper paramters""" for param, value in params.items(): assert '{}={}'.format(param, value) in uri return 200, headers, json.dumps({'result': 'test'}) httpretty.register_uri(httpretty.GET, 'http://test.com:7777/api/Test', body=request_callback, content_type='application/json') resp = basic_server().get('Test', params) assert Server.json_from_response(resp)['result'] == 'test'
def test_json_from_response(): """Test to make sure json gets parsed correctly from bytes""" response = FakeResponse(b'{"thing": "result"}') json_resp = Server.json_from_response(response) assert json_resp['thing'] == 'result'