Test AddrCache.
This commit is contained in:
1 parent
732e7317f1
commit
09b671f47b
1 file changed
+31
@@ -0,0 +1,31 @@
|
||||
import unittest
|
||||
import os
|
||||
|
||||
from Milter.cache import AddrCache
|
||||
|
||||
class AddrCacheTestCase(unittest.TestCase):
|
||||
|
||||
def setUp(self):
|
||||
self.fname = 'test.dat'
|
||||
self.cache = AddrCache(fname=self.fname)
|
||||
|
||||
def tearDown(self):
|
||||
os.remove(self.fname)
|
||||
|
||||
def testAdd(self):
|
||||
cache = self.cache
|
||||
cache['[email protected]'] = None
|
||||
cache.addperm('[email protected]')
|
||||
cache['[email protected]'] = 'testing'
|
||||
self.failUnless(cache.has_key('[email protected]'))
|
||||
self.failUnless(not cache.has_key('[email protected]'))
|
||||
self.assertEquals(cache['[email protected]'],'testing')
|
||||
s = open(self.fname).readlines()
|
||||
self.failUnless(len(s) == 2)
|
||||
self.failUnless(s[0].startswith('[email protected] '))
|
||||
self.assertEquals(s[1].strip(),'[email protected]')
|
||||
|
||||
def suite(): return unittest.makeSuite(AddrCacheTestCase,'test')
|
||||
|
||||
if __name__ == '__main__':
|
||||
unittest.main()
|
||||
Reference in new issue
Block a user