Add inventory controls and enhance dashboard

This commit is contained in:
Tanner
2026-11-05 03:00:13 +00:00
parent efa2140a91
commit 659fa49ea1
13 changed files with 979 additions and 36 deletions
+79
View File
@@ -0,0 +1,79 @@
import importlib.util
from pathlib import Path
import threading
import unittest
from mosfet.protocol.types import Slot
ENTRY = Path(__file__).resolve().parents[1] / 'mosfet.py'
SPEC = importlib.util.spec_from_file_location('mosfet_web_entry', ENTRY)
mosfet_web = importlib.util.module_from_spec(SPEC)
SPEC.loader.exec_module(mosfet_web)
class FakeGame:
def __init__(self):
self.drop_calls = []
self.dump_item = None
self.dump_result = 2
def drop_inventory_slot(self, slot, whole_stack=True):
self.drop_calls.append((slot, whole_stack))
return True
def start_dump(self, item_id):
self.dump_item = item_id
return self.dump_result
class TestWebInventoryAPI(unittest.TestCase):
def setUp(self):
self.g = mosfet_web.g
self.original = {
key: self.g.get(key) for key in ('connected', 'game', 'inv')
}
self.game = FakeGame()
self.g.connected = True
self.g.game = self.game
self.g.inv = {
9: Slot(True, 1145, 3,
components={'custom_name': {'text': 'Test Paper'}})
}
self.g.reload_lock = threading.RLock()
self.client = mosfet_web.app.test_client()
def tearDown(self):
for key, value in self.original.items():
self.g[key] = value
def test_inventory_api_exposes_slot_components(self):
response = self.client.get('/api/global')
self.assertEqual(response.status_code, 200)
item = response.get_json()['inv']['9']
self.assertEqual(item['components']['custom_name']['text'], 'Test Paper')
def test_drop_endpoint_supports_one_stack_and_matching_actions(self):
one = self.client.post('/api/inventory/drop', json={
'slot': 9, 'action': 'one'})
stack = self.client.post('/api/inventory/drop', json={
'slot': 9, 'action': 'stack'})
matching = self.client.post('/api/inventory/drop', json={
'slot': 9, 'action': 'matching'})
self.assertEqual(one.status_code, 200)
self.assertEqual(stack.status_code, 200)
self.assertEqual(matching.status_code, 200)
self.assertEqual(self.game.drop_calls, [(9, False), (9, True)])
self.assertEqual(self.game.dump_item, 1145)
self.assertEqual(matching.get_json()['stacks'], 2)
def test_drop_endpoint_rejects_cross_origin_actions(self):
response = self.client.post('/api/inventory/drop', json={
'slot': 9, 'action': 'stack'},
headers={'Origin': 'https://elsewhere.example'})
self.assertEqual(response.status_code, 403)
if __name__ == '__main__':
unittest.main()