mirror of
https://github.com/MarioSpore/Grinch-AP.git
synced 2025-10-21 04:01:32 -06:00
d471dcc067
* Core: lazy-load worlds in unpickler this should hopefully fix customserver's memory consumption * WebHost: move imports around to save memory in MP * MultiServer: prefer loading _speedups without pyximport This saves ~15MB per MP and speeds up module import if it was built in-place. * Tests: fix tests for changed WebHost imports * CustomServer: run GC after setup * CustomServer: cleanup exception handling
43 lines
1.3 KiB
Python
43 lines
1.3 KiB
Python
import unittest
|
|
import json
|
|
|
|
|
|
class TestDocs(unittest.TestCase):
|
|
@classmethod
|
|
def setUpClass(cls) -> None:
|
|
from WebHostLib import app as raw_app
|
|
from WebHost import get_app
|
|
raw_app.config["PONY"] = {
|
|
"provider": "sqlite",
|
|
"filename": ":memory:",
|
|
"create_db": True,
|
|
}
|
|
raw_app.config.update({
|
|
"TESTING": True,
|
|
})
|
|
app = get_app()
|
|
|
|
cls.client = app.test_client()
|
|
|
|
def testCorrectErrorEmptyRequest(self):
|
|
response = self.client.post("/api/generate")
|
|
self.assertIn("No options found. Expected file attachment or json weights.", response.text)
|
|
|
|
def testGenerationQueued(self):
|
|
options = {
|
|
"Tester1":
|
|
{
|
|
"game": "Archipelago",
|
|
"name": "Tester",
|
|
"Archipelago": {}
|
|
}
|
|
}
|
|
response = self.client.post(
|
|
"/api/generate",
|
|
data=json.dumps({"weights": options}),
|
|
content_type='application/json'
|
|
)
|
|
json_data = response.get_json()
|
|
self.assertTrue(json_data["text"].startswith("Generation of seed "))
|
|
self.assertTrue(json_data["text"].endswith(" started successfully."))
|