mirror of
https://github.com/Shu-Ying/main_plugin_farm.git
synced 2026-10-28 13:10:02 +08:00
82 lines
3.7 KiB
Python
82 lines
3.7 KiB
Python
from dataclasses import replace
|
||
from io import BytesIO
|
||
import unittest
|
||
|
||
from PIL import Image
|
||
|
||
from farm_core import Catalogue
|
||
from farm_core.domain import Account, InventoryItem, Plot, Snapshot
|
||
from farm_render import FarmRenderer
|
||
from farm_render.warehouse import (
|
||
render_warehouse, warehouse_page, warehouse_rows, warehouse_text,
|
||
)
|
||
|
||
|
||
class WarehouseTests(unittest.TestCase):
|
||
def setUp(self):
|
||
self.catalogue = Catalogue.bundled()
|
||
self.carrot = self.catalogue.get('胡萝卜')
|
||
self.berry = self.catalogue.get('草莓')
|
||
self.snapshot = Snapshot(Account('仓库', 500, 0, 0), (
|
||
InventoryItem('草莓', 3, 8, True, self.berry.plant_id),
|
||
InventoryItem('胡萝卜', 2, 10, False, self.carrot.plant_id),
|
||
InventoryItem('零库存', 0, 0, False, 999999),
|
||
), (Plot(1), Plot(2)))
|
||
|
||
def test_separate_positive_stocks_groups_and_exclude_locked_value(self):
|
||
crops = warehouse_rows(self.snapshot, self.catalogue, 'crops')
|
||
self.assertEqual([r.group for r in crops], ['可出售', '已锁定'])
|
||
text = warehouse_text(crops, self.snapshot, 'crops', 1, 1)
|
||
self.assertIn(f'可出售价值 {10*self.carrot.price:,}', text)
|
||
self.assertIn('胡萝卜（#1） 10个', text)
|
||
self.assertNotIn('零库存', text)
|
||
seeds = warehouse_rows(self.snapshot, self.catalogue, 'seeds')
|
||
self.assertEqual([r.group for r in seeds], ['单季种子', '多季种子'])
|
||
self.assertEqual([r.quantity for r in seeds], [2, 3])
|
||
self.assertIn('当前空地 2/2块', warehouse_text(
|
||
seeds, self.snapshot, 'seeds', 1, 1,
|
||
))
|
||
|
||
def test_pagination_preserves_ids_and_unknown_directory_is_not_zero(self):
|
||
items = tuple(InventoryItem('未知', 1, 2, False, i)
|
||
for i in range(9000, 9025))
|
||
snapshot = replace(self.snapshot, inventory=items)
|
||
rows = warehouse_rows(snapshot, self.catalogue, 'seeds')
|
||
self.assertEqual([r.plant_id for r in warehouse_page(rows, 2)[0]],
|
||
list(range(9012, 9024)))
|
||
self.assertEqual(warehouse_page(rows, 3)[1], 3)
|
||
self.assertEqual(rows[0].group, '信息缺失')
|
||
for page in (0, 4, True):
|
||
with self.assertRaises(ValueError):
|
||
warehouse_page(rows, page)
|
||
crops = warehouse_rows(snapshot, self.catalogue, 'crops')
|
||
self.assertIn('不含目录缺失项', warehouse_text(
|
||
crops, snapshot, 'crops', 1, 3,
|
||
))
|
||
|
||
def test_empty_stock_and_no_snapshot_mutation(self):
|
||
snapshot = replace(self.snapshot, inventory=())
|
||
rows = warehouse_rows(snapshot, self.catalogue, 'seeds')
|
||
self.assertEqual(warehouse_page(rows, 1), ((), 1))
|
||
self.assertIn('种子商店', warehouse_text(
|
||
rows, snapshot, 'seeds', 1, 1,
|
||
))
|
||
self.assertEqual(snapshot.inventory, ())
|
||
|
||
def test_render_crops_and_seeds_as_compressed_jpeg(self):
|
||
from scripts.ci_font import choose_font
|
||
renderer = FarmRenderer(choose_font())
|
||
for kind in ('crops', 'seeds'):
|
||
rows = warehouse_rows(self.snapshot, self.catalogue, kind)
|
||
content = render_warehouse(renderer, rows, self.snapshot,
|
||
kind, 1, 1)
|
||
with Image.open(BytesIO(content)) as image:
|
||
self.assertEqual(image.format, 'JPEG')
|
||
self.assertEqual(image.size, (768, 1152))
|
||
self.assertLess(len(content), 500000)
|
||
high = render_warehouse(renderer, rows, self.snapshot,
|
||
kind, 1, 1, high=True)
|
||
with Image.open(BytesIO(high)) as image:
|
||
self.assertEqual(image.size, (1024, 1536))
|
||
self.assertGreater(len(high), len(content))
|