feat(multiuser): implementar sistema multiusuario gRPC y configuración compartida de IA/MCP
- Servidor gRPC: Agregar interceptores de autenticación y UserRegistry para aislar sesiones por usuario.
- Contexto de Hilos: Corregir propagación de ContextVar _current_user a hilos secundarios en ExecutionServicer.
- Configuración Compartida: Implementar herencia y deep merge de settings de IA ('ai') y servidores MCP en configfile.
- Hot-Reload: Recarga automática en caliente de la configuración compartida global ante cambios en disco.
- CLI: Agregar comandos e interfaces de usuario para autenticación (login) y administración de usuarios.
- Pruebas: Desarrollar tests unitarios completos (test_shared_ai.py) y resolver regresiones en la suite existente.
This commit is contained in:
@@ -0,0 +1,186 @@
|
||||
import os
|
||||
import pytest
|
||||
import grpc
|
||||
import argparse
|
||||
from unittest.mock import MagicMock, patch
|
||||
from connpy.connapp import connapp
|
||||
from connpy.services.provider import ServiceProvider
|
||||
from connpy.cli.user_handler import UserHandler
|
||||
from connpy.cli.login_handler import LoginHandler
|
||||
from connpy.grpc_layer.stubs import AuthClientInterceptor, AuthStub
|
||||
|
||||
@pytest.fixture
|
||||
def mock_config():
|
||||
config = MagicMock()
|
||||
config.config = {"service_mode": "local", "remote_host": "localhost:8048"}
|
||||
config.defaultdir = "/mock/default/dir"
|
||||
return config
|
||||
|
||||
@pytest.fixture
|
||||
def app_instance(mock_config):
|
||||
with patch("connpy.services.provider.ServiceProvider") as mock_provider_cls:
|
||||
mock_provider = MagicMock()
|
||||
mock_provider.context = MagicMock()
|
||||
mock_provider.nodes = MagicMock()
|
||||
mock_provider.profiles = MagicMock()
|
||||
mock_provider.config_svc = MagicMock()
|
||||
mock_provider.plugins = MagicMock()
|
||||
mock_provider.sync = MagicMock()
|
||||
mock_provider.mode = "local"
|
||||
mock_provider.remote_host = "localhost:8048"
|
||||
mock_provider_cls.return_value = mock_provider
|
||||
|
||||
app = connapp(mock_config)
|
||||
# Mock UserService on app services
|
||||
app.services.users = MagicMock()
|
||||
return app
|
||||
|
||||
class TestCLIMultiUserParsing:
|
||||
def test_parser_contains_user_login_logout(self, app_instance):
|
||||
parser, _ = app_instance.get_parser()
|
||||
|
||||
# Verify subcommands exist by finding the _SubParsersAction
|
||||
subparsers_action = None
|
||||
for action in parser._actions:
|
||||
if isinstance(action, argparse._SubParsersAction):
|
||||
subparsers_action = action
|
||||
break
|
||||
|
||||
assert subparsers_action is not None
|
||||
subcommands = subparsers_action.choices.keys()
|
||||
assert "user" in subcommands
|
||||
assert "login" in subcommands
|
||||
assert "logout" in subcommands
|
||||
|
||||
def test_user_parser_arguments(self, app_instance):
|
||||
parser, _ = app_instance.get_parser()
|
||||
|
||||
# Parse add user
|
||||
args = parser.parse_args(["user", "--add", "newguy"])
|
||||
assert args.add == ["newguy"]
|
||||
assert args.func == app_instance._user.dispatch
|
||||
|
||||
# Parse delete user
|
||||
args = parser.parse_args(["user", "--del", "oldguy"])
|
||||
assert args.delete == ["oldguy"]
|
||||
|
||||
# Parse list users
|
||||
args = parser.parse_args(["user", "--list"])
|
||||
assert args.list is True
|
||||
|
||||
# Parse show user
|
||||
args = parser.parse_args(["user", "--show", "someguy"])
|
||||
assert args.show == ["someguy"]
|
||||
|
||||
# Parse regen-password
|
||||
args = parser.parse_args(["user", "--regen-password", "someguy"])
|
||||
assert args.regen_password == ["someguy"]
|
||||
|
||||
# Parse path
|
||||
args = parser.parse_args(["user", "--add", "newguy", "--path", "/some/path"])
|
||||
assert args.add == ["newguy"]
|
||||
assert args.path == ["/some/path"]
|
||||
|
||||
def test_login_logout_parser_arguments(self, app_instance):
|
||||
parser, _ = app_instance.get_parser()
|
||||
|
||||
args = parser.parse_args(["login", "someuser"])
|
||||
assert args.username == "someuser"
|
||||
assert args.func == app_instance._login.dispatch
|
||||
|
||||
args = parser.parse_args(["logout"])
|
||||
assert args.func == app_instance._login.dispatch
|
||||
|
||||
|
||||
class TestUserHandlerDispatch:
|
||||
def test_user_handler_fails_in_remote_mode(self, app_instance):
|
||||
app_instance.services.mode = "remote"
|
||||
handler = UserHandler(app_instance)
|
||||
|
||||
args = MagicMock()
|
||||
args.add = ["testuser"]
|
||||
|
||||
with pytest.raises(SystemExit) as excinfo:
|
||||
handler.dispatch(args)
|
||||
assert excinfo.value.code == 1
|
||||
|
||||
def test_user_handler_routes_add_correctly(self, app_instance):
|
||||
app_instance.services.mode = "local"
|
||||
handler = UserHandler(app_instance)
|
||||
|
||||
args = MagicMock()
|
||||
args.add = ["newuser"]
|
||||
args.delete = None
|
||||
args.list = False
|
||||
args.show = None
|
||||
args.regen_password = None
|
||||
|
||||
with patch.object(handler, "add_user") as mock_add:
|
||||
handler.dispatch(args)
|
||||
assert args.action == "add"
|
||||
assert args.username == "newuser"
|
||||
mock_add.assert_called_once_with(args)
|
||||
|
||||
def test_user_handler_routes_list_correctly(self, app_instance):
|
||||
app_instance.services.mode = "local"
|
||||
handler = UserHandler(app_instance)
|
||||
|
||||
args = MagicMock()
|
||||
args.add = None
|
||||
args.delete = None
|
||||
args.list = True
|
||||
args.show = None
|
||||
args.regen_password = None
|
||||
|
||||
with patch.object(handler, "list_users") as mock_list:
|
||||
handler.dispatch(args)
|
||||
assert args.action == "list"
|
||||
mock_list.assert_called_once_with(args)
|
||||
|
||||
|
||||
class TestAuthClientInterceptor:
|
||||
def test_auth_client_interceptor_adds_bearer_token(self):
|
||||
# Mock token provider
|
||||
token_provider = MagicMock(return_value="my-super-secret-token")
|
||||
interceptor = AuthClientInterceptor(token_provider)
|
||||
|
||||
# Mock ClientCallDetails using namedtuple
|
||||
from collections import namedtuple
|
||||
ClientCallDetails = namedtuple('ClientCallDetails', ['method', 'timeout', 'metadata', 'credentials', 'wait_for_ready', 'compression'])
|
||||
|
||||
mock_details = ClientCallDetails(
|
||||
method="/connpy.NodeService/list_nodes",
|
||||
timeout=10,
|
||||
metadata=[],
|
||||
credentials=None,
|
||||
wait_for_ready=True,
|
||||
compression=None
|
||||
)
|
||||
|
||||
intercepted_details = interceptor._add_metadata(mock_details)
|
||||
|
||||
# Verify metadata was injected
|
||||
metadata_dict = dict(intercepted_details.metadata)
|
||||
assert "authorization" in metadata_dict
|
||||
assert metadata_dict["authorization"] == "Bearer my-super-secret-token"
|
||||
|
||||
def test_auth_client_interceptor_no_token(self):
|
||||
token_provider = MagicMock(return_value=None)
|
||||
interceptor = AuthClientInterceptor(token_provider)
|
||||
|
||||
from collections import namedtuple
|
||||
ClientCallDetails = namedtuple('ClientCallDetails', ['method', 'timeout', 'metadata', 'credentials', 'wait_for_ready', 'compression'])
|
||||
|
||||
mock_details = ClientCallDetails(
|
||||
method="/connpy.NodeService/list_nodes",
|
||||
timeout=10,
|
||||
metadata=[],
|
||||
credentials=None,
|
||||
wait_for_ready=True,
|
||||
compression=None
|
||||
)
|
||||
|
||||
intercepted_details = interceptor._add_metadata(mock_details)
|
||||
|
||||
# Verify metadata remains empty
|
||||
assert len(intercepted_details.metadata) == 0
|
||||
Reference in New Issue
Block a user