wehub-resource-sync
09a3d3ab17
Copilot Setup Steps / copilot-setup-steps (push) Failing after 2s
Python check requirements.txt / check-requirements (push) Has been cancelled
Python Type-Check / python type-check (push) Has been cancelled
Update Operations Documentation / update-ops-docs (push) Has been cancelled
Check Pre-Tokenizer Hashes / pre-tokenizer-hashes (push) Has been cancelled
42 lines
992 B
Python
42 lines
992 B
Python
import pytest
|
|
from utils import *
|
|
|
|
server = ServerPreset.tinyllama2()
|
|
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def create_server():
|
|
global server
|
|
server = ServerPreset.tinyllama2()
|
|
|
|
|
|
def test_mcp_no_proxy():
|
|
global server
|
|
server.ui_mcp_proxy = False
|
|
server.start()
|
|
|
|
res = server.make_request("GET", "/cors-proxy")
|
|
assert res.status_code == 403
|
|
|
|
|
|
def test_mcp_proxy():
|
|
global server
|
|
server.ui_mcp_proxy = True
|
|
server.start()
|
|
|
|
url = f"http://{server.server_host}:{server.server_port}/cors-proxy?url=http://example.com"
|
|
res = requests.get(url)
|
|
assert res.status_code == 200
|
|
assert "Example Domain" in res.text
|
|
|
|
|
|
def test_mcp_proxy_custom_port():
|
|
global server
|
|
server.ui_mcp_proxy = True
|
|
server.start()
|
|
|
|
# try getting the server's models API via the proxy
|
|
res = server.make_request("GET", f"/cors-proxy?url=http://{server.server_host}:{server.server_port}/models")
|
|
assert res.status_code == 200
|
|
assert "data" in res.body
|