mirror of
https://github.com/langbot-app/LangBot.git
synced 2026-06-02 03:55:55 +00:00
43 lines
1.6 KiB
Python
43 lines
1.6 KiB
Python
"""
|
|
MessageAggregator unit tests.
|
|
"""
|
|
|
|
from importlib import import_module
|
|
|
|
import langbot_plugin.api.entities.builtin.platform.message as platform_message
|
|
import langbot_plugin.api.entities.builtin.provider.session as provider_session
|
|
|
|
|
|
def test_merge_messages_preserves_routed_by_rule_if_any_input_matches(sample_message_event, mock_adapter):
|
|
"""Merged PendingMessage should keep routed_by_rule when any input was rule-routed."""
|
|
aggregator = import_module('langbot.pkg.pipeline.aggregator')
|
|
message_aggregator = aggregator.MessageAggregator(ap=None)
|
|
|
|
first_message = aggregator.PendingMessage(
|
|
bot_uuid='test-bot-uuid',
|
|
launcher_type=provider_session.LauncherTypes.PERSON,
|
|
launcher_id=12345,
|
|
sender_id=12345,
|
|
message_event=sample_message_event,
|
|
message_chain=platform_message.MessageChain([platform_message.Plain(text='first')]),
|
|
adapter=mock_adapter,
|
|
pipeline_uuid='test-pipeline-uuid',
|
|
routed_by_rule=False,
|
|
)
|
|
second_message = aggregator.PendingMessage(
|
|
bot_uuid='test-bot-uuid',
|
|
launcher_type=provider_session.LauncherTypes.PERSON,
|
|
launcher_id=12345,
|
|
sender_id=12345,
|
|
message_event=sample_message_event,
|
|
message_chain=platform_message.MessageChain([platform_message.Plain(text='second')]),
|
|
adapter=mock_adapter,
|
|
pipeline_uuid='test-pipeline-uuid',
|
|
routed_by_rule=True,
|
|
)
|
|
|
|
merged_message = message_aggregator._merge_messages([first_message, second_message])
|
|
|
|
assert merged_message.routed_by_rule is True
|
|
assert str(merged_message.message_chain) == 'first\nsecond'
|