class TestScannedFile(FileUploadTestCase):
def setUp(self):
self.file_object = self._file_storage_wrapper(
"This is a test", browser_filename="abc.txt")
self.scanner = ScannedFile(self.file_object)
def tearDown(self):
try:
os.remove('/tmp/abc.txt')
except OSError:
pass
def test_has_virus_signature(self):
self.scanner._virus_signature = True
self.scanner._save_file_to_disk = Mock()
self.scanner._scan_file = Mock()
self.scanner._clean_up = Mock()
assert_that(self.scanner.has_virus_signature, is_(True))
self.scanner._save_file_to_disk.assert_called_once_with()
self.scanner._scan_file.assert_called_once_with()
self.scanner._clean_up.assert_called_once_with()
def test_saving_a_file_to_disk(self):
self.scanner._save_file_to_disk()
assert_that(file('/tmp/abc.txt').read(), is_("This is a test"))
def test_cleaning_up_after_scanning(self):
self.file_object.save('/tmp/abc.txt')
self.scanner._clean_up()
assert_that(os.path.exists('/tmp/abc.txt'), is_(False))
def test_scanning_a_file(self):
self.scanner._clamscan = Mock(return_value=True)
self.scanner._scan_file()
assert_that(self.scanner._virus_signature, is_(True))
self.scanner._clamscan.assert_called_once_with("/tmp/abc.txt")
class TestScannedFile(FileUploadTestCase):
def setUp(self):
self.file_object = self._file_storage_wrapper("This is a test", browser_filename="abc.txt")
self.scanner = ScannedFile(self.file_object)
def tearDown(self):
try:
os.remove('/tmp/abc.txt')
except OSError:
pass
def test_has_virus_signature(self):
self.scanner._virus_signature = True
self.scanner._save_file_to_disk = Mock()
self.scanner._scan_file = Mock()
self.scanner._clean_up = Mock()
assert_that(self.scanner.has_virus_signature, is_(True))
self.scanner._save_file_to_disk.assert_called_once_with()
self.scanner._scan_file.assert_called_once_with()
self.scanner._clean_up.assert_called_once_with()
def test_saving_a_file_to_disk(self):
self.scanner._save_file_to_disk()
assert_that(file('/tmp/abc.txt').read(), is_("This is a test"))
def test_cleaning_up_after_scanning(self):
self.file_object.save('/tmp/abc.txt')
self.scanner._clean_up()
assert_that(os.path.exists('/tmp/abc.txt'), is_(False))
def test_scanning_a_file(self):
self.scanner._clamscan = Mock(return_value=True)
self.scanner._scan_file()
assert_that(self.scanner._virus_signature, is_(True))
self.scanner._clamscan.assert_called_once_with("/tmp/abc.txt")
def setUp(self):
self.file_object = self._file_storage_wrapper(
"This is a test", browser_filename="abc.txt")
self.scanner = ScannedFile(self.file_object)
def setUp(self):
self.file_object = self._file_storage_wrapper("This is a test", browser_filename="abc.txt")
self.scanner = ScannedFile(self.file_object)