def user_group_id(request): resp = Response() client = AsyncWeChatClient(APPID, SECRET, timeout=10) try: group_id = yield from client.user.get_group_id(OPENID) except Exception as e: print(e) resp.body = str(e).encode('utf-8') else: resp.body = str(group_id).encode('utf-8') return resp
def user_info(request): resp = Response() client = AsyncWeChatClient(APPID, SECRET) try: user = yield from client.user.get(OPENID) except Exception as e: print(e) resp.body = str(e).encode('utf-8') else: resp.body = json.dumps(user).encode('utf-8') return resp
def test_assign_nonbyteish_body(self): resp = Response(body=b'data') with self.assertRaises(TypeError): resp.body = 123 self.assertEqual(b'data', resp.body) self.assertEqual(4, resp.content_length)
def test_assign_nonbyteish_body(): resp = Response(body=b'data') with pytest.raises(TypeError): resp.body = 123 assert b'data' == resp.body assert 4 == resp.content_length
def test_assign_nonbyteish_body(self): req = self.make_request('GET', '/') resp = Response(req, b'data') with self.assertRaises(TypeError): resp.body = 123 self.assertEqual(b'data', resp.body) self.assertEqual(4, resp.content_length)
def _make_response(cls, xml_response, status=200, reason=None): response = Response(status=status, reason=reason) response.headers["Content-Type"] = "text/xml; charset=utf-8" xml_data = cls._build_xml(xml_response) log.debug("Sending response:\n%s", xml_data) response.body = xml_data return response
async def transform_client_exception_to_json(request: web.Request, response: web.Response) -> None: if 400 <= response.status < 500: response.content_type = "application/json" try: json.loads(response.body) except json.JSONDecodeError: response.body = HTTPClientErrorSchema().dumps({ "detail": response.body.decode("utf-8") }).encode("utf-8")
def test_assign_nonbyteish_body(): resp = Response(body=b'data') with pytest.raises(AssertionError): resp.body = 123 assert b'data' == resp.body assert 4 == resp.content_length resp._send_headers = mock.Mock() resp._start(mock.Mock()) assert resp.headers['CONTENT-LENGTH'] == '4' assert 4 == resp.content_length
def test_assign_nonbyteish_body(): resp = Response(body=b'data') with pytest.raises(ValueError): resp.body = 123 assert b'data' == resp.body assert 4 == resp.content_length resp.headers['DATE'] = 'date' resp._start(mock.Mock(version=HttpVersion11)) assert resp.headers['CONTENT-LENGTH'] == '4' assert 4 == resp.content_length
async def test_assign_nonbyteish_body() -> None: resp = Response(body=b"data") with pytest.raises(ValueError): resp.body = 123 assert b"data" == resp.body assert 4 == resp.content_length resp.headers["DATE"] = "date" req = make_mocked_request("GET", "/", version=HttpVersion11) await resp._start(req) assert resp.headers["CONTENT-LENGTH"] == "4" assert 4 == resp.content_length
async def get(self): try: registry = CollectorRegistry() multiprocess.MultiProcessCollector(registry) response = Response() response.body = generate_latest(registry) response.content_type = CONTENT_TYPE_LATEST return response except Exception as ex: logging.warning(str(ex)) traceback.print_exc() raise web.HTTPBadRequest(body=json.dumps({'message': str(ex)}))
async def test_assign_nonbyteish_body() -> None: resp = Response(body=b'data') with pytest.raises(ValueError): resp.body = 123 assert b'data' == resp.body assert 4 == resp.content_length resp.headers['DATE'] = 'date' req = make_mocked_request('GET', '/', version=HttpVersion11) await resp._start(req) assert resp.headers['CONTENT-LENGTH'] == '4' assert 4 == resp.content_length
async def test_assign_nonbyteish_body(): resp = Response(body=b'data') with pytest.raises(ValueError): resp.body = 123 assert b'data' == resp.body assert 4 == resp.content_length resp.headers['DATE'] = 'date' req = make_mocked_request('GET', '/', version=HttpVersion11) await resp._start(req) assert resp.headers['CONTENT-LENGTH'] == '4' assert 4 == resp.content_length
async def test_rm_content_length_if_compression_http11() -> None: writer = mock.Mock() async def write_headers(status_line, headers): assert hdrs.CONTENT_LENGTH not in headers assert headers.get(hdrs.TRANSFER_ENCODING, "") == "chunked" writer.write_headers.side_effect = write_headers req = make_request("GET", "/", writer=writer) payload = BytesPayload(b"answer", headers={"X-Test-Header": "test"}) resp = Response(body=payload) assert resp.content_length == 6 resp.body = payload resp.enable_compression(ContentCoding.gzip) await resp.prepare(req) assert resp.content_length is None
async def test_rm_content_length_if_compression_enabled_on_payload_http11(): writer = mock.Mock() async def write_headers(status_line, headers): assert hdrs.CONTENT_LENGTH not in headers assert headers.get(hdrs.TRANSFER_ENCODING, '') == 'chunked' writer.write_headers.side_effect = write_headers req = make_request('GET', '/', writer=writer) payload = BytesPayload(b'answer', headers={"X-Test-Header": "test"}) resp = Response(body=payload) assert resp.content_length == 6 resp.body = payload resp.enable_compression(ContentCoding.gzip) await resp.prepare(req) assert resp.content_length is None
def test_remove_content_length_if_compression_enabled_on_payload_http11(): writer = mock.Mock() def write_headers(status_line, headers): assert hdrs.CONTENT_LENGTH not in headers assert headers.get(hdrs.TRANSFER_ENCODING, '') == 'chunked' writer.write_headers.side_effect = write_headers req = make_request('GET', '/', payload_writer=writer) payload = BytesPayload(b'answer', headers={"X-Test-Header": "test"}) resp = Response(body=payload) assert resp.content_length == 6 resp.body = payload resp.enable_compression(ContentCoding.gzip) yield from resp.prepare(req) assert resp.content_length is None
async def change_body(request): resp = Response() resp.body = b"Body changed" resp.content_type = 'text/plain' return resp
def change_body(request): resp = Response(request) resp.body = b"Body changed" return resp
def change_body(request): resp = Response() resp.body = b"Body changed" return resp