mirror of
https://github.com/Shu-Ying/main_plugin_farm.git
synced 2026-10-27 20:50:06 +08:00
42 lines
1.6 KiB
Python
42 lines
1.6 KiB
Python
import importlib.util
|
|
from pathlib import Path
|
|
import tempfile
|
|
import unittest
|
|
|
|
|
|
ROOT = Path(__file__).resolve().parents[1]
|
|
|
|
|
|
class DataDirectoryTests(unittest.TestCase):
|
|
def setUp(self):
|
|
spec = importlib.util.spec_from_file_location(
|
|
'farm_data_paths', ROOT / 'integrations/zhenxun/data_paths.py',
|
|
)
|
|
self.module = importlib.util.module_from_spec(spec)
|
|
spec.loader.exec_module(self.module)
|
|
self.temp = tempfile.TemporaryDirectory()
|
|
self.addCleanup(self.temp.cleanup)
|
|
self.base = Path(self.temp.name)
|
|
|
|
def test_fixed_directory(self):
|
|
self.assertEqual(
|
|
self.module.data_directory(self.base), self.base / 'farm_db',
|
|
)
|
|
custom = self.base / 'custom'
|
|
self.assertEqual(self.module.data_directory(self.base, custom), custom)
|
|
self.assertEqual(self.module.data_directory(self.base, ''),
|
|
self.base / 'farm_db')
|
|
|
|
def test_nonebot_directory_is_fixed(self):
|
|
spec = importlib.util.spec_from_file_location(
|
|
'nonebot_farm_paths', ROOT / 'integrations/nonebot/data_paths.py',
|
|
)
|
|
module = importlib.util.module_from_spec(spec)
|
|
spec.loader.exec_module(module)
|
|
self.assertEqual(module.data_directory(self.base),
|
|
self.base / 'nonebot_plugin_farm/farm_db')
|
|
custom = self.base / 'custom'
|
|
self.assertEqual(module.data_directory(self.base, custom), custom)
|
|
self.assertEqual(module.data_directory(self.base, ''),
|
|
self.base / 'nonebot_plugin_farm/farm_db')
|