def generate_chain(method='GET', expect_400=False):
base = chains.base(method=method)
chain = tester.BadLengthMessageChain(request=base.request,
expected_responses=[base.response],
forwarded_request=base.fwd_request,
server_response=base.server_response)
if expect_400:
chain.responses.append(chains.response_400(connection='close'))
return chain
def __init__(self, *args, **kwargs):
deproxy.Deproxy.__init__(self, *args, **kwargs)
base = chains.base(method='PUT')
base.request.headers['Content-Length'] = 'invalid'
base.request.build_message()
base.response = chains.response_400(connection='close')
base.fwd_request = None
self.message_chains = [base]
self.cookies = []
def create_chains(self):
chain_200 = chains.base(forward=True)
chain_200.request.body = ''.join(
['Arbitrary data ' for _ in range(300)])
chain_200.request.update()
chain_400 = deproxy.MessageChain(
request=deproxy.Request(), expected_response=chains.response_400())
return [chain_200, chain_400]
def create_chains(self):
chain_200 = chains.base(forward=True)
# Append some garbge to message.
chain_200.request.msg += ''.join(['Arbitrary data ' for _ in range(300)])
# Body is not declared in the request, so the garbage will be treated
# as a new request. 400 response will be sent and client connection
# will be closed.
chain_400 = deproxy.MessageChain(
request = deproxy.Request(),
expected_response = chains.response_400())
return [chain_200, chain_400]
def __init__(self, *args, **kwargs):
deproxy.Deproxy.__init__(self, *args, **kwargs)
base = chains.base(method='PUT')
cl = base.request.headers['Content-Length']
base.request.headers.add('Content-Length', cl)
base.request.build_message()
base.fwd_request = deproxy.Request()
base.response = chains.response_400(connection='keep-alive')
self.message_chains = [base]
self.cookies = []
def expected_response(self):
return chains.response_400(connection='close')
def expected_response(self):
return chains.response_400(connection='keep-alive')
