Add extra unit test to test #3271

evennia/scripts/tests.py

@@ -3,21 +3,20 @@ Unit tests for the scripts package
 
 """
 
-from unittest import TestCase, mock
 from collections import defaultdict
-
+from unittest import TestCase, mock
-from parameterized import parameterized
 
 from evennia import DefaultScript
 from evennia.objects.objects import DefaultObject
-from evennia.scripts.models import ObjectDoesNotExist, ScriptDB
-from evennia.scripts.scripts import DoNothing, ExtendedLoopingCall
-from evennia.utils.create import create_script
-from evennia.utils.test_resources import BaseEvenniaTest
-from evennia.scripts.tickerhandler import TickerHandler
-from evennia.scripts.monitorhandler import MonitorHandler
 from evennia.scripts.manager import ScriptDBManager
+from evennia.scripts.models import ObjectDoesNotExist, ScriptDB
+from evennia.scripts.monitorhandler import MonitorHandler
+from evennia.scripts.scripts import DoNothing, ExtendedLoopingCall
+from evennia.scripts.tickerhandler import TickerHandler
+from evennia.utils.create import create_script
 from evennia.utils.dbserialize import dbserialize
+from evennia.utils.test_resources import BaseEvenniaTest
+from parameterized import parameterized
 
 
 class TestScript(BaseEvenniaTest):
@@ -29,34 +28,38 @@ class TestScript(BaseEvenniaTest):
             self.assertFalse(errors, errors)
             mockinit.assert_called()
 
+
 class TestTickerHandler(TestCase):
-    """ Test the TickerHandler class """
+    """Test the TickerHandler class"""
 
     def test_store_key_raises_RunTimeError(self):
-        """ Test _store_key method raises RuntimeError for interval < 1 """
+        """Test _store_key method raises RuntimeError for interval < 1"""
         with self.assertRaises(RuntimeError):
-            th=TickerHandler()
+            th = TickerHandler()
             th._store_key(None, None, 0, None)
 
     def test_remove_raises_RunTimeError(self):
-       """ Test remove method raises RuntimeError for catching old ordering of arguments """
+        """Test remove method raises RuntimeError for catching old ordering of arguments"""
-       with self.assertRaises(RuntimeError):
+        with self.assertRaises(RuntimeError):
-            th=TickerHandler()
+            th = TickerHandler()
             th.remove(callback=1)
 
+
 class TestScriptDBManager(TestCase):
-    """ Test the ScriptDBManger class """
+    """Test the ScriptDBManger class"""
 
     def test_not_obj_return_empty_list(self):
-        """ Test get_all_scripts_on_obj returns empty list for falsy object """
+        """Test get_all_scripts_on_obj returns empty list for falsy object"""
         manager_obj = ScriptDBManager()
         returned_list = manager_obj.get_all_scripts_on_obj(False)
         self.assertEqual(returned_list, [])
 
+
 class TestingListIntervalScript(DefaultScript):
     """
     A script that does nothing. Used to test listing of script with nonzero intervals.
     """
+
     def at_script_creation(self):
         """
         Setup the script
@@ -66,11 +69,13 @@ class TestingListIntervalScript(DefaultScript):
         self.interval = 1
         self.repeats = 1
 
+
 class TestScriptHandler(BaseEvenniaTest):
     """
     Test the ScriptHandler class.
 
     """
+
     def setUp(self):
         self.obj, self.errors = DefaultObject.create("test_object")
 
@@ -90,6 +95,13 @@ class TestScriptHandler(BaseEvenniaTest):
         self.assertTrue("None/1" in self.str)
         self.assertTrue("1 repeats" in self.str)
 
+    def test_get_script(self):
+        "Checks that Scripthandler get function returns correct script"
+        self.obj.scripts.add(TestingListIntervalScript)
+        script = self.obj.scripts.get("interval_test")
+        self.assertTrue(bool(script))
+
+
 class TestScriptDB(TestCase):
     "Check the singleton/static ScriptDB object works correctly"
 
@@ -161,14 +173,14 @@ class TestExtendedLoopingCall(TestCase):
         loopcall._scheduleFrom.assert_called_with(121)
 
     def test_start_invalid_interval(self):
-        """ Test the .start method with interval less than zero """
+        """Test the .start method with interval less than zero"""
         with self.assertRaises(ValueError):
             callback = mock.MagicMock()
             loopcall = ExtendedLoopingCall(callback)
             loopcall.start(-1, now=True, start_delay=None, count_start=1)
 
     def test__call__when_delay(self):
