"""GraphitiClient 初始化已初始化测试."""
import pytest
from unittest.mock import Mock, patch
from src.graphiti_client import GraphitiClient
from src.config_manager import ConfigManager
class TestGraphitiClientInitializeAlreadyInitialized:
"""GraphitiClient 初始化已初始化测试类."""
@pytest.fixture
def config_manager(self, temp_config_dir):
"""创建配置管理器."""
return ConfigManager(config_path=temp_config_dir / ".graphitiace" / "config.json")
def test_initialize_graphiti_already_initialized(self, config_manager):
"""测试初始化 Graphiti(已经初始化)."""
client = GraphitiClient(config_manager)
# 设置已初始化标志
client._graphiti_initialized = True
client.graphiti = Mock()
# 再次初始化应该直接返回
result = client._initialize_graphiti()
# 应该返回 True(因为 graphiti 不为 None)
assert result is True
def test_initialize_graphiti_already_initialized_none(self, config_manager):
"""测试初始化 Graphiti(已初始化但 graphiti 为 None)."""
client = GraphitiClient(config_manager)
# 设置已初始化标志,但 graphiti 为 None
client._graphiti_initialized = True
client.graphiti = None
# 再次初始化应该直接返回 False
result = client._initialize_graphiti()
# 应该返回 False(因为 graphiti 为 None)
assert result is False