Add inventory controls and enhance dashboard
This commit is contained in:
@@ -0,0 +1,51 @@
|
||||
import unittest
|
||||
|
||||
from munch import Munch
|
||||
|
||||
from mosfet.commands import Commands
|
||||
|
||||
|
||||
class FakeChat:
|
||||
def __init__(self):
|
||||
self.sent = []
|
||||
|
||||
def set_handler(self, handler):
|
||||
self.handler = handler
|
||||
|
||||
def send(self, text):
|
||||
self.sent.append(text)
|
||||
|
||||
|
||||
class FakeGame:
|
||||
def __init__(self):
|
||||
self.dump_result = 2
|
||||
self.drop_result = True
|
||||
self.dump_item = None
|
||||
|
||||
def drop_stack(self):
|
||||
return self.drop_result
|
||||
|
||||
def start_dump(self, item_id):
|
||||
self.dump_item = item_id
|
||||
return self.dump_result
|
||||
|
||||
|
||||
class TestInventoryCommands(unittest.TestCase):
|
||||
def setUp(self):
|
||||
self.chat = FakeChat()
|
||||
self.game = FakeGame()
|
||||
self.commands = Commands(Munch(
|
||||
name='mosfet1', chat=self.chat, game=self.game))
|
||||
|
||||
def test_drop_command_is_enabled_for_bot(self):
|
||||
self.commands.handle_chat(('CHAT', 'tester', 'mosfet1: 1drop'))
|
||||
self.assertEqual(self.chat.sent, ['ok'])
|
||||
|
||||
def test_dump_command_starts_dump_for_item_id(self):
|
||||
self.commands.handle_chat(('CHAT', 'tester', 'mosfet1: 1dump 62'))
|
||||
self.assertEqual(self.game.dump_item, 62)
|
||||
self.assertEqual(self.chat.sent, ['ok'])
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
unittest.main()
|
||||
Reference in New Issue
Block a user