From 09b671f47bee9cbb5a92849787eccca1aa7bd3d7 Mon Sep 17 00:00:00 2001 From: Stuart Gathman <stuart@gathman.org> Date: Fri, 5 Jan 2007 21:26:03 +0000 Subject: [PATCH] Test AddrCache. --- testcache.py | 31 +++++++++++++++++++++++++++++++ 1 file changed, 31 insertions(+) create mode 100644 testcache.py diff --git a/testcache.py b/testcache.py new file mode 100644 index 0000000..e14ca95 --- /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() -- GitLab