4176448fb8* Studio: enable stdio MCP servers on a loopback bind * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * Studio: address codex review on stdio MCP loopback gate * Studio: fix banner URL and preserve stdio MCP env opt-in on network binds * Studio: scope loopback to exact aliases and honor force-disable on run_server reuse * Studio: cover force-disable across a public re-bind and fix a stale test comment * Studio: keep stdio MCP off on Colab loopback launches * Studio: set tool policy before server startup --------- Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com> Co-authored-by: Lee Jackson <130007945+Imagineer99@users.noreply.github.com> Co-authored-by: imagineer99 <samleejackson0@gmail.com>
24 lines
1,002 B
Python
24 lines
1,002 B
Python
# SPDX-License-Identifier: AGPL-3.0-only
|
|
# Copyright 2026-present the Unsloth AI Inc. team. All rights reserved. See /studio/LICENSE.AGPL-3.0
|
|
|
|
"""Regression for PR #6295: the banner's canned http://127.0.0.1 URL is valid
|
|
only for the exact loopback aliases, so any other bind (e.g. a specific LAN IP)
|
|
must show its real address."""
|
|
|
|
import pytest
|
|
|
|
from startup_banner import print_studio_access_banner
|
|
|
|
|
|
def test_non_alias_loopback_shows_real_address(capsys):
|
|
# A server bound to 127.0.0.2 does not listen on 127.0.0.1.
|
|
print_studio_access_banner(port = 8891, bind_host = "127.0.0.2", display_host = "127.0.0.2")
|
|
out = capsys.readouterr().out
|
|
assert "http://127.0.0.2:8891" in out
|
|
assert "http://127.0.0.1" not in out
|
|
|
|
|
|
@pytest.mark.parametrize("host", ["127.0.0.1", "localhost"])
|
|
def test_alias_loopback_shows_canned_url(capsys, host):
|
|
print_studio_access_banner(port = 8891, bind_host = host, display_host = host)
|
|
assert "http://127.0.0.1:8891" in capsys.readouterr().out
|