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_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)