Spaces:
Running
Running
File size: 1,947 Bytes
69fb140 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 |
"""
測試 services/voice_binding.py 語音綁定狀態機
"""
import pytest
from services.voice_binding import VoiceBindingStateMachine
class TestVoiceBindingStateMachine:
"""測試 VoiceBindingStateMachine 類別"""
def test_init(self):
"""測試初始化"""
fsm = VoiceBindingStateMachine()
assert fsm.user_states == {}
def test_trigger_keywords(self):
"""測試觸發關鍵字"""
fsm = VoiceBindingStateMachine()
# 測試各種觸發關鍵字
assert fsm.check_binding_trigger("user1", "我要綁定語音登入") == "TRIGGER"
assert fsm.check_binding_trigger("user2", "語音登入綁定") == "TRIGGER"
assert fsm.check_binding_trigger("user3", "綁定語音") == "TRIGGER"
assert fsm.check_binding_trigger("user4", "設定語音登入") == "TRIGGER"
def test_no_trigger(self):
"""測試非觸發訊息"""
fsm = VoiceBindingStateMachine()
assert fsm.check_binding_trigger("user1", "你好") is None
assert fsm.check_binding_trigger("user1", "今天天氣如何") is None
assert fsm.check_binding_trigger("user1", "語音") is None
def test_clear_state(self):
"""測試清理狀態"""
fsm = VoiceBindingStateMachine()
# 觸發後清理
fsm.check_binding_trigger("user1", "綁定語音登入")
assert "user1" in fsm.user_states
fsm.clear_state("user1")
assert "user1" not in fsm.user_states
def test_is_awaiting_voice(self):
"""測試等待語音狀態"""
fsm = VoiceBindingStateMachine()
# 初始狀態
assert fsm.is_awaiting_voice("user1") is False
# 觸發後
fsm.check_binding_trigger("user1", "綁定語音登入")
assert fsm.is_awaiting_voice("user1") is True
# 清理後
fsm.clear_state("user1")
assert fsm.is_awaiting_voice("user1") is False
|