From 56892f885b1a306e5f78226c7b7caabccaa5ebcd Mon Sep 17 00:00:00 2001 From: Shadab Zafar Date: Mon, 8 Feb 2016 23:06:47 +0530 Subject: Tests for flow_export.raw_request --- test/test_export_flow.py | 24 +++++++++++++++++++++++- 1 file changed, 23 insertions(+), 1 deletion(-) (limited to 'test/test_export_flow.py') diff --git a/test/test_export_flow.py b/test/test_export_flow.py index 31f2a207..d026e214 100644 --- a/test/test_export_flow.py +++ b/test/test_export_flow.py @@ -4,7 +4,7 @@ from . import tutils req_get = netlib.tutils.treq( method='GET', - content=None, + content='', ) req_post = netlib.tutils.treq( @@ -77,3 +77,25 @@ def test_python_code(): """)\n\n""" """print(response.text)""") assert flow_export.python_code(flow) == result + + +def test_raw_request(): + flow = tutils.tflow(req=req_get) + result = ("""GET /path HTTP/1.1\r\n""" + """header: qvalue\r\n""" + """host: address:22\r\n\r\n""" + """""") + assert flow_export.raw_request(flow) == result + + flow = tutils.tflow(req=req_post) + result = ("""POST /path HTTP/1.1\r\n""" + """host: address:22\r\n\r\n""" + """content""") + assert flow_export.raw_request(flow) == result + + flow = tutils.tflow(req=req_patch) + result = ("""PATCH /path?query=param HTTP/1.1\r\n""" + """header: qvalue\r\n""" + """host: address:22\r\n\r\n""" + """content""") + assert flow_export.raw_request(flow) == result -- cgit v1.2.3