21 lines
650 B
Python
21 lines
650 B
Python
import unittest
|
|
|
|
from core.websocket_client import QingjianAPIClient
|
|
|
|
|
|
class TransferGreetingContextTest(unittest.TestCase):
|
|
def test_transfer_greeting_is_non_empty(self):
|
|
c = QingjianAPIClient(enable_agent=False)
|
|
text = c._pick_transfer_greeting()
|
|
self.assertTrue(isinstance(text, str) and len(text) > 0)
|
|
|
|
def test_transfer_greeting_contains_presence_phrase(self):
|
|
c = QingjianAPIClient(enable_agent=False)
|
|
for _ in range(10):
|
|
text = c._pick_transfer_greeting()
|
|
self.assertTrue(("在" in text) or ("我在" in text))
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main(verbosity=2)
|