forked from mttu-developers/konabot
feat: evolve textfx into a mini shell
This commit is contained in:
33
tests/test_textfx_runtime_limits.py
Normal file
33
tests/test_textfx_runtime_limits.py
Normal file
@ -0,0 +1,33 @@
|
||||
import nonebot
|
||||
|
||||
nonebot.init()
|
||||
|
||||
from konabot.plugins.handle_text.__init__ import _get_textfx_user_key
|
||||
|
||||
|
||||
class DummyEvent:
|
||||
def __init__(self, self_id=None, user_id=None, group_id=None, session_id=None):
|
||||
self.self_id = self_id
|
||||
self.user_id = user_id
|
||||
self.group_id = group_id
|
||||
self._session_id = session_id
|
||||
|
||||
def get_session_id(self):
|
||||
if self._session_id is None:
|
||||
raise RuntimeError('no session')
|
||||
return self._session_id
|
||||
|
||||
|
||||
def test_textfx_user_key_group():
|
||||
evt = DummyEvent(self_id='123', user_id='456', group_id='789')
|
||||
assert _get_textfx_user_key(evt) == '123:789:456'
|
||||
|
||||
|
||||
def test_textfx_user_key_private():
|
||||
evt = DummyEvent(self_id='123', user_id='456')
|
||||
assert _get_textfx_user_key(evt) == '123:private:456'
|
||||
|
||||
|
||||
def test_textfx_user_key_session_fallback():
|
||||
evt = DummyEvent(session_id='console:alice')
|
||||
assert _get_textfx_user_key(evt) == 'session:console:alice'
|
||||
Reference in New Issue
Block a user