Skip to content

Commit

Permalink
Add test_secure_url
Browse files Browse the repository at this point in the history
  • Loading branch information
DjangoCRM committed Sep 16, 2024
1 parent d6030b8 commit 551357e
Showing 1 changed file with 30 additions and 0 deletions.
30 changes: 30 additions & 0 deletions tests/common/utils/test_secure_url.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,30 @@
from django.contrib.sites.models import Site
from django.test import RequestFactory
from django.test import TestCase
from django.urls import reverse

from common.utils.secure_url import secure_url


class TestSecureUrl(TestCase):

def test_secure_url(self):
relative_url = reverse("site:crm_company_changelist")
factory = RequestFactory()
# Create an instance of a GET request.
request = factory.get(relative_url)
self.assertEqual(relative_url, secure_url(relative_url, request))
site = Site.objects.get_current()
https_url = f"https://{site.domain}{relative_url}"
self.assertEqual(https_url, secure_url(
https_url, factory.get(https_url)))

http_url = f"http://{site.domain}{relative_url}"
self.assertEqual(https_url, secure_url(
http_url, factory.get(https_url)))

# alien domain
url = f"https://www.google.com{relative_url}"

self.assertEqual('/', secure_url(url, factory.get(url)))

0 comments on commit 551357e

Please sign in to comment.