python – 寻找assertURLEquals

2024年1月28日 214次阅读

在单元测试期间,我想将生成的URL与测试中定义的静态URL进行比较.对于这种比较,最好有一个TestCase.assertURLEqual或类似的,它可以让你比较字符串格式的两个URL,如果所有查询和片段组件都存在且相等但不一定按顺序,则结果为True.

在我自己实现之前,这个功能已经存在吗？

最佳答案 我不知道是否有内置的东西,但您可以简单地使用urlparse并检查自己的查询参数,因为默认情况下会考虑订单.

>>> import urlparse
>>> url1 = 'http://google.com/?a=1&b=2'
>>> url2 = 'http://google.com/?b=2&a=1'
>>> # parse url ignoring query params order
... def parse_url(url):
...   u = urlparse.urlparse(url)
...   q = u.query
...   u = urlparse.urlparse(u.geturl().replace(q, ''))
...   return (u, urlparse.parse_qs(q))
... 
>>> parse_url(url1)
(ParseResult(scheme='http', netloc='google.com', path='/', params='', query='', fragment=''), {'a': ['1'], 'b': ['2']})
>>> def assert_url_equals(url1, url2):
...   return parse_url(url1) == parse_url(url1)
...
>>> assert_url_equals(url1, url2)
True
点赞