amyangfei / gormw Goto Github PK

# coding: utf-8
import sys
from gor.middleware import AsyncioGor

def log(msg):
    """
    Logging to STDERR as STDOUT and STDIN used for data transfer
    @type msg: str or byte string
    @param msg: Message to log to STDERR
    """
    try:
        msg = str(msg) + '\n'
    except:
        pass
    sys.stderr.write(msg)
    sys.stderr.flush()


def on_request(proxy, msg, **kwargs):
    log("fromtype:1 id: {}".format(msg.id))
    proxy.on('response', on_response, idx=msg.id, req=msg)


def on_response(proxy, msg, **kwargs):
    log("fromtype:2 id: {}".format(msg.id))
    proxy.on('replay', on_replay, idx=kwargs['req'].id, req=kwargs['req'], resp=msg)


def on_replay(proxy, msg, **kwargs):
    log("fromtype:3 id: {}".format(msg.id))
    

if __name__ == '__main__':
    proxy = AsyncioGor()
    proxy.on('request', on_request)
    proxy.run()

=============================== warnings summary ===============================
tests/test_asyncio_gor.py::TestAsyncioGor::test_run
  /home/travis/build/amyangfei/GorMW/gor/asyncio_impl.py:56: DeprecationWarning: There is no current event loop
    self.io_loop = asyncio.get_event_loop()
-- Docs: https://docs.pytest.org/en/stable/how-to/capture-warnings.html

# coding: utf-8
import sys
from gor.middleware import AsyncioGor


def on_request(proxy, msg, **kwargs):
    proxy.on('response', on_response, idx=msg.id, req=msg)

def on_response(proxy, msg, **kwargs):
    proxy.on('replay', on_replay, idx=kwargs['req'].id, req=kwargs['req'], resp=msg)

def on_replay(proxy, msg, **kwargs):
    replay_status = proxy.http_status(msg.http)
    resp_status = proxy.http_status(kwargs['resp'].http)
    if replay_status != resp_status:
        sys.stderr.write('replay status [%s] diffs from response status [%s]\n' % (replay_status, resp_status))
    else:
        sys.stderr.write('replay status is same as response status\n')
    sys.stderr.flush()

if __name__ == '__main__':
    proxy = AsyncioGor()
    proxy.on('request', on_request)
    proxy.run()