74 lines
2.0 KiB
Python
74 lines
2.0 KiB
Python
|
from __future__ import annotations
|
||
|
|
|
||
|
|
import pytest
|
||
|
|
|
||
|
|
from yuxi.channels.adapters.mattermost.session import (
|
||
|
|
resolve_chat_id,
|
||
|
|
resolve_chat_type,
|
||
|
|
)
|
||
|
|
from yuxi.channels.models import ChatType
|
||
|
|
|
||
|
|
|
||
|
|
class TestResolveChatId:
|
||
|
|
def test_dm_channel(self):
|
||
|
|
assert resolve_chat_id(
|
||
|
|
{"user_id": "U001", "channel_id": "abc"},
|
||
|
|
{"type": "D"},
|
||
|
|
) == "dm_U001"
|
||
|
|
|
||
|
|
def test_dm_without_user_id(self):
|
||
|
|
assert resolve_chat_id(
|
||
|
|
{"channel_id": "abc"},
|
||
|
|
{"type": "D"},
|
||
|
|
) == "dm_unknown"
|
||
|
|
|
||
|
|
def test_public_channel(self):
|
||
|
|
assert resolve_chat_id(
|
||
|
|
{"channel_id": "ch123"},
|
||
|
|
{"type": "O"},
|
||
|
|
) == "channel_ch123"
|
||
|
|
|
||
|
|
def test_private_channel(self):
|
||
|
|
assert resolve_chat_id(
|
||
|
|
{"channel_id": "ch456"},
|
||
|
|
{"type": "P"},
|
||
|
|
) == "channel_ch456"
|
||
|
|
|
||
|
|
def test_thread_message(self):
|
||
|
|
assert resolve_chat_id(
|
||
|
|
{"channel_id": "ch123", "root_id": "post1"},
|
||
|
|
{"type": "O"},
|
||
|
|
) == "channel_ch123:thread_post1"
|
||
|
|
|
||
|
|
def test_thread_priority_over_dm(self):
|
||
|
|
assert resolve_chat_id(
|
||
|
|
{"channel_id": "abc", "root_id": "post1", "user_id": "U001"},
|
||
|
|
{"type": "D"},
|
||
|
|
) == "channel_abc:thread_post1"
|
||
|
|
|
||
|
|
|
||
|
|
class TestResolveChatType:
|
||
|
|
def test_direct_message(self):
|
||
|
|
assert resolve_chat_type({}, {"type": "D"}) == ChatType.DIRECT
|
||
|
|
|
||
|
|
def test_thread(self):
|
||
|
|
assert resolve_chat_type(
|
||
|
|
{"root_id": "post1"},
|
||
|
|
{"type": "O"},
|
||
|
|
) == ChatType.THREAD
|
||
|
|
|
||
|
|
def test_thread_priority_over_dm(self):
|
||
|
|
assert resolve_chat_type(
|
||
|
|
{"root_id": "post1"},
|
||
|
|
{"type": "D"},
|
||
|
|
) == ChatType.THREAD
|
||
|
|
|
||
|
|
def test_public_channel(self):
|
||
|
|
assert resolve_chat_type({}, {"type": "O"}) == ChatType.GROUP
|
||
|
|
|
||
|
|
def test_private_channel(self):
|
||
|
|
assert resolve_chat_type({}, {"type": "P"}) == ChatType.GROUP
|
||
|
|
|
||
|
|
def test_missing_type_defaults_to_group(self):
|
||
|
|
assert resolve_chat_type({}, {}) == ChatType.GROUP
|