refactor(query)

h2non · Jan 14, 2017 · bf95274 · bf95274
1 parent 55b32af
commit bf95274
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/pook/matchers/query.py b/pook/matchers/query.py
@@ -18,10 +18,12 @@ def test(key, param):
             if match is None:
                 return False
 
+            # Normalize param value
+            param = [param] if not isinstance(param, list) else param
+
             # Compare query params
-            for index, value in enumerate(param):
-                # Perform query param comparison
-                [self.compare(value, expect) for expect in match]
+            [[self.compare(value, expect)
+              for expect in match] for value in param]
 
             return True