class mw_greylistTest(unittest.TestCase):
def setUp(self):
self.settings = Settings()
self.settings.connection_url = 'sqlite:///:memory:'
engine = create_engine(self.settings.connection_url)
GLEntry.metadata.create_all(engine)
Session.configure(bind=engine)
self.session = Session()
self.glc = GLCandidate(self.settings, self.session)
self.glc.plugins = ActionProvider.plugins
script_path = sys.argv[0]
script_path = os.path.split(script_path)[0]
self.header_file = script_path + "/header_file.txt"
def testReadHeadersFromExistingFile(self):
self.glc.read_headers(self.header_file)
self.assertEqual('1.2.3.4',
self.glc.headers['client_address'])
def testInvalidHeaderLine(self):
self.failUnlessRaises(GLHeaderException,
self.glc._split_headers,
'foo')
def testValidHeaderLine(self):
self.assertEqual(['name', 'value'],
self.glc._split_headers('name=value\n'))
def testMultipleEqualSignHeaderLine(self):
self.assertEqual(['name', 'value==value'],
self.glc._split_headers('name=value==value\n'))
def testEmptyHeaderLine(self):
self.assertEqual(None,
self.glc._split_headers('\n'))
def testHeaderFromFileLineCount(self):
fh = open(self.header_file)
lines = fh.readlines()
self.glc.read_headers(self.header_file)
self.assertEqual(len(lines),
len(self.glc.headers))
def testHeaderAddedCorrectly(self):
self.glc.read_headers(self.header_file)
self.assertEqual(self.glc.headers['client_address'], '1.2.3.4')
def testActionShouldReturnTestForInvalidEntry(self):
entry = GLEntry(client='1.2.3.4', helo='some.domain.tld', sender='bar.tld')
self.session.add(entry)
self.session.commit()
self.glc.read_headers(self.header_file)
self.assertEqual('TEST', self.glc.get_action())
def testActionShouldReturnAllowForCurrentWL(self):
entry = GLEntry(client='1.2.3.4', helo='some.domain.tld', sender='bar.tld')
entry.status = 'W'
entry.expiry_date = datetime.now() + timedelta(hours=1)
self.session.add(entry)
self.session.commit()
self.glc.read_headers(self.header_file)
self.assertEqual('ALLOW', self.glc.get_action())
def testPerformActionShouldReturnDunnoForExpiredGL(self):
entry = GLEntry(client='1.2.3.4', helo='some.domain.tld', sender='bar.tld')
entry.status = 'G'
entry.expiry_date = datetime.now() - timedelta(minutes=5)
entry.score = 1
entry.count = 1
self.session.add(entry)
self.session.commit()
self.glc.read_headers(self.header_file)
self.assertEqual('action=DUNNO\n\n', self.glc.perform_action())
self.assertEqual(None, self.glc.score)
def testPerformActionShouldReturn450ForActiveGL(self):
entry = GLEntry(client='1.2.3.4', helo='some.domain.tld', sender='bar.tld')
entry.status = 'G'
entry.expiry_date = datetime.now() + timedelta(minutes=5)
entry.score = 1
entry.count = 1
self.session.add(entry)
self.session.commit()
self.glc.read_headers(self.header_file)
self.assertEqual("action=%s\n\n" % self.glc.settings.greylist_message,
self.glc.perform_action())
self.assertEqual(None, self.glc.score)
def testPerformActionShouldWriteScoreForInactiveWL(self):
entry = GLEntry(client='1.2.3.4', helo='some.domain.tld', sender='bar.tld')
entry.status = 'W'
entry.expiry_date = datetime.now() - timedelta(minutes=5)
entry.score = 1
entry.count = 1
self.session.add(entry)
self.session.commit()
self.glc.read_headers(self.header_file)
self.glc.perform_action()
self.assertNotEqual(None, self.glc.score)
def testShouldAddTestEntryAsGreylisted(self):
self.glc.read_headers(self.header_file)
self.assertEqual("action=%s\n\n" % self.glc.settings.greylist_message,
self.glc.perform_action())
query = self.session.query(GLEntry)
entry = query.first()
self.assertEqual('G', entry.status)
self.assertEqual(1, entry.count)
def testShouldAddTestEntryAsWhitelisted(self):
self.glc.read_headers(self.header_file)
self.glc.get_action()
self.glc.score = 0
self.glc._handle_score()
query = self.session.query(GLEntry)
entry = query.first()
self.assertEqual('W', entry.status)
self.assertEqual(1, entry.count)
def testShouldUpdateGreylistedTestEntryCount(self):
entry = GLEntry(client='1.2.3.4', helo='some.domain.tld', sender='bar.tld')
entry.status = 'G'
last_activated = datetime.now() - timedelta(minutes=5)
entry.last_activated = last_activated
expiry_date = datetime.now() + timedelta(minutes=5)
entry.expiry_date = expiry_date
entry.score = 1
entry.count = 1
self.session.add(entry)
self.glc.read_headers(self.header_file)
self.assertEqual("action=%s\n\n" % self.glc.settings.greylist_message,
self.glc.perform_action())
query = self.session.query(GLEntry)
entry = query.first()
self.assertEqual('G', entry.status)
self.assertEqual(2, entry.count)
self.failUnless(entry.last_activated > last_activated)
self.assertEqual(entry.expiry_date, expiry_date)
def testShouldUpdateWhitelistedTestEntryCount(self):
entry = GLEntry(client='1.2.3.4', helo='some.domain.tld', sender='bar.tld')
entry.status = 'W'
last_activated = datetime.now() - timedelta(minutes=5)
entry.last_activated = last_activated
expiry_date = datetime.now() + timedelta(minutes=5)
entry.expiry_date = expiry_date
entry.score = 1
entry.count = 1
self.session.add(entry)
self.glc.read_headers(self.header_file)
self.assertEqual('action=DUNNO\n\n',
self.glc.perform_action())
query = self.session.query(GLEntry)
entry = query.first()
self.assertEqual('W', entry.status)
self.assertEqual(2, entry.count)
self.failUnless(entry.last_activated > last_activated)
self.assertEqual(entry.expiry_date, expiry_date)
def testShouldChangeExpiredGreylistedTestEntryToWhitelisted(self):
entry = GLEntry(client='1.2.3.4', helo='some.domain.tld', sender='bar.tld')
entry.status = 'G'
last_activated = datetime.now() - timedelta(minutes=5)
entry.last_activated = last_activated
expiry_date = datetime.now() - timedelta(minutes=5)
entry.expiry_date = expiry_date
entry.score = 2
entry.count = 3
self.session.add(entry)
self.glc.read_headers(self.header_file)
self.assertEqual('action=DUNNO\n\n',
self.glc.perform_action())
query = self.session.query(GLEntry)
entry = query.first()
self.assertEqual('W', entry.status)
self.assertEqual(1, entry.count)
self.failUnless(entry.last_activated > last_activated)
self.failUnless(entry.expiry_date > expiry_date)
def testShouldHandlePluginExceptionsGracefully(self):
self.glc.headers = []
try:
self.glc._do_tests()
except:
self.fail()
from mw_greylist.settings import Settings from mw_greylist.glcandidate import GLCandidate from sqlalchemy import create_engine from sqlalchemy.orm import sessionmaker settings = Settings() engine = create_engine(settings.connection_url) Session = sessionmaker(bind=engine) session = Session() candidate = GLCandidate(settings, session) candidate.read_headers() print candidate.perform_action() session.commit()