-        """ Test __call__ modifies start_delay and starttime if start_delay was previously set """
+        """Test __call__ modifies start_delay and starttime if start_delay was previously set"""
         callback = mock.MagicMock()
         loopcall = ExtendedLoopingCall(callback)
         loopcall.clock.seconds = mock.MagicMock(return_value=1)
@@ -181,7 +193,7 @@ class TestExtendedLoopingCall(TestCase):
         self.assertEqual(loopcall.starttime, 1)
 
     def test_force_repeat(self):
-        """ Test forcing script to run that is scheduled to run in the future """
+        """Test forcing script to run that is scheduled to run in the future"""
         callback = mock.MagicMock()
         loopcall = ExtendedLoopingCall(callback)
         loopcall.clock.seconds = mock.MagicMock(return_value=0)
@@ -192,10 +204,12 @@ class TestExtendedLoopingCall(TestCase):
 
         callback.assert_called_once()
 
+
 def dummy_func():
-    """ Dummy function used as callback parameter """
+    """Dummy function used as callback parameter"""
     return 0
 
+
 class TestMonitorHandler(TestCase):
     """
     Test the MonitorHandler class.
@@ -220,13 +234,13 @@ class TestMonitorHandler(TestCase):
     def test_remove(self):
         """Tests that removing an object from the monitor handler works correctly"""
         obj = mock.Mock()
-        fieldname = 'db_remove'
+        fieldname = "db_remove"
         callback = dummy_func
-        idstring = 'test_remove'
+        idstring = "test_remove"
 
         """Add an object to the monitor handler and then remove it"""
-        self.handler.add(obj,fieldname,callback,idstring=idstring)
+        self.handler.add(obj, fieldname, callback, idstring=idstring)
-        self.handler.remove(obj,fieldname,idstring=idstring)
+        self.handler.remove(obj, fieldname, idstring=idstring)
         self.assertEquals(self.handler.monitors[obj][fieldname], {})
 
     def test_add_with_invalid_function(self):
@@ -240,18 +254,22 @@ class TestMonitorHandler(TestCase):
 
     def test_all(self):
         """Tests that all method correctly returns information about added objects"""
-        obj = [mock.Mock(),mock.Mock()]
+        obj = [mock.Mock(), mock.Mock()]
-        fieldname = ["db_all1","db_all2"]
+        fieldname = ["db_all1", "db_all2"]
         callback = dummy_func
-        idstring = ["test_all1","test_all2"]
+        idstring = ["test_all1", "test_all2"]
 
         self.handler.add(obj[0], fieldname[0], callback, idstring=idstring[0])
-        self.handler.add(obj[1], fieldname[1], callback, idstring=idstring[1],persistent=True)
+        self.handler.add(obj[1], fieldname[1], callback, idstring=idstring[1], persistent=True)
 
         output = self.handler.all()
-        self.assertEquals(output, 
+        self.assertEquals(
-                          [(obj[0], fieldname[0], idstring[0], False, {}),
+            output,
-                           (obj[1], fieldname[1], idstring[1], True, {})])
+            [
+                (obj[0], fieldname[0], idstring[0], False, {}),
+                (obj[1], fieldname[1], idstring[1], True, {}),
+            ],
+        )
 
     def test_clear(self):
         """Tests that the clear function correctly clears the monitor handler"""
@@ -277,7 +295,7 @@ class TestMonitorHandler(TestCase):
         category = "testattribute"
 
         """Add attribute to handler and assert that it has been added"""
-        self.handler.add(obj, fieldname, callback, idstring=idstring,category=category)
+        self.handler.add(obj, fieldname, callback, idstring=idstring, category=category)
 
         index = obj.attributes.get(fieldname, return_obj=True)
         name = "db_value[testattribute]"
@@ -287,5 +305,5 @@ class TestMonitorHandler(TestCase):
         self.assertEqual(self.handler.monitors[index][name][idstring], (callback, False, {}))
 
         """Remove attribute from the handler and assert that it is gone"""
-        self.handler.remove(obj,fieldname,idstring=idstring,category=category)
+        self.handler.remove(obj, fieldname, idstring=idstring, category=category)
         self.assertEquals(self.handler.monitors[index][name], {})