import pytest

from yuxi.storage.neo4j import Neo4jConnectionManager, safe_neo4j_label


def test_storage_neo4j_exports_manager():
    import yuxi.storage.neo4j as neo4j_storage

    assert neo4j_storage.Neo4jConnectionManager is Neo4jConnectionManager
    assert neo4j_storage.safe_neo4j_label is safe_neo4j_label


@pytest.mark.parametrize("label", ["kb_test", "MilvusKB", "_internal_1"])
def test_safe_neo4j_label_accepts_valid_labels(label):
    assert safe_neo4j_label(label) == label


@pytest.mark.parametrize("label", ["", "1invalid", "has-dash", "has space", "中文"])
def test_safe_neo4j_label_rejects_invalid_labels(label):
    with pytest.raises(ValueError, match="非法 Neo4j 标签"):
        safe_neo4j_label(label)


def test_neo4j_connection_manager_skips_connection_in_lite_mode(monkeypatch):
    monkeypatch.setenv("LITE_MODE", "true")

    manager = Neo4jConnectionManager()

    assert manager.driver is None
    assert manager.status == "closed"