From db658b12edf9a44e40ca79209a652e839ffa78dd Mon Sep 17 00:00:00 2001
From: Rajat Gupta <35985127+rjt-gupta@users.noreply.github.com>
Date: Tue, 4 Dec 2018 02:35:59 +0530
Subject: fix query array

fixes #3072
closes #3254
---
 test/mitmproxy/contentviews/test_base.py  | 15 +++++++++++++++
 test/mitmproxy/contentviews/test_query.py |  4 ++--
 2 files changed, 17 insertions(+), 2 deletions(-)

(limited to 'test')

diff --git a/test/mitmproxy/contentviews/test_base.py b/test/mitmproxy/contentviews/test_base.py
index c94d8be2..cd879bfd 100644
--- a/test/mitmproxy/contentviews/test_base.py
+++ b/test/mitmproxy/contentviews/test_base.py
@@ -15,3 +15,18 @@ def test_format_dict():
     f_d = base.format_dict(d)
     with pytest.raises(StopIteration):
         next(f_d)
+
+
+def test_format_pairs():
+    d = [("a", "c"), ("b", "d")]
+    f_d = base.format_pairs(d)
+    assert next(f_d)
+
+    d = [("abc", "")]
+    f_d = base.format_pairs(d)
+    assert next(f_d)
+
+    d = []
+    f_d = base.format_pairs(d)
+    with pytest.raises(StopIteration):
+        next(f_d)
diff --git a/test/mitmproxy/contentviews/test_query.py b/test/mitmproxy/contentviews/test_query.py
index 741b23f1..1ae1b3ac 100644
--- a/test/mitmproxy/contentviews/test_query.py
+++ b/test/mitmproxy/contentviews/test_query.py
@@ -6,8 +6,8 @@ from . import full_eval
 def test_view_query():
     d = ""
     v = full_eval(query.ViewQuery())
-    f = v(d, query=multidict.MultiDict([("foo", "bar")]))
+    f = v(d, query=multidict.MultiDict([("foo", "bar"), ("foo", "baz")]))
     assert f[0] == "Query"
-    assert f[1] == [[("header", "foo: "), ("text", "bar")]]
+    assert f[1] == [[("header", "foo: "), ("text", "bar")], [("header", "foo: "), ("text", "baz")]]
 
     assert v(d) == ("Query", [])
-- 
cgit v1.2.3