Removed deprecated modules and updated imports

This commit is contained in:
Alishahryar1 2026-02-19 20:38:11 -08:00
parent 2b0495dd08
commit 0c8d59e33e
18 changed files with 22 additions and 55 deletions

View file

@ -4,8 +4,8 @@ import pytest
from messaging.handler import ClaudeMessageHandler
from messaging.models import IncomingMessage
from messaging.tree_data import MessageNode, MessageTree
from messaging.tree_queue import MessageState
from messaging.trees.data import MessageNode, MessageTree
from messaging.trees.queue_manager import MessageState
@pytest.fixture

View file

@ -3,7 +3,7 @@ from unittest.mock import AsyncMock, MagicMock
import pytest
from messaging.handler import ClaudeMessageHandler
from messaging.tree_data import MessageState
from messaging.trees.data import MessageState
async def _gen_session(events):

View file

@ -4,7 +4,7 @@ from unittest.mock import MagicMock
import pytest
from messaging.handler import ClaudeMessageHandler
from messaging.tree_data import MessageState
from messaging.trees.data import MessageState
@pytest.fixture

View file

@ -5,7 +5,7 @@ import pytest
from messaging.handler import ClaudeMessageHandler
from messaging.models import IncomingMessage
from messaging.telegram_markdown import render_markdown_to_mdv2
from messaging.tree_data import MessageNode, MessageState
from messaging.trees.data import MessageNode, MessageState
def test_render_markdown_to_mdv2_empty_returns_empty():

View file

@ -174,21 +174,21 @@ class TestTreeQueueManager:
def test_tree_queue_manager_init(self):
"""Test TreeQueueManager initialization."""
from messaging.tree_queue import TreeQueueManager
from messaging.trees.queue_manager import TreeQueueManager
mgr = TreeQueueManager()
assert mgr.get_tree_count() == 0
def test_tree_not_busy_initially(self):
"""Test tree is not busy when no messages."""
from messaging.tree_queue import TreeQueueManager
from messaging.trees.queue_manager import TreeQueueManager
mgr = TreeQueueManager()
assert mgr.is_tree_busy("nonexistent") is False
def test_get_queue_size_empty(self):
"""Test queue size is 0 for non-existent node."""
from messaging.tree_queue import TreeQueueManager
from messaging.trees.queue_manager import TreeQueueManager
mgr = TreeQueueManager()
assert mgr.get_queue_size("nonexistent") == 0
@ -197,7 +197,7 @@ class TestTreeQueueManager:
async def test_create_tree_and_enqueue(self):
"""Test creating a tree and enqueueing."""
from messaging.models import IncomingMessage
from messaging.tree_queue import TreeQueueManager
from messaging.trees.queue_manager import TreeQueueManager
mgr = TreeQueueManager()
processed = []
@ -217,7 +217,7 @@ class TestTreeQueueManager:
def test_cancel_tree_empty(self):
"""Test cancelling non-existent tree."""
from messaging.tree_queue import TreeQueueManager
from messaging.trees.queue_manager import TreeQueueManager
mgr = TreeQueueManager()
cancelled = mgr.cancel_tree("nonexistent")

View file

@ -5,7 +5,7 @@ import pytest
from messaging.handler import ClaudeMessageHandler
from messaging.models import IncomingMessage
from messaging.session import SessionStore
from messaging.tree_queue import TreeQueueManager
from messaging.trees.queue_manager import TreeQueueManager
@pytest.mark.asyncio

View file

@ -5,8 +5,8 @@ import asyncio
import pytest
from messaging.models import IncomingMessage
from messaging.tree_data import MessageNode, MessageState, MessageTree
from messaging.tree_queue import TreeQueueManager
from messaging.trees.data import MessageNode, MessageState, MessageTree
from messaging.trees.queue_manager import TreeQueueManager
def _make_incoming(text: str = "hello", msg_id: str = "m1") -> IncomingMessage:

View file

@ -5,8 +5,8 @@ from unittest.mock import AsyncMock, MagicMock
import pytest
from messaging.models import IncomingMessage
from messaging.tree_data import MessageNode, MessageState, MessageTree
from messaging.tree_processor import TreeQueueProcessor
from messaging.trees.data import MessageNode, MessageState, MessageTree
from messaging.trees.processor import TreeQueueProcessor
@pytest.fixture

View file

@ -5,7 +5,7 @@ import asyncio
import pytest
from messaging.models import IncomingMessage
from messaging.tree_queue import (
from messaging.trees.queue_manager import (
MessageNode,
MessageState,
MessageTree,

View file

@ -3,8 +3,8 @@ from unittest.mock import MagicMock
import pytest
from messaging.models import IncomingMessage
from messaging.tree_data import MessageNode, MessageState, MessageTree
from messaging.tree_repository import TreeRepository
from messaging.trees.data import MessageNode, MessageState, MessageTree
from messaging.trees.repository import TreeRepository
@pytest.fixture