def test_upload_ftp_forbidden_extension(self, mock_shutil, mock_hash, mock_magic, mock_rename, mock_remove):
filehash = 'hash_code'
mock_hash.return_value = filehash
mock_magic.return_value = 'application/x-msdos-program'
from mod_upload.controllers import upload_ftp
upload_ftp(g.db, '/home/1/sample1.js')
queued_sample = QueuedSample.query.filter(
QueuedSample.sha == filehash).first()
self.assertEqual(queued_sample, None)
def test_upload_ftp_forbidden_extension(self, mock_shutil, mock_hash, mock_magic, mock_rename, mock_remove):
filehash = 'hash_code'
mock_hash.return_value = filehash
mock_magic.return_value = 'application/x-msdos-program'
from mod_upload.controllers import upload_ftp
upload_ftp(g.db, '/home/1/sample1.js')
queued_sample = QueuedSample.query.filter(
QueuedSample.sha == filehash).first()
self.assertEqual(queued_sample, None)
def test_upload_ftp(self, mock_shutil, mock_hash, mock_magic, mock_rename, mock_remove):
filehash = 'hash_code'
mock_hash.return_value = filehash
mock_magic.return_value = 'video/mp4'
from mod_upload.controllers import upload_ftp
upload_ftp(g.db, '/home/1/sample1.ts')
queued_sample = QueuedSample.query.filter(
QueuedSample.sha == filehash).first()
self.assertEqual(queued_sample.filename, 'hash_code.ts')
self.assertEqual(queued_sample.extension, '.ts')
mock_remove.assert_called_with('/home/1/sample1.ts')
def test_upload_ftp(self, mock_shutil, mock_hash, mock_magic, mock_rename, mock_remove):
filehash = 'hash_code'
mock_hash.return_value = filehash
mock_magic.return_value = 'video/mp4'
from mod_upload.controllers import upload_ftp
upload_ftp(g.db, '/home/1/sample1.ts')
queued_sample = QueuedSample.query.filter(
QueuedSample.sha == filehash).first()
self.assertEqual(queued_sample.filename, 'hash_code.ts')
self.assertEqual(queued_sample.extension, '.ts')
mock_remove.assert_called_with('/home/1/sample1.ts')
def test_upload_ftp_forbidden_mimetype(self, mock_shutil, mock_hash,
mock_magic, mock_rename,
mock_remove):
"""Test sample upload via ftp with forbidden mimetype."""
filehash = 'hash_code'
mock_hash.return_value = filehash
mock_magic.return_value = 'application/javascript'
from mod_upload.controllers import upload_ftp
upload_ftp(g.db, '/home/1/sample1.js')
queued_sample = QueuedSample.query.filter(
QueuedSample.sha == filehash).first()
self.assertEqual(queued_sample, None)
def process(database, file_to_process):
"""
Call ftp upload method.
:param database: database
:type database: database cursor
:param file_to_process: path of file to upload
:type file_to_process: str
"""
from mod_upload.controllers import upload_ftp
from run import log
log.debug("Calling the FTP upload method from the controller!")
upload_ftp(database, file_to_process)
def process(database, file_to_process):
from mod_upload.controllers import upload_ftp
from run import log
log.debug("Calling the FTP upload method from the controller!")
upload_ftp(database, file_to_process)
