diff --git a/test/test_ircdb.py b/test/test_ircdb.py index 1de359177..71bfdfde7 100644 --- a/test/test_ircdb.py +++ b/test/test_ircdb.py @@ -97,32 +97,60 @@ class UserCapabilitySetTestCase(unittest.TestCase): class CapabilitySetTestCase(unittest.TestCase): - def test(self): + def testContains(self): s = ircdb.CapabilitySet() + self.failIf('foo' in s) + self.failIf('!foo' in s) s.add('foo') self.failUnless('foo' in s) + self.failUnless('!foo' in s) + s.remove('foo') + self.failIf('foo' in s) self.failIf('!foo' in s) - s.add('!bar') - self.failUnless('!bar' in s) - self.failIf('bar' in s) - self.assertRaises(KeyError, s.check, 'baz') - self.assertRaises(KeyError, s.check, '!baz') - s.remove('!bar') - self.assertRaises(KeyError, s.check, '!bar') - self.assertRaises(KeyError, s.check, 'bar') + s.add('!foo') + self.failUnless('foo' in s) + self.failUnless('!foo' in s) + + def testCheck(self): + s = ircdb.CapabilitySet() + self.assertRaises(KeyError, s.check, 'foo') + self.assertRaises(KeyError, s.check, '!foo') + s.add('foo') + self.failUnless(s.check('foo')) + self.failIf(s.check('!foo')) s.remove('foo') self.assertRaises(KeyError, s.check, 'foo') self.assertRaises(KeyError, s.check, '!foo') + s.add('!foo') + self.failIf(s.check('foo')) + self.failUnless(s.check('!foo')) + s.remove('!foo') + self.assertRaises(KeyError, s.check, 'foo') + self.assertRaises(KeyError, s.check, '!foo') + + def testAdd(self): + s = ircdb.CapabilitySet() + s.add('foo') + s.add('!foo') + self.failIf(s.check('foo')) + self.failUnless(s.check('!foo')) + s.add('foo') + self.failUnless(s.check('foo')) + self.failIf(s.check('!foo')) + class UserCapabilitySetTestCase(unittest.TestCase): - def test(self): + def testOwner(self): s = ircdb.UserCapabilitySet() s.add('owner') + self.failUnless('foo' in s) + self.failUnless('!foo' in s) self.failUnless(s.check('owner')) self.failIf(s.check('!owner')) self.failIf(s.check('!foo')) self.failUnless(s.check('foo')) + class IrcUserTestCase(unittest.TestCase): def testCapabilities(self): u = ircdb.IrcUser()