mirror of
https://github.com/kevin1024/vcrpy.git
synced 2025-12-09 01:03:24 +00:00
Use Content-Type based approach for body matcher
When converting objects to body, dicts and sets order can change resulting in a different but same body. This fixes the issue by comparing the enclosed data in the body rather than the body itself while still allowing raw body matching with the raw_body matcher.
This commit is contained in:
@@ -1,3 +1,5 @@
|
||||
import json
|
||||
from six.moves import urllib, xmlrpc_client
|
||||
import logging
|
||||
log = logging.getLogger(__name__)
|
||||
|
||||
@@ -30,12 +32,29 @@ def query(r1, r2):
|
||||
return r1.query == r2.query
|
||||
|
||||
|
||||
def body(r1, r2):
|
||||
def raw_body(r1, r2):
|
||||
if hasattr(r1.body, 'read') and hasattr(r2.body, 'read'):
|
||||
return r1.body.read() == r2.body.read()
|
||||
return r1.body == r2.body
|
||||
|
||||
|
||||
def body(r1, r2):
|
||||
if hasattr(r1.body, 'read') and hasattr(r2.body, 'read'):
|
||||
r1_body = r1.body.read()
|
||||
r2_body = r2.body.read()
|
||||
else:
|
||||
r1_body = r1.body
|
||||
r2_body = r2.body
|
||||
if r1.headers.get('Content-Type') == r2.headers.get('Content-Type') == 'application/x-www-form-urlencoded':
|
||||
return urllib.parse.parse_qs(r1_body) == urllib.parse.parse_qs(r2_body)
|
||||
if r1.headers.get('Content-Type') == r2.headers.get('Content-Type') == 'application/json':
|
||||
return json.loads(r1_body) == json.loads(r2_body)
|
||||
if ('xmlrpc' in r1.headers.get('User-Agent', '') and 'xmlrpc' in r2.headers.get('User-Agent', '') and
|
||||
r1.headers.get('Content-Type') == r2.headers.get('Content-Type') == 'text/xml'):
|
||||
return xmlrpc_client.loads(r1_body) == xmlrpc_client.loads(r2_body)
|
||||
return r1_body == r2_body
|
||||
|
||||
|
||||
def headers(r1, r2):
|
||||
return r1.headers == r2.headers
|
||||
|
||||
|
||||
Reference in New Issue
Block a user