fix: test failed

This commit is contained in:
Junyan Qin
2025-12-05 22:54:13 +08:00
parent cb7c9af25c
commit daf56e5dc2

View File

@@ -14,6 +14,8 @@ from unittest.mock import AsyncMock, Mock
import langbot_plugin.api.entities.builtin.pipeline.query as pipeline_query import langbot_plugin.api.entities.builtin.pipeline.query as pipeline_query
import langbot_plugin.api.entities.builtin.platform.message as platform_message import langbot_plugin.api.entities.builtin.platform.message as platform_message
import langbot_plugin.api.entities.builtin.platform.events as platform_events
import langbot_plugin.api.entities.builtin.platform.entities as platform_entities
import langbot_plugin.api.entities.builtin.provider.session as provider_session import langbot_plugin.api.entities.builtin.provider.session as provider_session
from langbot.pkg.pipeline import entities as pipeline_entities from langbot.pkg.pipeline import entities as pipeline_entities
@@ -159,13 +161,18 @@ def sample_message_chain():
@pytest.fixture @pytest.fixture
def sample_message_event(sample_message_chain): def sample_message_event(sample_message_chain):
"""Provides sample message event""" """Provides sample message event (FriendMessage)"""
event = Mock() sender = platform_entities.Friend(
event.sender = Mock() id=12345,
event.sender.id = 12345 nickname='TestUser',
event.time = 1609459200 # 2021-01-01 00:00:00 remark=None,
event.message_chain = sample_message_chain )
return event return platform_events.FriendMessage(
type='FriendMessage',
sender=sender,
message_chain=sample_message_chain,
time=1609459200, # 2021-01-01 00:00:00
)
@pytest.fixture @pytest.fixture