diff --git a/testcache.py b/testcache.py new file mode 100644 index 0000000000000000000000000000000000000000..e14ca9513df61d64735ee41c55f72f10428a2ed6 --- /dev/null +++ b/testcache.py @@ -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['foo@bar.com'] = None + cache.addperm('baz@bar.com') + cache['temp@bar.com'] = 'testing' + self.failUnless(cache.has_key('foo@bar.com')) + self.failUnless(not cache.has_key('hello@bar.com')) + self.assertEquals(cache['temp@bar.com'],'testing') + s = open(self.fname).readlines() + self.failUnless(len(s) == 2) + self.failUnless(s[0].startswith('foo@bar.com ')) + self.assertEquals(s[1].strip(),'baz@bar.com') + +def suite(): return unittest.makeSuite(AddrCacheTestCase,'test') + +if __name__ == '__main__': + unittest.main()