Tidy up logging and rework tests from scratch
This commit is contained in:
172
tests/legacy/scenario_1.py
Normal file
172
tests/legacy/scenario_1.py
Normal file
@@ -0,0 +1,172 @@
|
||||
import os, sys, click, glob
|
||||
|
||||
from .scenario import Scenario
|
||||
|
||||
from .helper import createMediaTestFile
|
||||
from ffx.process import executeProcess
|
||||
|
||||
from ffx.file_properties import FileProperties
|
||||
|
||||
from ffx.media_descriptor import MediaDescriptor
|
||||
from ffx.track_descriptor import TrackDescriptor
|
||||
|
||||
from ffx.track_type import TrackType
|
||||
from ffx.track_disposition import TrackDisposition
|
||||
|
||||
from .media_combinator_0 import MediaCombinator0
|
||||
|
||||
from .basename_combinator import BasenameCombinator
|
||||
|
||||
|
||||
class Scenario1(Scenario):
|
||||
"""Creating file VAa, h264/aac/aac
|
||||
Converting to VaA, vp9/opus/opus
|
||||
No tmdb, default parameters"""
|
||||
|
||||
TEST_FILE_EXTENSION = 'mkv'
|
||||
EXPECTED_FILE_EXTENSION = 'webm'
|
||||
|
||||
|
||||
def __init__(self, context):
|
||||
|
||||
context['use_tmdb'] = False
|
||||
context['use_pattern'] = False
|
||||
|
||||
super().__init__(context)
|
||||
|
||||
def getScenario(self):
|
||||
return self.__class__.__name__[8:]
|
||||
|
||||
|
||||
def job(self, yieldObj: dict):
|
||||
|
||||
testContext = self._context.copy()
|
||||
|
||||
identifier = yieldObj['identifier']
|
||||
variantList = yieldObj['variants']
|
||||
|
||||
variantIdentifier = '-'.join(variantList)
|
||||
variantLabel = f"{self.__class__.__name__} Variant {variantIdentifier}"
|
||||
|
||||
mc0 = MediaCombinator0(context = testContext)
|
||||
sourceMediaDescriptor: MediaDescriptor = mc0.getPayload()
|
||||
|
||||
assertSelectorList: list = yieldObj['assertSelectors']
|
||||
assertFuncList = yieldObj['assertFuncs']
|
||||
shouldFail = yieldObj['shouldFail']
|
||||
|
||||
variantPayload = yieldObj['payload']
|
||||
variantBasename = variantPayload['basename']
|
||||
variantFilenameLabel = variantPayload['label']
|
||||
expectedBasename = variantPayload['expectedBasename']
|
||||
|
||||
variantFilename = f"{variantBasename}.{Scenario1.TEST_FILE_EXTENSION}"
|
||||
expectedFilename = f"{expectedBasename}.{Scenario1.EXPECTED_FILE_EXTENSION}"
|
||||
|
||||
|
||||
if self._context['test_variant'] and not variantIdentifier.startswith(self._context['test_variant']):
|
||||
return
|
||||
|
||||
if (self._context['test_limit'] and (self._context['test_passed_counter'] + self._context['test_failed_counter'])
|
||||
>= self._context['test_limit']):
|
||||
return
|
||||
|
||||
self._logger.debug(f"Running Job: {variantLabel}")
|
||||
|
||||
|
||||
# Phase 1: Setup source files
|
||||
|
||||
if not variantBasename:
|
||||
raise ValueError(f"{variantLabel}: Testfile basename is falsy")
|
||||
|
||||
|
||||
self.clearTestDirectory()
|
||||
|
||||
self._logger.debug(f"Creating test file: {variantFilename}")
|
||||
mediaFilePath = createMediaTestFile(mediaDescriptor=sourceMediaDescriptor,
|
||||
baseName=variantBasename,
|
||||
directory=self._testDirectory,
|
||||
logger=self._logger,
|
||||
length = 2)
|
||||
|
||||
|
||||
# Phase 2: Run ffx
|
||||
|
||||
commandSequence = [sys.executable, '-m', self._ffxModuleName]
|
||||
|
||||
if self._context['verbosity']:
|
||||
commandSequence += ['--verbose',
|
||||
str(self._context['verbosity'])]
|
||||
|
||||
commandSequence += ['convert',
|
||||
mediaFilePath,
|
||||
'--no-prompt',
|
||||
'--no-signature']
|
||||
|
||||
if variantFilenameLabel:
|
||||
commandSequence += ['--label', variantFilenameLabel]
|
||||
|
||||
|
||||
commandSequence += ['--no-pattern']
|
||||
commandSequence += ['--no-tmdb']
|
||||
|
||||
out, err, rc = executeProcess(commandSequence, directory = self._testDirectory, context = self._context)
|
||||
|
||||
if out and self._context['verbosity'] >= 9:
|
||||
self._logger.debug(f"{variantLabel}: Process output: {out}")
|
||||
if rc:
|
||||
self._logger.debug(f"{variantLabel}: Process returned ERROR {rc} ({err})")
|
||||
|
||||
|
||||
# Phase 3: Evaluate results
|
||||
|
||||
resultFilenames = [rf for rf in self.getFilenamesInTestDirectory() if rf != 'ffmpeg2pass-0.log' and rf != variantFilename]
|
||||
self._logger.debug(f"{variantLabel}: Result filenames: {resultFilenames}")
|
||||
|
||||
|
||||
try:
|
||||
|
||||
jobFailed = bool(rc)
|
||||
|
||||
self._logger.debug(f"{variantLabel}: Should fail: {shouldFail} / actually failed: {jobFailed}")
|
||||
|
||||
assert (jobFailed == shouldFail
|
||||
), f"Process {'failed' if jobFailed else 'did not fail'}"
|
||||
|
||||
if not jobFailed:
|
||||
|
||||
expectedResultFilePath = os.path.join(self._testDirectory, f"{expectedFilename}")
|
||||
|
||||
assert (os.path.isfile(expectedResultFilePath)
|
||||
), f"Result file {expectedFilename} in path '{self._testDirectory}' wasn't created"
|
||||
|
||||
for assertIndex in range(len(assertSelectorList)):
|
||||
|
||||
assertSelector = assertSelectorList[assertIndex]
|
||||
assertFunc = assertFuncList[assertIndex]
|
||||
assertVariant = variantList[assertIndex]
|
||||
|
||||
if assertSelector == 'B':
|
||||
#TODO: per file find
|
||||
testObj = {'filenames': resultFilenames}
|
||||
assertFunc(testObj=testObj)
|
||||
if assertSelector == 'L':
|
||||
assertFunc()
|
||||
if assertSelector == 'I':
|
||||
assertFunc()
|
||||
|
||||
self._context['test_passed_counter'] += 1
|
||||
self._reportLogger.info(f"{variantLabel}: Test passed")
|
||||
|
||||
except AssertionError as ae:
|
||||
|
||||
self._context['test_failed_counter'] += 1
|
||||
self._reportLogger.error(f"{variantLabel}: Test FAILED ({ae})")
|
||||
|
||||
|
||||
def run(self):
|
||||
for BC in BasenameCombinator.getAllClassReferences():
|
||||
self._logger.debug(f"BC={BC.__name__}")
|
||||
bc = BC(context = self._context)
|
||||
for y in bc.getYield():
|
||||
self.job(y)
|
||||
Reference in New Issue
Block a user