feat: added tests for all,clear #2
This commit is contained in:
parent
5541b94688
commit
c481ae772d
1 changed files with 33 additions and 2 deletions
|
|
@ -12,6 +12,7 @@ from evennia.scripts.tickerhandler import TickerHandler
|
||||||
from evennia.scripts.monitorhandler import MonitorHandler
|
from evennia.scripts.monitorhandler import MonitorHandler
|
||||||
import inspect
|
import inspect
|
||||||
from evennia.scripts.manager import ScriptDBManager
|
from evennia.scripts.manager import ScriptDBManager
|
||||||
|
from collections import defaultdict
|
||||||
|
|
||||||
class TestScript(BaseEvenniaTest):
|
class TestScript(BaseEvenniaTest):
|
||||||
def test_create(self):
|
def test_create(self):
|
||||||
|
|
@ -145,16 +146,16 @@ class TestExtendedLoopingCall(TestCase):
|
||||||
|
|
||||||
def dummy_func():
|
def dummy_func():
|
||||||
return 0
|
return 0
|
||||||
|
|
||||||
class TestMonitorHandler(TestCase):
|
class TestMonitorHandler(TestCase):
|
||||||
def setUp(self):
|
def setUp(self):
|
||||||
self.handler = MonitorHandler()
|
self.handler = MonitorHandler()
|
||||||
|
|
||||||
def test_add(self):
|
def test_add(self):
|
||||||
obj = mock.Mock()
|
obj = mock.Mock()
|
||||||
fieldname = "db_add"
|
fieldname = "db_add"
|
||||||
callback = dummy_func
|
callback = dummy_func
|
||||||
idstring = "test"
|
idstring = "test"
|
||||||
self.assertEquals(inspect.isfunction(callback),True)
|
|
||||||
|
|
||||||
self.handler.add(obj, fieldname, callback, idstring=idstring)
|
self.handler.add(obj, fieldname, callback, idstring=idstring)
|
||||||
|
|
||||||
|
|
@ -183,6 +184,36 @@ class TestMonitorHandler(TestCase):
|
||||||
self.handler.add(obj, fieldname, callback)
|
self.handler.add(obj, fieldname, callback)
|
||||||
self.assertNotIn(fieldname, self.handler.monitors[obj])
|
self.assertNotIn(fieldname, self.handler.monitors[obj])
|
||||||
|
|
||||||
|
def test_all(self):
|
||||||
|
obj = [mock.Mock(),mock.Mock()]
|
||||||
|
fieldname = ["db_all1","db_all2"]
|
||||||
|
callback = dummy_func
|
||||||
|
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)
|
||||||
|
|
||||||
|
output = self.handler.all()
|
||||||
|
self.assertEquals(output,
|
||||||
|
[(obj[0], fieldname[0], idstring[0], False, {}),
|
||||||
|
(obj[1], fieldname[1], idstring[1], True, {})])
|
||||||
|
|
||||||
|
def test_clear(self):
|
||||||
|
obj = mock.Mock()
|
||||||
|
fieldname = "db_add"
|
||||||
|
callback = dummy_func
|
||||||
|
idstring = "test"
|
||||||
|
|
||||||
|
self.handler.add(obj, fieldname, callback, idstring=idstring)
|
||||||
|
self.assertIn(obj, self.handler.monitors)
|
||||||
|
|
||||||
|
self.handler.clear()
|
||||||
|
self.assertNotIn(obj, self.handler.monitors)
|
||||||
|
self.assertEquals(defaultdict(lambda: defaultdict(dict)), self.handler.monitors)
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue