"""
Unit tests for deephaven_mcp.client.__init__
Covers importability, __all__, and is_enterprise_available attribute.
Should pass in both community-only and enterprise environments by patching sys.modules if needed.
"""
import sys
import types
# Patch sys.modules for enterprise imports BEFORE any tested imports
mock_enterprise = types.ModuleType("deephaven_enterprise")
mock_client = types.ModuleType("deephaven_enterprise.client")
mock_controller = types.ModuleType("deephaven_enterprise.client.controller")
mock_proto = types.ModuleType("deephaven_enterprise.proto")
mock_auth_pb2 = types.ModuleType("deephaven_enterprise.proto.auth_pb2")
mock_common_pb2 = types.ModuleType("deephaven_enterprise.proto.common_pb2")
mock_controller.ControllerClient = type("ControllerClient", (), {})
sys.modules["deephaven_enterprise"] = mock_enterprise
sys.modules["deephaven_enterprise.client"] = mock_client
sys.modules["deephaven_enterprise.client.controller"] = mock_controller
sys.modules["deephaven_enterprise.proto"] = mock_proto
sys.modules["deephaven_enterprise.proto.auth_pb2"] = mock_auth_pb2
sys.modules["deephaven_enterprise.proto.common_pb2"] = mock_common_pb2
import pytest
def test_import_client_module():
import deephaven_mcp.client
def test___all__():
import deephaven_mcp.client
assert isinstance(deephaven_mcp.client.__all__, list)
def test_expected_symbols_in_all():
import deephaven_mcp.client
expected = {
"ClientObjectWrapper",
"BaseSession",
"CoreSession",
"CorePlusSession",
"CorePlusAuthClient",
"CorePlusSessionFactory",
"CorePlusControllerClient",
"is_enterprise_available",
"ProtobufWrapper",
"CorePlusQueryStatus",
"CorePlusQuerySerial",
"CorePlusQueryConfig",
"CorePlusQueryState",
"CorePlusQueryInfo",
"CorePlusToken",
}
assert expected <= set(deephaven_mcp.client.__all__)
@pytest.mark.parametrize(
"symbol",
[
"ClientObjectWrapper",
"BaseSession",
"CoreSession",
"CorePlusSession",
"CorePlusAuthClient",
"CorePlusSessionFactory",
"CorePlusControllerClient",
"is_enterprise_available",
"ProtobufWrapper",
"CorePlusQueryStatus",
"CorePlusQuerySerial",
"CorePlusQueryConfig",
"CorePlusQueryState",
"CorePlusQueryInfo",
"CorePlusToken",
],
)
def test_symbol_in_module(symbol):
import deephaven_mcp.client
assert hasattr(deephaven_mcp.client, symbol)
def test_import_client_init():
import deephaven_mcp.client as client
assert hasattr(client, "CoreSession")
assert hasattr(client, "CorePlusSession")
assert hasattr(client, "CorePlusAuthClient")
assert hasattr(client, "is_enterprise_available")
def test___all__():
import deephaven_mcp.client as client
expected = {
"ClientObjectWrapper",
"BaseSession",
"CoreSession",
"CorePlusSession",
"CorePlusAuthClient",
"CorePlusSessionFactory",
"CorePlusControllerClient",
"is_enterprise_available",
"ProtobufWrapper",
"CorePlusQueryStatus",
"CorePlusQuerySerial",
"CorePlusQueryConfig",
"CorePlusQueryState",
"CorePlusQueryInfo",
"CorePlusToken",
}
assert set(client.__all__) == expected
def test_import_star_behavior():
expected = {
"ClientObjectWrapper",
"BaseSession",
"CoreSession",
"CorePlusSession",
"CorePlusAuthClient",
"CorePlusSessionFactory",
"CorePlusControllerClient",
"is_enterprise_available",
"ProtobufWrapper",
"CorePlusQueryStatus",
"CorePlusQuerySerial",
"CorePlusQueryConfig",
"CorePlusQueryState",
"CorePlusQueryInfo",
"CorePlusToken",
}
ns = {}
exec("from deephaven_mcp.client import *", ns)
for symbol in expected:
assert symbol in ns
# Optionally, check that no unexpected symbols are present (no leading underscores)
for k in ns:
if not k.startswith("__"):
assert k in expected
def test_is_enterprise_available_type():
import deephaven_mcp.client as client
assert isinstance(client.is_enterprise_available, bool)
