def init():
auxiliary.install(aux_types.AUX_TYPE_ENTITY, "inventory", InventoryAux)
command.add_command("inventory", inventory_command)
command.add_command("fit", fit_command)
command.add_command("unfit", unfit_command)
command.add_command("test.add_active_warp_scanner", test_add_active_warp_scanner_command)
hook.add(hook_types.REMOVE_MOBILE, clear_inventory_hook)
def setUp(self):
UnitTestCase.setUp(self)
self.hook_star_system_reset = mock.MagicMock()
self.hook_star_system_reset.__name__ = 'hook_star_system_reset'
hook.add(hook_types.STAR_SYSTEM_RESET, self.hook_star_system_reset)
self.star_system = StarSystem()
self.star_system.reset_interval = 10
def setUp(self):
session.init()
self.open_brain_hook = mock.MagicMock()
self.open_brain_hook.__name__ = "open_brain_hook"
self.close_brain_hook = mock.MagicMock()
self.close_brain_hook.__name__ = "close_brain_hook"
hook.add(hook.HOOK_OPEN_BRAIN, self.open_brain_hook)
hook.add(hook.HOOK_CLOSE_BRAIN, self.close_brain_hook)
def test_run_catches_exceptions_in_hook_functions(self):
def raiser(name):
raise Exception()
hook.add(self.name, raiser)
try:
hook.run(self.name)
except Exception:
self.fail("hook.run must catch all exceptions in hook functions.")
def test_run_does_not_call_functions_of_other_hook_types(self):
func1 = mock.MagicMock()
func1.__name__ = "func1"
func2 = mock.MagicMock()
func2.__name__ = "func2"
hook.add(self.name, func1)
hook.add("other", func2)
hook.run(self.name)
func1.assert_called_once_with(self.name)
self.assertEqual(func2.call_count, 0)
def test_run_calls_all_functions_for_given_hook_type(self):
func1 = mock.MagicMock()
func1.__name__ = "func1"
func2 = mock.MagicMock()
func2.__name__ = "func2"
hook.add(self.name, func1)
hook.add(self.name, func2)
hook.run(self.name)
func1.assert_called_once_with(self.name)
func2.assert_called_once_with(self.name)
def init():
hook.add(hook.HOOK_OPEN_BRAIN, open_brain_hook)
hook.add(hook.HOOK_CLOSE_BRAIN, close_brain_hook)
def init(eng, env):
global engine, environment
engine = eng
environment = env
hook.add(hook.HOOK_SHUTDOWN, shutdown_hook)
def test_run_passes_arguments_to_hook_functions(self):
func = mock.MagicMock()
func.__name__ = "hook"
hook.add(self.name, func)
hook.run(self.name, "foo", "bar", baz="omg")
func.assert_called_once_with(self.name, "foo", "bar", baz="omg")
def init():
auxiliary.install(aux_types.AUX_TYPE_ENTITY, "warp", WarpAux)
hook.add(hook_types.CELESTIAL_EXIT, clear_warp_scanner)
hook.add(hook_types.STAR_SYSTEM_EXIT, clear_warp_scanner)