1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
|
from nose.tools import *
from lamson import spam
from lamson_tests.message_tests import *
from lamson.routing import Router
import os
ham_db = "tests/sddb"
def setup():
Router.clear_states()
Router.clear_routes()
if os.path.exists(ham_db):
os.unlink(ham_db)
def teardown():
setup()
def test_Filter():
sf = spam.Filter(ham_db, 'tests/.hammierc')
ham_msg = test_mail_request().to_message()
spam_msg = test_mail_response_plain_text().to_message()
sf.train_ham(ham_msg)
sf.train_spam(spam_msg)
sf.untrain_ham(test_mail_request().to_message())
sf.untrain_spam(spam_msg)
def test_spam_filter():
import spam_filtered_mod
sf = spam.Filter(ham_db, 'tests/.hammierc')
msg = test_mail_request()
sf.train_spam(msg.to_message())
Router.deliver(msg)
assert Router.in_state(spam_filtered_mod.SPAMMING, msg), "Spam got through"
Router.clear_states()
sf.untrain_spam(msg.to_message())
sf.train_ham(msg.to_message())
Router.deliver(msg)
assert Router.in_state(spam_filtered_mod.END, msg), "Ham didn't go through."
del spam_filtered_mod
def test_spam_filter_without_db_file():
import spam_filtered_mod
msg = test_mail_request()
Router.deliver(msg)
assert Router.in_state(spam_filtered_mod.END, msg), "Spam got through"
|
