Stuart Gathman
45 lines
1.3 KiB
Python
45 lines
1.3 KiB
Python
import unittest
|
|
import doctest
|
|
import os
|
|
import Milter.greylist
|
|
from Milter.greylist import Greylist
|
|
|
|
class GreylistTestCase(unittest.TestCase):
|
|
|
|
def setUp(self):
|
|
self.fname = 'test.db'
|
|
|
|
def tearDown(self):
|
|
os.remove(self.fname)
|
|
|
|
def testGrey(self):
|
|
grey = Greylist(self.fname)
|
|
# first time
|
|
rc = grey.check('1.2.3.4','foo@bar.com','baz@spat.com')
|
|
self.failUnless(rc == 0)
|
|
# not in window yet
|
|
rc = grey.check('1.2.3.4','foo@bar.com','baz@spat.com',timeinc=5*60)
|
|
self.failUnless(rc == 0)
|
|
# within window
|
|
rc = grey.check('1.2.3.4','foo@bar.com','baz@spat.com',timeinc=15*60)
|
|
self.failUnless(rc == 1)
|
|
# new triple
|
|
rc = grey.check('1.2.3.5','foo@bar.com','baz@spat.com',timeinc=15*60)
|
|
self.failUnless(rc == 0)
|
|
# seen again
|
|
rc = grey.check('1.2.3.4','foo@bar.com','baz@spat.com',timeinc=5*3600)
|
|
self.failUnless(rc == 2)
|
|
# new one past expire
|
|
rc = grey.check('1.2.3.5','foo@bar.com','baz@spat.com',timeinc=5*3600)
|
|
self.failUnless(rc == 0)
|
|
# original past retain
|
|
rc = grey.check('1.2.3.4','foo@bar.com','baz@spat.com',timeinc=37*24*3600)
|
|
self.failUnless(rc == 0)
|
|
|
|
def suite():
|
|
s = unittest.makeSuite(GreylistTestCase,'test')
|
|
return s
|
|
|
|
if __name__ == '__main__':
|
|
unittest.TextTestRunner().run(suite())
|