class SAPluginTestCase(unittest.TestCase): """Testcases for the Stub Plugin""" def setUp(self): from ConfigParser import RawConfigParser config = RawConfigParser() config.add_section('main') config.set('main', 'prependaddedheaders', 'X-Fuglu-') config.add_section('SAPlugin') config.set('SAPlugin', 'host', '127.0.0.1') config.set('SAPlugin', 'port', '783') config.set('SAPlugin', 'timeout', '5') config.set('SAPlugin', 'retries', '3') config.set('SAPlugin', 'peruserconfig', '0') config.set('SAPlugin', 'maxsize', '500000') config.set('SAPlugin', 'spamheader', 'X-Spam-Status') config.set('SAPlugin', 'lowspamaction', 'DUNNO') config.set('SAPlugin', 'highspamaction', 'REJECT') config.set('SAPlugin', 'problemaction', 'DEFER') config.set('SAPlugin', 'highspamlevel', '15') config.set('SAPlugin', 'forwardoriginal', 'False') config.set('SAPlugin', 'scanoriginal', 'False') config.set('SAPlugin', 'rejectmessage', '') # sql blacklist testfile = "/tmp/sa_test.db" if os.path.exists(testfile): os.remove(testfile) # important: 4 slashes for absolute paths! self.testdb = "sqlite:///%s" % testfile sql = """SELECT value FROM userpref WHERE preference='blacklist_from' AND username in ('@GLOBAL','%' || ${to_domain},${to_address})""" config.set('SAPlugin', 'sql_blacklist_dbconnectstring', self.testdb) config.set('SAPlugin', 'sql_blacklist_sql', sql) config.set('SAPlugin', 'check_sql_blacklist', 'False') self.candidate = SAPlugin(config) def test_score(self): suspect = Suspect('*****@*****.**', '*****@*****.**', '/dev/null') stream = """Date: Mon, 08 Sep 2008 17:33:54 +0200 To: [email protected] From: [email protected] Subject: test scanner XJS*C4JDBQADN1.NSBN3*2IDNEN*GTUBE-STANDARD-ANTI-UBE-TEST-EMAIL*C.34X """ suspect.set_source(stream) result = self.candidate.examine(suspect) if type(result) is tuple: result, message = result score = int(suspect.get_tag('SAPlugin.spamscore')) self.failUnless(score > 999, "GTUBE mails should score ~1000 , we got %s" % score) self.failUnless(result == REJECT, 'High spam should be rejected') def test_symbols(self): stream = """Date: Mon, 08 Sep 2008 17:33:54 +0200 To: [email protected] From: [email protected] Subject: test scanner XJS*C4JDBQADN1.NSBN3*2IDNEN*GTUBE-STANDARD-ANTI-UBE-TEST-EMAIL*C.34X """ spamstatus, spamscore, rules = self.candidate.safilter_symbols( stream, '*****@*****.**') self.failUnless('GTUBE' in rules, "GTUBE not found in SYMBOL scan") self.failIf(spamscore < 500) self.failUnless(spamstatus) stream2 = """Received: from mail.python.org (mail.python.org [82.94.164.166]) by bla.fuglu.org (Postfix) with ESMTPS id 395743E03A5 for <*****@*****.**>; Sun, 22 Aug 2010 18:15:11 +0200 (CEST) Date: Tue, 24 Aug 2010 09:20:57 +0200 To: [email protected] From: [email protected] Subject: test Tue, 24 Aug 2010 09:20:57 +0200 X-Mailer: swaks v20061116.0 jetmore.org/john/code/#swaks Message-Id: <*****@*****.**> This is a test mailing """ spamstatus, spamscore, rules = self.candidate.safilter_symbols( stream2, '*****@*****.**') # print rules self.failIf(spamstatus, "This message should not be detected as spam") def test_sql_blacklist(self): self.candidate.config.set('SAPlugin', 'check_sql_blacklist', 'True') suspect = Suspect('*****@*****.**', '*****@*****.**', '/dev/null') import fuglu.extensions.sql if not fuglu.extensions.sql.ENABLED: print "Excluding test that needs sqlalchemy extension" return session = fuglu.extensions.sql.get_session(self.testdb) createsql = """CREATE TABLE userpref ( username varchar(100) NOT NULL DEFAULT '', preference varchar(30) NOT NULL DEFAULT '', value varchar(100) NOT NULL DEFAULT '' )""" session.execute(createsql) self.assertEquals(self.candidate.check_sql_blacklist(suspect), DUNNO), 'sender is not blacklisted' insertsql = """INSERT INTO userpref (username,preference,value) VALUES ('%unittests.fuglu.org','blacklist_from','*@unittests.fuglu.org')""" session.execute(insertsql) self.assertEquals(self.candidate.check_sql_blacklist(suspect), REJECT), 'sender should be blacklisted' fuglu.extensions.sql.ENABLED = False self.assertEquals(self.candidate.check_sql_blacklist(suspect), DUNNO), 'problem if sqlalchemy is not available' fuglu.extensions.sql.ENABLED = True self.candidate.config.set('SAPlugin', 'sql_blacklist_sql', 'this is a buggy sql statement') self.assertEquals(self.candidate.check_sql_blacklist(suspect), DUNNO), 'error coping with db problems' # simulate unavailable db self.candidate.config.set('SAPlugin', 'sql_blacklist_dbconnectstring', 'mysql://127.0.0.1:9977/idonotexist') self.assertEquals(self.candidate.check_sql_blacklist(suspect), DUNNO), 'error coping with db problems'
class SAPluginTestCase(unittest.TestCase): """Testcases for the Stub Plugin""" def setUp(self): from ConfigParser import RawConfigParser config = RawConfigParser() config.add_section("main") config.set("main", "prependaddedheaders", "X-Fuglu-") config.add_section("SAPlugin") config.set("SAPlugin", "host", "127.0.0.1") config.set("SAPlugin", "port", "783") config.set("SAPlugin", "timeout", "5") config.set("SAPlugin", "retries", "3") config.set("SAPlugin", "peruserconfig", "0") config.set("SAPlugin", "maxsize", "500000") config.set("SAPlugin", "spamheader", "X-Spam-Status") config.set("SAPlugin", "lowspamaction", "DUNNO") config.set("SAPlugin", "highspamaction", "REJECT") config.set("SAPlugin", "problemaction", "DEFER") config.set("SAPlugin", "highspamlevel", "15") config.set("SAPlugin", "forwardoriginal", "False") config.set("SAPlugin", "scanoriginal", "False") config.set("SAPlugin", "rejectmessage", "") # sql blacklist testfile = "/tmp/sa_test.db" if os.path.exists(testfile): os.remove(testfile) # important: 4 slashes for absolute paths! self.testdb = "sqlite:///%s" % testfile sql = """SELECT value FROM userpref WHERE preference='blacklist_from' AND username in ('@GLOBAL','%' || ${to_domain},${to_address})""" config.set("SAPlugin", "sql_blacklist_dbconnectstring", self.testdb) config.set("SAPlugin", "sql_blacklist_sql", sql) config.set("SAPlugin", "check_sql_blacklist", "False") self.candidate = SAPlugin(config) def test_score(self): suspect = Suspect("*****@*****.**", "*****@*****.**", "/dev/null") stream = """Date: Mon, 08 Sep 2008 17:33:54 +0200 To: [email protected] From: [email protected] Subject: test scanner XJS*C4JDBQADN1.NSBN3*2IDNEN*GTUBE-STANDARD-ANTI-UBE-TEST-EMAIL*C.34X """ suspect.set_source(stream) result = self.candidate.examine(suspect) if type(result) is tuple: result, message = result score = int(suspect.get_tag("SAPlugin.spamscore")) self.failUnless(score > 999, "GTUBE mails should score ~1000 , we got %s" % score) self.failUnless(result == REJECT, "High spam should be rejected") def test_symbols(self): stream = """Date: Mon, 08 Sep 2008 17:33:54 +0200 To: [email protected] From: [email protected] Subject: test scanner XJS*C4JDBQADN1.NSBN3*2IDNEN*GTUBE-STANDARD-ANTI-UBE-TEST-EMAIL*C.34X """ spamstatus, spamscore, rules = self.candidate.safilter_symbols(stream, "*****@*****.**") self.failUnless("GTUBE" in rules, "GTUBE not found in SYMBOL scan") self.failIf(spamscore < 500) self.failUnless(spamstatus) stream2 = """Received: from mail.python.org (mail.python.org [82.94.164.166]) by bla.fuglu.org (Postfix) with ESMTPS id 395743E03A5 for <*****@*****.**>; Sun, 22 Aug 2010 18:15:11 +0200 (CEST) Date: Tue, 24 Aug 2010 09:20:57 +0200 To: [email protected] From: [email protected] Subject: test Tue, 24 Aug 2010 09:20:57 +0200 X-Mailer: swaks v20061116.0 jetmore.org/john/code/#swaks Message-Id: <*****@*****.**> This is a test mailing """ spamstatus, spamscore, rules = self.candidate.safilter_symbols(stream2, "*****@*****.**") # print rules self.failIf(spamstatus, "This message should not be detected as spam") def test_sql_blacklist(self): self.candidate.config.set("SAPlugin", "check_sql_blacklist", "True") suspect = Suspect("*****@*****.**", "*****@*****.**", "/dev/null") import fuglu.extensions.sql if not fuglu.extensions.sql.ENABLED: print "Excluding test that needs sqlalchemy extension" return session = fuglu.extensions.sql.get_session(self.testdb) createsql = """CREATE TABLE userpref ( username varchar(100) NOT NULL DEFAULT '', preference varchar(30) NOT NULL DEFAULT '', value varchar(100) NOT NULL DEFAULT '' )""" session.execute(createsql) self.assertEquals(self.candidate.check_sql_blacklist(suspect), DUNNO), "sender is not blacklisted" insertsql = """INSERT INTO userpref (username,preference,value) VALUES ('%unittests.fuglu.org','blacklist_from','*@unittests.fuglu.org')""" session.execute(insertsql) self.assertEquals(self.candidate.check_sql_blacklist(suspect), REJECT), "sender should be blacklisted" fuglu.extensions.sql.ENABLED = False self.assertEquals(self.candidate.check_sql_blacklist(suspect), DUNNO), "problem if sqlalchemy is not available" fuglu.extensions.sql.ENABLED = True self.candidate.config.set("SAPlugin", "sql_blacklist_sql", "this is a buggy sql statement") self.assertEquals(self.candidate.check_sql_blacklist(suspect), DUNNO), "error coping with db problems" # simulate unavailable db self.candidate.config.set("SAPlugin", "sql_blacklist_dbconnectstring", "mysql://127.0.0.1:9977/idonotexist") self.assertEquals(self.candidate.check_sql_blacklist(suspect), DUNNO), "error coping with db problems"