Coverage for z.s.checker.undefineChecker.

1 files changed, 29 insertions, 0 deletions

diff --git a/src/zope/security/tests/test_checker.py b/src/zope/security/tests/test_checker.py
index 892147d..d3d7503 100644
--- a/src/zope/security/tests/test_checker.py
+++ b/src/zope/security/tests/test_checker.py
@@ -853,6 +853,34 @@ class Test_defineChecker(unittest.TestCase):
         self.assertTrue(_checkers[Foo] is checker)
 
 
+class Test_undefineChecker(unittest.TestCase):
+
+    def setUp(self):
+        from zope.security.checker import _clear
+        _clear()
+
+    def tearDown(self):
+        from zope.security.checker import _clear
+        _clear()
+
+    def _callFUT(self, type_):
+        from zope.security.checker import undefineChecker
+        return undefineChecker(type_)
+
+    def test_miss(self):
+        class Foo(object):
+            pass
+        self.assertRaises(KeyError, self._callFUT, Foo)
+
+    def test_hit(self):
+        from zope.security.checker import _checkers
+        class Foo(object):
+            pass
+        checker = _checkers[Foo] = object()
+        self._callFUT(Foo)
+        self.failIf(Foo in _checkers)
+
+
 # Pre-geddon tests start here
 
 class Test(unittest.TestCase):
@@ -1528,6 +1556,7 @@ def test_suite():
         unittest.makeSuite(Test_selectChecker),
         unittest.makeSuite(Test_getCheckerForInstancesOf),
         unittest.makeSuite(Test_defineChecker),
+        unittest.makeSuite(Test_defineChecker),
         # pre-geddon fossils
         unittest.makeSuite(Test),
         unittest.makeSuite(TestCheckerPublic),