Test AddrCache.

This commit is contained in:
Stuart Gathman committed 2007-01-05 21:26:03 +00:00
1 parent 732e7317f1
commit 09b671f47b
1 file changed
+31
+31
View File
@@ -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()