70 lines
3.2 KiB
Python
70 lines
3.2 KiB
Python
import importlib.util
|
|
import os
|
|
from pathlib import Path
|
|
import shutil
|
|
import subprocess
|
|
import tempfile
|
|
import unittest
|
|
from unittest.mock import patch
|
|
|
|
spec = importlib.util.spec_from_file_location('file_preview', Path(__file__).resolve().parents[1] / 'scripts/file-preview.py')
|
|
module = importlib.util.module_from_spec(spec)
|
|
spec.loader.exec_module(module)
|
|
|
|
|
|
class FilePreviewTests(unittest.TestCase):
|
|
def test_text_is_bounded_and_path_preserved(self):
|
|
with tempfile.TemporaryDirectory() as directory:
|
|
path = Path(directory) / 'a " quoted.txt'
|
|
path.write_text('a' * 10000)
|
|
data = module.preview(str(path))
|
|
self.assertEqual(data['bytes'], 10000)
|
|
self.assertEqual(len(data['text']), 8192)
|
|
self.assertTrue(data['truncated'])
|
|
self.assertEqual(data['path'], str(path))
|
|
|
|
def test_missing_file_and_folder(self):
|
|
with tempfile.TemporaryDirectory() as directory:
|
|
self.assertEqual(module.preview(directory)['kind'], 'Folder')
|
|
self.assertIn('error', module.preview(directory + '/missing'))
|
|
|
|
def test_special_file_is_not_read(self):
|
|
with tempfile.TemporaryDirectory() as directory:
|
|
path = directory + '/fifo.txt'
|
|
os.mkfifo(path)
|
|
self.assertEqual(module.preview(path)['kind'], 'Special file')
|
|
|
|
def test_missing_media_tools_preserve_basic_metadata(self):
|
|
with tempfile.NamedTemporaryFile(suffix='.mp4') as source:
|
|
with patch.object(module, 'read_bounded', return_value=None):
|
|
data = module.preview(source.name)
|
|
self.assertEqual(data['mime'], 'video/mp4')
|
|
self.assertIn('bytes', data)
|
|
self.assertNotIn('thumbnail', data)
|
|
|
|
@unittest.skipUnless(shutil.which('ffmpeg') and shutil.which('ffprobe'), 'ffmpeg optional')
|
|
def test_video_resolution_duration_and_thumbnail(self):
|
|
with tempfile.TemporaryDirectory() as directory:
|
|
path = directory + '/clip.mkv'
|
|
subprocess.run(['ffmpeg', '-v', 'error', '-f', 'lavfi', '-i', 'color=c=red:s=160x90:d=1', '-c:v', 'ffv1', path], check=True, timeout=10)
|
|
data = module.preview(path)
|
|
self.assertEqual((data['width'], data['height']), (160, 90))
|
|
self.assertAlmostEqual(data['duration'], 1, places=1)
|
|
self.assertTrue(data['thumbnail'].startswith('data:image/jpeg;base64,'))
|
|
|
|
@unittest.skipUnless(shutil.which('ffmpeg') and shutil.which('ffprobe'), 'ffmpeg optional')
|
|
def test_audio_details(self):
|
|
with tempfile.TemporaryDirectory() as directory:
|
|
path = directory + '/audio.wav'
|
|
subprocess.run(['ffmpeg', '-v', 'error', '-f', 'lavfi', '-i', 'sine=duration=1', path], check=True, timeout=10)
|
|
data = module.preview(path)
|
|
self.assertAlmostEqual(data['duration'], 1, places=1)
|
|
self.assertIn('44100 Hz', data['audio'])
|
|
self.assertNotIn('thumbnail', data)
|
|
|
|
def test_generic_binary_not_shown_as_text(self):
|
|
with tempfile.NamedTemporaryFile(suffix='.txt') as source:
|
|
source.write(b'abc\x00def')
|
|
source.flush()
|
|
self.assertNotIn('text', module.preview(source.name))
|