23 lines
914 B
Python
23 lines
914 B
Python
from requests.cookies import MockRequest
|
|
|
|
from middleware.vendor_context import VendorContextManager
|
|
|
|
|
|
def test_custom_domain_detection():
|
|
# Mock request with custom domain
|
|
request = MockRequest(host="customdomain1.com")
|
|
context = VendorContextManager.detect_vendor_context(request)
|
|
assert context["detection_method"] == "custom_domain"
|
|
assert context["domain"] == "customdomain1.com"
|
|
|
|
def test_subdomain_detection():
|
|
request = MockRequest(host="vendor1.platform.com")
|
|
context = VendorContextManager.detect_vendor_context(request)
|
|
assert context["detection_method"] == "subdomain"
|
|
assert context["subdomain"] == "vendor1"
|
|
|
|
def test_path_detection():
|
|
request = MockRequest(host="localhost", path="/vendor/vendor1/")
|
|
context = VendorContextManager.detect_vendor_context(request)
|
|
assert context["detection_method"] == "path"
|
|
assert context["subdomain"] == "vendor1" |