def setUp(self):
config = RawConfigParser()
config.add_section('main')
config.add_section('virus')
config.set('main', 'prependaddedheaders', 'X-Fuglu-')
config.set('virus', 'defaultvirusaction', 'DELETE')
config.add_section('ICAPPlugin')
config.set('ICAPPlugin', 'host', '127.0.0.1')
config.set('ICAPPlugin', 'port', '1344')
config.set('ICAPPlugin', 'timeout', '5')
config.set('ICAPPlugin', 'retries', '3')
config.set('ICAPPlugin', 'maxsize', '22000000')
config.set('ICAPPlugin', 'virusaction', 'DEFAULTVIRUSACTION')
config.set('ICAPPlugin', 'problemaction', 'DEFER')
config.set('ICAPPlugin', 'rejectmessage', '')
config.set('ICAPPlugin', 'service', 'AVSCAN')
config.set('ICAPPlugin', 'enginename', 'sophos')
self.candidate = ICAPPlugin(config)
def setUp(self):
from ConfigParser import RawConfigParser
config = RawConfigParser()
config.add_section('main')
config.add_section('virus')
config.set('main', 'prependaddedheaders', 'X-Fuglu-')
config.set('virus', 'defaultvirusaction', 'DELETE')
config.add_section('ICAPPlugin')
config.set('ICAPPlugin', 'host', '127.0.0.1')
config.set('ICAPPlugin', 'port', '1344')
config.set('ICAPPlugin', 'timeout', '5')
config.set('ICAPPlugin', 'retries', '3')
config.set('ICAPPlugin', 'maxsize', '22000000')
config.set('ICAPPlugin', 'virusaction', 'DEFAULTVIRUSACTION')
config.set('ICAPPlugin', 'problemaction', 'DEFER')
config.set('ICAPPlugin', 'rejectmessage', '')
config.set('ICAPPlugin', 'service', 'AVSCAN')
config.set('ICAPPlugin', 'enginename', 'sophos')
self.candidate = ICAPPlugin(config)
class ICAPPluginTestCase(unittest.TestCase):
def setUp(self):
try:
from configparser import RawConfigParser
except ImportError:
from ConfigParser import RawConfigParser
config = RawConfigParser()
config.add_section('main')
config.add_section('virus')
config.set('main', 'prependaddedheaders', 'X-Fuglu-')
config.set('virus', 'defaultvirusaction', 'DELETE')
config.add_section('ICAPPlugin')
config.set('ICAPPlugin', 'host', '127.0.0.1')
config.set('ICAPPlugin', 'port', '1344')
config.set('ICAPPlugin', 'timeout', '5')
config.set('ICAPPlugin', 'retries', '3')
config.set('ICAPPlugin', 'maxsize', '22000000')
config.set('ICAPPlugin', 'virusaction', 'DEFAULTVIRUSACTION')
config.set('ICAPPlugin', 'problemaction', 'DEFER')
config.set('ICAPPlugin', 'rejectmessage', '')
config.set('ICAPPlugin', 'service', 'AVSCAN')
config.set('ICAPPlugin', 'enginename', 'sophos')
self.candidate = ICAPPlugin(config)
def test_result(self):
"""Test if EICAR virus is detected and message deleted"""
suspect = Suspect(
'*****@*****.**', '*****@*****.**', '/dev/null')
stream = """Date: Mon, 08 Sep 2008 17:33:54 +0200
To: [email protected]
From: [email protected]
Subject: test eicar attachment
X-Mailer: swaks v20061116.0 jetmore.org/john/code/#swaks
MIME-Version: 1.0
Content-Type: multipart/mixed; boundary="----=_MIME_BOUNDARY_000_12140"
------=_MIME_BOUNDARY_000_12140
Content-Type: text/plain
Eicar test
------=_MIME_BOUNDARY_000_12140
Content-Type: application/octet-stream
Content-Transfer-Encoding: BASE64
Content-Disposition: attachment
UEsDBAoAAAAAAGQ7WyUjS4psRgAAAEYAAAAJAAAAZWljYXIuY29tWDVPIVAlQEFQWzRcUFpYNTQo
UF4pN0NDKTd9JEVJQ0FSLVNUQU5EQVJELUFOVElWSVJVUy1URVNULUZJTEUhJEgrSCoNClBLAQIU
AAoAAAAAAGQ7WyUjS4psRgAAAEYAAAAJAAAAAAAAAAEAIAD/gQAAAABlaWNhci5jb21QSwUGAAAA
AAEAAQA3AAAAbQAAAAAA
------=_MIME_BOUNDARY_000_12140--"""
suspect.setMessageRep(email.message_from_string(stream))
result = self.candidate.examine(suspect)
if type(result) is tuple:
result, message = result
strresult = actioncode_to_string(result)
if strresult == "DEFER":
import logging
logging.warn(
"ICAP Scan returned DEFER -> daemon is probably not running. treating this as test ok anyway")
return
self.assertEqual(strresult, "DELETE")
class ICAPPluginTestCase(unittest.TestCase):
def setUp(self):
config = RawConfigParser()
config.add_section('main')
config.add_section('virus')
config.set('main', 'prependaddedheaders', 'X-Fuglu-')
config.set('virus', 'defaultvirusaction', 'DELETE')
config.add_section('ICAPPlugin')
config.set('ICAPPlugin', 'host', '127.0.0.1')
config.set('ICAPPlugin', 'port', '1344')
config.set('ICAPPlugin', 'timeout', '5')
config.set('ICAPPlugin', 'retries', '3')
config.set('ICAPPlugin', 'maxsize', '22000000')
config.set('ICAPPlugin', 'virusaction', 'DEFAULTVIRUSACTION')
config.set('ICAPPlugin', 'problemaction', 'DEFER')
config.set('ICAPPlugin', 'rejectmessage', '')
config.set('ICAPPlugin', 'service', 'AVSCAN')
config.set('ICAPPlugin', 'enginename', 'sophos')
self.candidate = ICAPPlugin(config)
def test_result(self):
"""Test if EICAR virus is detected and message deleted"""
suspect = Suspect('*****@*****.**',
'*****@*****.**', '/dev/null')
stream = """Date: Mon, 08 Sep 2008 17:33:54 +0200
To: [email protected]
From: [email protected]
Subject: test eicar attachment
X-Mailer: swaks v20061116.0 jetmore.org/john/code/#swaks
MIME-Version: 1.0
Content-Type: multipart/mixed; boundary="----=_MIME_BOUNDARY_000_12140"
------=_MIME_BOUNDARY_000_12140
Content-Type: text/plain
Eicar test
------=_MIME_BOUNDARY_000_12140
Content-Type: application/octet-stream
Content-Transfer-Encoding: BASE64
Content-Disposition: attachment
UEsDBAoAAAAAAGQ7WyUjS4psRgAAAEYAAAAJAAAAZWljYXIuY29tWDVPIVAlQEFQWzRcUFpYNTQo
UF4pN0NDKTd9JEVJQ0FSLVNUQU5EQVJELUFOVElWSVJVUy1URVNULUZJTEUhJEgrSCoNClBLAQIU
AAoAAAAAAGQ7WyUjS4psRgAAAEYAAAAJAAAAAAAAAAEAIAD/gQAAAABlaWNhci5jb21QSwUGAAAA
AAEAAQA3AAAAbQAAAAAA
------=_MIME_BOUNDARY_000_12140--"""
suspect.setMessageRep(email.message_from_string(stream))
result = self.candidate.examine(suspect)
if type(result) is tuple:
result, message = result
strresult = actioncode_to_string(result)
if strresult == "DEFER":
import logging
logging.warn(
"ICAP Scan returned DEFER -> daemon is probably not running. treating this as test ok anyway"
)
return
self.assertEqual(strresult, "DELETE")
