fix: migrate legacy state to the first Discord server
This commit is contained in:
6 files changed
+58
-2
No files matched your search
@@ -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)
|
||||
|
||||
@@ -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()
|
||||
Reference in new issue
Block a user