From ddeb3731a8f65201321d56c454d55bd00220f057 Mon Sep 17 00:00:00 2001
From: Shadab Zafar <dufferzafar0@gmail.com>
Date: Mon, 8 Feb 2016 21:56:43 +0530
Subject: [PATCH] Tests for flow_export.python_code

---
 test/test_export_flow.py | 46 ++++++++++++++++++++++++++++++++++++++++
 1 file changed, 46 insertions(+)

diff --git a/test/test_export_flow.py b/test/test_export_flow.py
index d93912c8a..31f2a2075 100644
--- a/test/test_export_flow.py
+++ b/test/test_export_flow.py
@@ -31,3 +31,49 @@ def test_curl_command():
     result = """curl -H 'header:qvalue' -X PATCH 'http://address/path?query=param' --data-binary 'content'"""
     assert flow_export.curl_command(flow) == result
 
+
+def test_python_code():
+    flow = tutils.tflow(req=req_get)
+    result = ("""import requests\n\n"""
+              """url = 'http://address/path'\n\n"""
+              """headers = {\n"""
+              """    'header': 'qvalue',\n"""
+              """}\n\n"""
+              """response = requests.request(\n"""
+              """    method='GET',\n"""
+              """    url=url,\n"""
+              """    headers=headers,\n"""
+              """)\n\n"""
+              """print(response.text)""")
+    assert flow_export.python_code(flow) == result
+
+    flow = tutils.tflow(req=req_post)
+    result = ("""import requests\n\n"""
+              """url = 'http://address/path'\n\n"""
+              """data = '''content'''\n\n"""
+              """response = requests.request(\n"""
+              """    method='POST',\n"""
+              """    url=url,\n"""
+              """    data=data,\n)\n\n"""
+              """print(response.text)""")
+    assert flow_export.python_code(flow) == result
+
+    flow = tutils.tflow(req=req_patch)
+    result = ("""import requests\n\n"""
+              """url = 'http://address/path'\n\n"""
+              """headers = {\n"""
+              """    'header': 'qvalue',\n"""
+              """}\n\n"""
+              """params = {\n"""
+              """    'query': 'param',\n"""
+              """}\n\n"""
+              """data = '''content'''\n\n"""
+              """response = requests.request(\n"""
+              """    method='PATCH',\n"""
+              """    url=url,\n"""
+              """    headers=headers,\n"""
+              """    params=params,\n"""
+              """    data=data,\n"""
+              """)\n\n"""
+              """print(response.text)""")
+    assert flow_export.python_code(flow) == result