Add inventory controls and enhance dashboard
This commit is contained in:
@@ -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()
|
||||
Reference in New Issue
Block a user