22 lines
518 B
Python
22 lines
518 B
Python
|
import json
|
||
|
|
import os
|
||
|
|
import sys
|
||
|
|
|
||
|
|
import pytest
|
||
|
|
from fastapi.testclient import TestClient
|
||
|
|
|
||
|
|
sys.path.insert(
|
||
|
|
0, os.path.abspath("../../..")
|
||
|
|
) # Adds the parent directory to the system path
|
||
|
|
|
||
|
|
|
||
|
|
from unittest.mock import MagicMock
|
||
|
|
|
||
|
|
from litellm.proxy.utils import get_custom_url
|
||
|
|
|
||
|
|
|
||
|
|
def test_get_custom_url(monkeypatch):
|
||
|
|
monkeypatch.setenv("SERVER_ROOT_PATH", "/litellm")
|
||
|
|
custom_url = get_custom_url(request_base_url="http://0.0.0.0:4000", route="ui/")
|
||
|
|
assert custom_url == "http://0.0.0.0:4000/litellm/ui/"
|