def test_modified_message(self):
"""Test if the modified message gets archived correctly"""
from fuglu.shared import Suspect
import shutil
import tempfile
tempfilename = tempfile.mktemp(
suffix='archive', prefix='fuglu-unittest', dir='/tmp')
shutil.copy(TESTDATADIR + '/helloworld.eml', tempfilename)
self.tempfiles.append(tempfilename)
self.config.set('ArchivePlugin', 'storeoriginal', '0')
candidate = ArchivePlugin(self.config)
suspect = Suspect(
'*****@*****.**', '*****@*****.**', tempfilename)
origmessage = suspect.get_source()
# modify the mesg
msgrep = suspect.get_message_rep()
msgrep['X-Changed-Something'] = 'Yes'
suspect.setMessageRep(msgrep)
filename = candidate.archive(suspect)
self.assertTrue(filename != None and filename)
self.tempfiles.append(filename)
archivedmessage = open(filename, 'r').read()
self.assertNotEqual(origmessage.strip(), archivedmessage.strip(
)), 'Archived message should have stored modified message'
def test_modified_message(self):
"""Test if the modified message gets archived correctly"""
from fuglu.shared import Suspect
import shutil
import tempfile
tempfilename = tempfile.mktemp(suffix='archive',
prefix='fuglu-unittest',
dir='/tmp')
shutil.copy(TESTDATADIR + '/helloworld.eml', tempfilename)
self.tempfiles.append(tempfilename)
self.config.set('ArchivePlugin', 'storeoriginal', '0')
candidate = ArchivePlugin(self.config)
suspect = Suspect('*****@*****.**',
'*****@*****.**', tempfilename)
origmessage = suspect.get_source()
# modify the mesg
msgrep = suspect.get_message_rep()
msgrep['X-Changed-Something'] = 'Yes'
suspect.setMessageRep(msgrep)
filename = candidate.archive(suspect)
self.assertTrue(filename != None and filename)
self.tempfiles.append(filename)
archivedmessage = open(filename, 'r').read()
self.assertNotEqual(origmessage.strip(), archivedmessage.strip(
)), 'Archived message should have stored modified message'
message = ""
if type(ans) is tuple:
result, message = ans
else:
result = ans
if result == None:
result = DUNNO
logging.info("Result: %s %s", actioncode_to_string(result), message)
suspect.tags['decisions'].append((pluginstance.section, result))
logging.info(suspect)
for pluginstance in mc.appenders:
logging.info("*** Running appender: %s ***" % pluginstance)
pluginstance.process(suspect, DUNNO)
message = ""
logging.info(suspect)
if suspect.is_modified():
outfilename = '/tmp/fuglu_dummy_message_out.eml'
out = open(outfilename, 'wb')
out.write(suspect.get_source())
out.close()
logging.info(
"Plugin modified the source -> modified message available as %s" % outfilename)
if opts.console:
run_debugconsole(
suspect=suspect, plugin=pluginstance, result=result, config=config, prependers=mc.prependers, scanners=scannerlist, appenders=mc.appenders)
if result == None:
result = DUNNO
logging.info("Result: %s %s", actioncode_to_string(result), message)
suspect.tags['decisions'].append((pluginstance.section, result))
logging.info(suspect)
for pluginstance in mc.appenders:
logging.info("*** Running appender: %s ***" % pluginstance)
pluginstance.process(suspect, DUNNO)
message = ""
logging.info(suspect)
if suspect.is_modified():
outfilename = '/tmp/fuglu_dummy_message_out.eml'
out = open(outfilename, 'wb')
out.write(suspect.get_source())
out.close()
logging.info(
"Plugin modified the source -> modified message available as %s" %
outfilename)
if opts.console:
run_debugconsole(suspect=suspect,
plugin=pluginstance,
result=result,
config=config,
prependers=mc.prependers,
scanners=scannerlist,
appenders=mc.appenders)
