def test_default(self):
with requests_mock.mock() as m:
for prot in SCHEMES + ['alt']:
m.get('{}://domain.com'.format(prot),
headers={'Location': 'alt://domain.com'},
status_code=301)
self.assertEqual(force_url('domain.com'), 'http://domain.com')
def test_http(self):
with requests_mock.mock() as m:
m.get('http://domain.com',
text='',
headers={'Content-Type': 'text/html'})
self.assertEqual(force_url('domain.com'), 'http://domain.com')
self.assertEqual(m.call_count, 1)
def test_https(self):
with requests_mock.mock() as m:
m.get('http://domain.com',
headers={'Location': 'https://domain.com'},
status_code=301)
m.get('https://domain.com',
text='',
headers={'Content-Type': 'text/html'})
self.assertEqual(force_url('domain.com'), 'https://domain.com')
def test_full_url(self):
self.assertEqual(force_url('http://domain.com'), 'http://domain.com')