27 lines
1 KiB
Python
27 lines
1 KiB
Python
from unittest.mock import Mock
|
|
from .. import utils
|
|
|
|
|
|
class TestOrigin:
|
|
def test_simple_http(self):
|
|
"""should return the correct origin for a vanilla HTTP site"""
|
|
req = Mock(scheme='http', site=Mock(domain='lemoncurry.test'))
|
|
assert utils.origin(req) == 'http://lemoncurry.test'
|
|
|
|
def test_simple_https(self):
|
|
"""should return the correct origin for a vanilla HTTPS site"""
|
|
req = Mock(scheme='https', site=Mock(domain='secure.lemoncurry.test'))
|
|
assert utils.origin(req) == 'https://secure.lemoncurry.test'
|
|
|
|
|
|
class TestUri:
|
|
def test_siteroot(self):
|
|
"""should return correct full URI for requests to the site root"""
|
|
req = Mock(scheme='https', path='/', site=Mock(domain='l.test'))
|
|
assert utils.uri(req) == 'https://l.test/'
|
|
|
|
def test_path(self):
|
|
"""should return correct full URI for requests with a path"""
|
|
req = Mock(scheme='https', path='/notes/23', site=Mock(domain='l.tst'))
|
|
assert utils.uri(req) == 'https://l.tst/notes/23'
|