def test_move_to_whitelist(self):
request = {
"client_name": "example.com",
"sender": "*****@*****.**",
"recipient": "*****@*****.**"
}
greylist.greylist_timeout = 1
greylist.auto_whitelist_threshold = 1
greylist.move_to_whitelist = True
self.assertEqual(
greylist.FAILED,
greylist.process_request(request))
time.sleep(1)
self.assertEqual(
greylist.PASSED,
greylist.process_request(request))
self.assertEqual(
greylist.PASSED,
greylist.process_request(request))
self.assertSequenceEqual(
[(0,)],
list(greylist.get_db().cursor().execute(
"SELECT COUNT(*) FROM greylist")))
def test_whitelisting(self):
request = {
"client_name": "example.com",
"sender": "*****@*****.**",
"recipient": "*****@*****.**"
}
greylist.greylist_timeout = 1
greylist.auto_whitelist_threshold = 2
self.assertEqual(
greylist.FAILED,
greylist.process_request(request))
time.sleep(1)
self.assertEqual(
greylist.PASSED,
greylist.process_request(request))
self.assertEqual(
greylist.PASSED,
greylist.process_request(request))
self.assertEqual(
greylist.PASSED,
greylist.process_request(request))
greylist.greylist_timeout = 100
self.assertEqual(
greylist.PASSED,
greylist.process_request(request))
