RomanNum3ral
/
scrapy
mirror of https://github.com/scrapy/scrapy.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

added unit tests

This commit is contained in:
Umair Ashraf 2016-05-21 04:46:02 -04:00 committed by Umair Ashraf
parent 05952d00d9
commit 2c8441957e
1 changed files with 15 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
tests/test_downloadermiddleware_auth.py
View File

@ -30,3 +30,18 @@ class AuthMiddlewareTest(unittest.TestCase):
headers=dict(Authorization='Digest 123'))
assert self.mw.process_request(req, self.spider) is None
self.assertEquals(req.headers['Authorization'], b'Digest 123')
def test_auth_from_http_url(self):
req = Request('http://username:password@scrapytest.org/')
new_req = self.mw.process_request(req, self.spider)
assert new_req is not None
self.assertEquals(new_req.headers['Authorization'], b'Basic dXNlcm5hbWU6cGFzc3dvcmQ=')
self.assertTrue('@' not in new_req.url)
def test_auth_from_ftp_url(self):
req = Request('ftp://username:password@scrapytest.org/')
new_req = self.mw.process_request(req, self.spider)
assert new_req is not None
self.assertTrue('ftp_user' in new_req.meta)
self.assertTrue('ftp_password' in new_req.meta)
self.assertTrue('@' not in new_req.url)