58 lines
1.9 KiB
Python
58 lines
1.9 KiB
Python
"""当前 Client 本地身份服务测试。"""
|
|
|
|
import tempfile
|
|
import unittest
|
|
from pathlib import Path
|
|
|
|
from src.current_client_service import CurrentClientService
|
|
from src.settings_repository import SettingsRepository
|
|
|
|
|
|
class CurrentClientServiceTest(unittest.TestCase):
|
|
def setUp(self):
|
|
self.temp_directory = tempfile.TemporaryDirectory()
|
|
path = Path(self.temp_directory.name) / "client.db"
|
|
self.repository = SettingsRepository(path)
|
|
self.generated_count = 0
|
|
|
|
def generate_id():
|
|
self.generated_count += 1
|
|
return f"CLIENT-GENERATED-{self.generated_count}"
|
|
|
|
self.service = CurrentClientService(self.repository, generate_id)
|
|
|
|
def tearDown(self):
|
|
self.temp_directory.cleanup()
|
|
|
|
def test_first_save_generates_id_and_later_save_reuses_it(self):
|
|
first = self.service.save(" 办公室电脑 ")
|
|
second = self.service.save("仓库电脑")
|
|
|
|
self.assertEqual(first.client_id, "CLIENT-GENERATED-1")
|
|
self.assertEqual(second.client_id, first.client_id)
|
|
self.assertEqual(second.client_name, "仓库电脑")
|
|
self.assertEqual(self.generated_count, 1)
|
|
self.assertEqual(self.service.load(), second)
|
|
|
|
def test_empty_name_is_allowed_but_more_than_50_chars_is_rejected(self):
|
|
saved = self.service.save(" ")
|
|
self.assertEqual(saved.client_name, "")
|
|
|
|
with self.assertRaisesRegex(ValueError, "最多 50"):
|
|
self.service.save("测" * 51)
|
|
|
|
self.assertEqual(self.service.load(), saved)
|
|
|
|
def test_non_string_corrupted_values_are_treated_as_missing(self):
|
|
self.repository.set("admin.client_id", {"bad": "value"})
|
|
self.repository.set("admin.client_name", 123)
|
|
|
|
saved = self.service.save("新设备")
|
|
|
|
self.assertEqual(saved.client_id, "CLIENT-GENERATED-1")
|
|
self.assertEqual(saved.client_name, "新设备")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|