Python ProxyFlask.format_response Examples

Example #1

File: test_proxy_flask.py Project: avivey/shadow

def test_format_response_resp():
    from shadow.proxy.web import ProxyFlask
    svc = mock.Mock()

    app = ProxyFlask(
        svc,
        ['localhost'],
        ['localhost'],
    )

    resp, expected, elapsed_time = _mock_response()

    eq_(app.format_response(resp, elapsed_time=elapsed_time), expected)

Example #2

def test_format_response_resp():
    from shadow.proxy.web import ProxyFlask
    svc = mock.Mock()

    app = ProxyFlask(
        svc,
        ['localhost'],
        ['localhost'],
    )

    resp, expected, elapsed_time = _mock_response()

    eq_(app.format_response(resp, elapsed_time=elapsed_time), expected)

Example #3

def test_format_response_unknown():
    from shadow.proxy.web import ProxyFlask
    svc = mock.Mock()

    resp = mock.Mock()

    app = ProxyFlask(svc, ['localhost'], ['localhost'])
    elapsed_time = 1337.0

    expected = {
        'repr': repr(resp),
        'elapsed_time': elapsed_time,
        'type': 'unknown'
    }

    eq_(app.format_response(resp, elapsed_time=elapsed_time), expected)

Example #4

def test_format_response_exception():
    from shadow.proxy.web import ProxyFlask
    svc = mock.Mock()

    resp = mock.Mock(spec=Exception)
    resp.message = "Exception Message!!!!!"

    app = ProxyFlask(svc, ['localhost'], ['localhost'])
    elapsed_time = 1337.0

    expected = {
        'message': resp.message,
        'repr': repr(resp),
        'elapsed_time': elapsed_time,
        'type': 'exception'
    }

    eq_(app.format_response(resp, elapsed_time=elapsed_time), expected)

Example #5

File: test_proxy_flask.py Project: avivey/shadow

def test_format_response_unknown():
    from shadow.proxy.web import ProxyFlask
    svc = mock.Mock()

    resp = mock.Mock()

    app = ProxyFlask(
        svc,
        ['localhost'],
        ['localhost']
    )
    elapsed_time = 1337.0

    expected = {
                'repr': repr(resp),
                'elapsed_time': elapsed_time,
                'type': 'unknown'
            }

    eq_(app.format_response(resp, elapsed_time=elapsed_time), expected)

Example #6

File: test_proxy_flask.py Project: avivey/shadow

def test_format_response_exception():
    from shadow.proxy.web import ProxyFlask
    svc = mock.Mock()

    resp = mock.Mock(spec=Exception)
    resp.message = "Exception Message!!!!!"

    app = ProxyFlask(
        svc,
        ['localhost'],
        ['localhost']
    )
    elapsed_time = 1337.0

    expected = {
                'message': resp.message,
                'repr': repr(resp),
                'elapsed_time': elapsed_time,
                'type': 'exception'
            }

    eq_(app.format_response(resp, elapsed_time=elapsed_time), expected)