fix: migrate legacy state to the first Discord server

This commit is contained in:
mmacierzanka committed 2026-08-21 16:18:57 +02:00
1 parent 64b428555c
commit 32fd296d9e
6 files changed
+58 -2

No files matched your search

+1
View File
@@ -14,6 +14,7 @@ class GuildConfigRepositoryTest(unittest.TestCase):
path.write_text(json.dumps({"default_plays": 7, "10": 123}), encoding="utf-8")
repository = GuildConfigRepository(JsonFileStore(str(path)), fallback_plays=5)
repository.migrate_legacy(10)
self.assertEqual(repository.get_default_plays(10), 7)
self.assertEqual(repository.get_notification_channel_id(10), 123)
self.assertEqual(repository.get_default_plays(20), 5)
+23
View File
@@ -0,0 +1,23 @@
import tempfile
import unittest
from pathlib import Path
from zyrarbot.storage.uploads_repository import UploadsRepository
class UploadsRepositoryTest(unittest.TestCase):
def test_legacy_root_files_are_assigned_to_first_guild(self):
with tempfile.TemporaryDirectory() as directory:
root = Path(directory)
(root / "legacy.mp3").write_bytes(b"audio")
repository = UploadsRepository(str(root))
repository.migrate_legacy(10)
self.assertEqual([item.filename for item in repository.list(10)], ["legacy.mp3"])
self.assertFalse((root / "legacy.mp3").exists())
self.assertEqual(repository.list(20), [])
if __name__ == "__main__":
unittest.main()