Rekushi
e4bb0b00ba
* Update server-side database and dynamic resource code for Python 3 * Preserve existing Flipnote, UGO, NTFT, CSS, localization, and Hatena assets * Port Flipnote upload and movie endpoints * Update front page UGO menu generators * Preserve existing database and sample Flipnote data * Add regression tests for database operations and binary formats * Add server resource loading tests * Maintain compatibility with the original DSi Flipnote Studio protocol
60 lines
2.1 KiB
Python
60 lines
2.1 KiB
Python
from __future__ import annotations
|
|
|
|
import os
|
|
import shutil
|
|
import tempfile
|
|
import unittest
|
|
from pathlib import Path
|
|
|
|
import database as database_module
|
|
|
|
|
|
ROOT = Path(__file__).resolve().parents[1]
|
|
SAMPLE_CREATOR = "913304C0CC74CC1F"
|
|
SAMPLE_NAME = "74CC1F_0D3AA40F2B3A0_000"
|
|
|
|
|
|
class DatabaseTests(unittest.TestCase):
|
|
def setUp(self):
|
|
self.tmp = tempfile.TemporaryDirectory()
|
|
self.old_cwd = os.getcwd()
|
|
temp_root = Path(self.tmp.name)
|
|
shutil.copytree(ROOT / "database", temp_root / "database")
|
|
os.chdir(temp_root)
|
|
self.db = database_module.Database.__class__()
|
|
|
|
def tearDown(self):
|
|
# Prevent the per-test instance from having anything left to flush at interpreter exit.
|
|
self.db.Creator.clear()
|
|
self.db.new = False
|
|
os.chdir(self.old_cwd)
|
|
self.tmp.cleanup()
|
|
|
|
def test_read_sample(self):
|
|
self.assertTrue(self.db.CreatorExists(SAMPLE_CREATOR))
|
|
row = self.db.GetFlipnote(SAMPLE_CREATOR, SAMPLE_NAME)
|
|
self.assertTrue(row)
|
|
self.assertEqual(row[0], SAMPLE_NAME)
|
|
self.assertEqual(len(row), 9)
|
|
self.assertTrue(self.db.GetFlipnotePPM(SAMPLE_CREATOR, SAMPLE_NAME).startswith(b"PARA"))
|
|
self.assertEqual(len(self.db.GetFlipnoteTMB(SAMPLE_CREATOR, SAMPLE_NAME)), 0x6A0)
|
|
|
|
def test_counters_and_write(self):
|
|
self.assertTrue(self.db.AddView(SAMPLE_CREATOR, SAMPLE_NAME))
|
|
self.assertTrue(self.db.AddStar(SAMPLE_CREATOR, SAMPLE_NAME, 5, "green"))
|
|
self.assertTrue(self.db.AddDownload(SAMPLE_CREATOR, SAMPLE_NAME))
|
|
self.assertEqual(self.db.Views, 1)
|
|
self.assertEqual(self.db.Stars, 1)
|
|
self.assertEqual(self.db.Downloads, 1)
|
|
row = self.db.GetFlipnote(SAMPLE_CREATOR, SAMPLE_NAME, True)
|
|
self.assertEqual(int(row[1]), 3)
|
|
self.assertEqual(int(row[3]), 5)
|
|
self.assertEqual(int(row[8]), 1)
|
|
self.db.write()
|
|
text = Path("database/Creators") / SAMPLE_CREATOR / "flipnotes.dat"
|
|
self.assertIn("\t3\t", text.read_text(encoding="utf-8"))
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|