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'