oak-team/oak-editor
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
Files
b4db6ebc8828adaaa62b1a0aa5847236f061e2d1
oak-editor/tests/gtest/proxy_manager_test.cpp
T
Mike-Solar 6aaf37e2e5 proxy: per-footage presets, audio in proxies, configurable ffmpeg path 
- Footage can store custom proxy parameters (width/height/crf/preset/
  extension/audio) that override the global settings; they are
  serialized with the project and used by every generation entry point
  via Footage::GetEffectiveProxyParams()
- Proxies now include the source audio streams (AAC) unless disabled;
  the proxy filename records the audio flag and offline audio rendering
  decodes from the proxy when present
- ProxyTask resolves ffmpeg from the new FFmpegPath config key first,
  then PATH, then common install locations (e.g. Homebrew on macOS),
  instead of relying on PATH only; the error message points at the
  preferences when no executable is found
- ProxyTask::BuildArguments() is extracted for testability
- Add ProxyIncludeAudio and FFmpegPath config defaults
- Add regression tests for the filename audio marker, config-backed
  params, ffmpeg resolution, argument building, and custom-param
  persistence
2026-07-16 22:31:37 +08:00

477 lines
16 KiB
C++
Raw Blame History

#include <gtest/gtest.h>
#include <QCoreApplication>
#include <QDir>
#include <QFile>
#include <QTemporaryDir>
#include <QXmlStreamReader>
#include <QXmlStreamWriter>
#include "codec/proxymanager.h"
#include "config/config.h"
#include "node/project/footage/footage.h"
#include "render/job/footagejob.h"
#include "task/proxy/proxy.h"
namespace
{
QVariant ProxyConfigValue(const char *key)
{
return olive::Config::Current()[QString::fromUtf8(key)];
}
} // namespace
TEST(ProxyManager, BuildsStableProxyFilename)
{
olive::ProxyManager::ProxyParams params;
params.width = 1280;
params.height = 720;
params.version = 1;
const QString first = olive::ProxyManager::GetProxyFilename(
QStringLiteral("/tmp/oak-cache"), QStringLiteral("/media/source.mov"),
0, params);
const QString second = olive::ProxyManager::GetProxyFilename(
QStringLiteral("/tmp/oak-cache"), QStringLiteral("/media/source.mov"),
0, params);
const QString other_stream = olive::ProxyManager::GetProxyFilename(
QStringLiteral("/tmp/oak-cache"), QStringLiteral("/media/source.mov"),
1, params);
EXPECT_EQ(first, second);
EXPECT_NE(first, other_stream);
EXPECT_TRUE(first.contains(QStringLiteral("/proxy/")));
EXPECT_TRUE(first.endsWith(QStringLiteral(".mp4")));
}
TEST(ProxyManager, ProxyFilenameIncludesPresetParameters)
{
olive::ProxyManager::ProxyParams mp4_720p;
mp4_720p.width = 1280;
mp4_720p.height = 720;
mp4_720p.version = 1;
mp4_720p.extension = QStringLiteral("mp4");
olive::ProxyManager::ProxyParams mov_540p = mp4_720p;
mov_540p.width = 960;
mov_540p.height = 540;
mov_540p.version = 2;
mov_540p.extension = QStringLiteral("mov");
const QString first = olive::ProxyManager::GetProxyFilename(
QStringLiteral("/tmp/oak-cache"), QStringLiteral("/media/source.mov"),
0, mp4_720p);
const QString second = olive::ProxyManager::GetProxyFilename(
QStringLiteral("/tmp/oak-cache"), QStringLiteral("/media/source.mov"),
0, mov_540p);
EXPECT_NE(first, second);
EXPECT_TRUE(first.contains(QStringLiteral(".1280x720.v1.")));
EXPECT_TRUE(second.contains(QStringLiteral(".960x540.v2.")));
EXPECT_TRUE(second.endsWith(QStringLiteral(".mov")));
}
TEST(ProxyManager, DetectsProxyState)
{
QTemporaryDir dir;
ASSERT_TRUE(dir.isValid());
const QString proxy =
QDir(dir.path()).filePath(QStringLiteral("proxy-file.mp4"));
EXPECT_EQ(olive::ProxyManager::GetProxyState(proxy),
olive::ProxyManager::kProxyMissing);
QFile working(olive::ProxyManager::GetWorkingProxyFilename(proxy));
ASSERT_TRUE(working.open(QFile::WriteOnly));
working.close();
EXPECT_EQ(olive::ProxyManager::GetProxyState(proxy),
olive::ProxyManager::kProxyGenerating);
QFile ready(proxy);
ASSERT_TRUE(ready.open(QFile::WriteOnly));
ready.close();
EXPECT_EQ(olive::ProxyManager::GetProxyState(proxy),
olive::ProxyManager::kProxyReady);
}
TEST(ProxyManager, ReadyStateTakesPrecedenceOverWorkingFile)
{
QTemporaryDir dir;
ASSERT_TRUE(dir.isValid());
const QString proxy =
QDir(dir.path()).filePath(QStringLiteral("proxy-file.mp4"));
QFile ready(proxy);
ASSERT_TRUE(ready.open(QFile::WriteOnly));
ready.close();
QFile working(olive::ProxyManager::GetWorkingProxyFilename(proxy));
ASSERT_TRUE(working.open(QFile::WriteOnly));
working.close();
EXPECT_EQ(olive::ProxyManager::GetProxyState(proxy),
olive::ProxyManager::kProxyReady);
}
TEST(ProxyManager, ConvertsProxyStateToAndFromStrings)
{
EXPECT_EQ(olive::ProxyManager::ProxyStateToString(
olive::ProxyManager::kProxyMissing),
QStringLiteral("missing"));
EXPECT_EQ(olive::ProxyManager::ProxyStateToString(
olive::ProxyManager::kProxyGenerating),
QStringLiteral("generating"));
EXPECT_EQ(olive::ProxyManager::ProxyStateToString(
olive::ProxyManager::kProxyReady),
QStringLiteral("ready"));
EXPECT_EQ(olive::ProxyManager::ProxyStateToString(
olive::ProxyManager::kProxyFailed),
QStringLiteral("failed"));
EXPECT_EQ(
olive::ProxyManager::ProxyStateFromString(QStringLiteral("missing")),
olive::ProxyManager::kProxyMissing);
EXPECT_EQ(
olive::ProxyManager::ProxyStateFromString(QStringLiteral("generating")),
olive::ProxyManager::kProxyGenerating);
EXPECT_EQ(
olive::ProxyManager::ProxyStateFromString(QStringLiteral("ready")),
olive::ProxyManager::kProxyReady);
EXPECT_EQ(
olive::ProxyManager::ProxyStateFromString(QStringLiteral("failed")),
olive::ProxyManager::kProxyFailed);
EXPECT_EQ(
olive::ProxyManager::ProxyStateFromString(QStringLiteral("unknown")),
olive::ProxyManager::kProxyMissing);
}
TEST(ProxyManager, FootagePersistsProxyMetadata)
{
QString xml;
QXmlStreamWriter writer(&xml);
writer.writeStartDocument();
writer.writeStartElement(QStringLiteral("custom"));
writer.writeTextElement(QStringLiteral("timestamp"), QStringLiteral("0"));
writer.writeStartElement(QStringLiteral("proxy"));
writer.writeAttribute(QStringLiteral("enabled"), QStringLiteral("1"));
writer.writeAttribute(QStringLiteral("state"), QStringLiteral("ready"));
writer.writeAttribute(QStringLiteral("stream"), QStringLiteral("0"));
writer.writeAttribute(QStringLiteral("preset"), QStringLiteral("1"));
writer.writeCharacters(QStringLiteral("/cache/proxy/example.mp4"));
writer.writeEndElement();
writer.writeEndElement();
writer.writeEndDocument();
QXmlStreamReader reader(xml);
ASSERT_TRUE(reader.readNextStartElement());
ASSERT_EQ(reader.name(), QStringLiteral("custom"));
olive::Footage footage;
ASSERT_TRUE(footage.LoadCustom(&reader, nullptr));
EXPECT_TRUE(footage.proxy_enabled());
EXPECT_EQ(footage.proxy_path(), QStringLiteral("/cache/proxy/example.mp4"));
EXPECT_EQ(footage.proxy_state(), olive::ProxyManager::kProxyReady);
EXPECT_EQ(footage.proxy_video_stream_index(), 0);
EXPECT_EQ(footage.proxy_preset_version(), 1);
}
TEST(ProxyManager, FootageSavesProxyMetadata)
{
olive::Footage footage;
footage.set_timestamp(42);
footage.SetProxy(QStringLiteral("/cache/proxy/example.mp4"),
olive::ProxyManager::kProxyReady, 2, 3, true);
QString xml;
QXmlStreamWriter writer(&xml);
writer.writeStartDocument();
writer.writeStartElement(QStringLiteral("custom"));
footage.SaveCustom(&writer);
writer.writeEndElement();
writer.writeEndDocument();
EXPECT_TRUE(xml.contains(QStringLiteral("<proxy")));
EXPECT_TRUE(xml.contains(QStringLiteral("enabled=\"1\"")));
EXPECT_TRUE(xml.contains(QStringLiteral("state=\"ready\"")));
EXPECT_TRUE(xml.contains(QStringLiteral("stream=\"2\"")));
EXPECT_TRUE(xml.contains(QStringLiteral("preset=\"3\"")));
EXPECT_TRUE(xml.contains(QStringLiteral("/cache/proxy/example.mp4")));
}
TEST(ProxyManager, FootageClearRemovesProxyMetadata)
{
olive::Footage footage;
footage.SetProxy(QStringLiteral("/cache/proxy/example.mp4"),
olive::ProxyManager::kProxyReady, 0, 1, true);
footage.Clear();
EXPECT_FALSE(footage.proxy_enabled());
EXPECT_TRUE(footage.proxy_path().isEmpty());
EXPECT_EQ(footage.proxy_state(), olive::ProxyManager::kProxyMissing);
EXPECT_EQ(footage.proxy_video_stream_index(), -1);
EXPECT_EQ(footage.proxy_preset_version(), 0);
}
TEST(ProxyManager, EmitsProxyFinishedState)
{
olive::ProxyManager::CreateInstance();
bool received = false;
QString received_source;
int received_stream = -1;
QString received_proxy;
olive::ProxyManager::ProxyState received_state =
olive::ProxyManager::kProxyMissing;
QObject::connect(
olive::ProxyManager::instance(), &olive::ProxyManager::ProxyFinished,
[&received, &received_source, &received_stream, &received_proxy,
&received_state](const QString &source_filename, int stream_index,
const QString &proxy_filename,
olive::ProxyManager::ProxyState state) {
received = true;
received_source = source_filename;
received_stream = stream_index;
received_proxy = proxy_filename;
received_state = state;
});
emit olive::ProxyManager::instance()
-> ProxyFinished(QStringLiteral("/media/source.mov"), 0,
QStringLiteral("/cache/proxy/example.mp4"),
olive::ProxyManager::kProxyFailed);
EXPECT_TRUE(received);
EXPECT_EQ(received_source, QStringLiteral("/media/source.mov"));
EXPECT_EQ(received_stream, 0);
EXPECT_EQ(received_proxy, QStringLiteral("/cache/proxy/example.mp4"));
EXPECT_EQ(received_state, olive::ProxyManager::kProxyFailed);
olive::ProxyManager::DestroyInstance();
}
TEST(ProxyManager, WorkingProxyFilenamePrependsExtension)
{
const QString proxy = QStringLiteral("/cache/proxy/example.mp4");
const QString working = olive::ProxyManager::GetWorkingProxyFilename(proxy);
EXPECT_EQ(working, QStringLiteral("/cache/proxy/example.mp4.working.mp4"));
}
TEST(ProxyManager, ProxyStateFromStringDefaultsForEmpty)
{
EXPECT_EQ(olive::ProxyManager::ProxyStateFromString(QString()),
olive::ProxyManager::kProxyMissing);
}
TEST(ProxyManager, FootageProxyCanBeDisabled)
{
olive::Footage footage;
footage.SetProxy(QStringLiteral("/cache/proxy/example.mp4"),
olive::ProxyManager::kProxyReady, 0, 1, false);
EXPECT_FALSE(footage.proxy_enabled());
EXPECT_FALSE(footage.proxy_path().isEmpty());
}
TEST(ProxyManager, FootageJobWithoutProxyHasEmptyProxyFields)
{
olive::FootageJob job(olive::TimeRange(), QStringLiteral("source-decoder"),
QStringLiteral("/media/source.mov"),
olive::Track::kVideo, olive::rational(10),
olive::LoopMode::kLoopModeOff);
EXPECT_FALSE(job.has_proxy());
EXPECT_TRUE(job.proxy_filename().isEmpty());
EXPECT_TRUE(job.proxy_decoder().isEmpty());
EXPECT_EQ(job.proxy_stream_index(), -1);
}
TEST(ProxyManager, ProxyFilenameIncludesAudioFlag)
{
olive::ProxyManager::ProxyParams params;
params.include_audio = true;
const QString with_audio = olive::ProxyManager::GetProxyFilename(
QStringLiteral("/tmp/oak-cache"), QStringLiteral("/media/source.mov"),
0, params);
EXPECT_TRUE(with_audio.contains(QStringLiteral(".a1.")));
EXPECT_TRUE(olive::ProxyManager::ProxyFilenameHasAudio(with_audio));
params.include_audio = false;
const QString without_audio = olive::ProxyManager::GetProxyFilename(
QStringLiteral("/tmp/oak-cache"), QStringLiteral("/media/source.mov"),
0, params);
EXPECT_TRUE(without_audio.contains(QStringLiteral(".a0.")));
EXPECT_FALSE(olive::ProxyManager::ProxyFilenameHasAudio(without_audio));
// Legacy proxy filenames (no audio marker) must be treated as video-only
EXPECT_FALSE(olive::ProxyManager::ProxyFilenameHasAudio(
QStringLiteral("/tmp/oak-cache/proxy/abc-0.1280x720.v1.mp4")));
// The audio flag distinguishes otherwise identical proxy filenames
EXPECT_NE(with_audio, without_audio);
}
TEST(ProxyManager, ProxyParamsFromConfigReadsDefaults)
{
const olive::ProxyManager::ProxyParams params =
olive::ProxyManager::ProxyParamsFromConfig();
EXPECT_EQ(params.width, ProxyConfigValue("ProxyWidth").value<int>());
EXPECT_EQ(params.height, ProxyConfigValue("ProxyHeight").value<int>());
EXPECT_EQ(params.crf, ProxyConfigValue("ProxyCRF").value<int>());
EXPECT_EQ(params.preset, ProxyConfigValue("ProxyPreset").toString());
EXPECT_EQ(params.include_audio,
ProxyConfigValue("ProxyIncludeAudio").toBool());
}
TEST(ProxyManager, FindFFmpegExecutablePrefersConfiguredPath)
{
// The test executable itself is guaranteed to be an existing executable
// file, making it a safe stand-in for an ffmpeg binary
const QString self = QCoreApplication::applicationFilePath();
ASSERT_FALSE(self.isEmpty());
EXPECT_EQ(olive::ProxyManager::FindFFmpegExecutable(self), self);
}
TEST(ProxyManager, FindFFmpegExecutableRejectsInvalidConfiguredPath)
{
const QString bogus = QStringLiteral("/nonexistent/ffmpeg-binary");
const QString result = olive::ProxyManager::FindFFmpegExecutable(bogus);
// Must not return the invalid configured path; any fallback is acceptable
EXPECT_NE(result, bogus);
}
TEST(ProxyTask, BuildArgumentsIncludesAudioWhenEnabled)
{
olive::ProxyManager::ProxyParams params;
params.include_audio = true;
const QStringList args = olive::ProxyTask::BuildArguments(
QStringLiteral("/media/source.mov"), 1, params,
QStringLiteral("/cache/proxy/out.mp4"));
EXPECT_FALSE(args.contains(QStringLiteral("-an")));
const int audio_map = args.indexOf(QStringLiteral("0:a?"));
EXPECT_GE(audio_map, 0);
EXPECT_GT(audio_map, args.indexOf(QStringLiteral("-map")));
EXPECT_TRUE(args.contains(QStringLiteral("-c:a")));
EXPECT_TRUE(args.contains(QStringLiteral("aac")));
// The requested video stream must be mapped before the audio streams
EXPECT_LT(args.indexOf(QStringLiteral("0:1")), audio_map);
}
TEST(ProxyTask, BuildArgumentsDisablesAudioWhenDisabled)
{
olive::ProxyManager::ProxyParams params;
params.include_audio = false;
const QStringList args = olive::ProxyTask::BuildArguments(
QStringLiteral("/media/source.mov"), 1, params,
QStringLiteral("/cache/proxy/out.mp4"));
EXPECT_TRUE(args.contains(QStringLiteral("-an")));
EXPECT_FALSE(args.contains(QStringLiteral("0:a?")));
}
TEST(ProxyManager, FootagePersistsCustomProxyParams)
{
olive::Footage footage;
olive::ProxyManager::ProxyParams params;
params.width = 640;
params.height = 360;
params.crf = 30;
params.preset = QStringLiteral("faster");
params.extension = QStringLiteral("mov");
params.include_audio = false;
footage.SetCustomProxyParams(params);
footage.SetProxy(QStringLiteral("/cache/proxy/example.mp4"),
olive::ProxyManager::kProxyReady, 0, 1, true);
QString xml;
QXmlStreamWriter writer(&xml);
writer.writeStartDocument();
writer.writeStartElement(QStringLiteral("custom"));
footage.SaveCustom(&writer);
writer.writeEndElement();
writer.writeEndDocument();
EXPECT_TRUE(xml.contains(QStringLiteral("custom=\"1\"")));
EXPECT_TRUE(xml.contains(QStringLiteral("pwidth=\"640\"")));
EXPECT_TRUE(xml.contains(QStringLiteral("pheight=\"360\"")));
EXPECT_TRUE(xml.contains(QStringLiteral("pcrf=\"30\"")));
EXPECT_TRUE(xml.contains(QStringLiteral("ppreset=\"faster\"")));
EXPECT_TRUE(xml.contains(QStringLiteral("pext=\"mov\"")));
EXPECT_TRUE(xml.contains(QStringLiteral("paudio=\"0\"")));
QXmlStreamReader reader(xml);
ASSERT_TRUE(reader.readNextStartElement());
ASSERT_EQ(reader.name(), QStringLiteral("custom"));
olive::Footage loaded;
ASSERT_TRUE(loaded.LoadCustom(&reader, nullptr));
ASSERT_TRUE(loaded.has_custom_proxy_params());
EXPECT_EQ(loaded.custom_proxy_params().width, 640);
EXPECT_EQ(loaded.custom_proxy_params().height, 360);
EXPECT_EQ(loaded.custom_proxy_params().crf, 30);
EXPECT_EQ(loaded.custom_proxy_params().preset, QStringLiteral("faster"));
EXPECT_EQ(loaded.custom_proxy_params().extension, QStringLiteral("mov"));
EXPECT_FALSE(loaded.custom_proxy_params().include_audio);
}
TEST(ProxyManager, FootageWithoutCustomParamsOmitsThemFromXml)
{
olive::Footage footage;
footage.SetProxy(QStringLiteral("/cache/proxy/example.mp4"),
olive::ProxyManager::kProxyReady, 0, 1, true);
QString xml;
QXmlStreamWriter writer(&xml);
writer.writeStartDocument();
writer.writeStartElement(QStringLiteral("custom"));
footage.SaveCustom(&writer);
writer.writeEndElement();
writer.writeEndDocument();
EXPECT_FALSE(xml.contains(QStringLiteral("custom=")));
// Loading old project files without custom params must not enable them
QXmlStreamReader reader(xml);
ASSERT_TRUE(reader.readNextStartElement());
olive::Footage loaded;
ASSERT_TRUE(loaded.LoadCustom(&reader, nullptr));
EXPECT_FALSE(loaded.has_custom_proxy_params());
}
TEST(ProxyManager, FootageEffectiveProxyParams)
{
olive::Footage footage;
// Without custom params, the global config values apply
const olive::ProxyManager::ProxyParams global_params =
footage.GetEffectiveProxyParams();
EXPECT_EQ(global_params.width, ProxyConfigValue("ProxyWidth").value<int>());
EXPECT_EQ(global_params.include_audio,
ProxyConfigValue("ProxyIncludeAudio").toBool());
// Custom params take precedence
olive::ProxyManager::ProxyParams custom;
custom.width = 320;
custom.height = 180;
footage.SetCustomProxyParams(custom);
EXPECT_TRUE(footage.has_custom_proxy_params());
EXPECT_EQ(footage.GetEffectiveProxyParams().width, 320);
EXPECT_EQ(footage.GetEffectiveProxyParams().height, 180);
// Clearing reverts to the global config values
footage.ClearCustomProxyParams();
EXPECT_FALSE(footage.has_custom_proxy_params());
EXPECT_EQ(footage.GetEffectiveProxyParams().width,
ProxyConfigValue("ProxyWidth").value<int>());
}
Reference in New Issue View Git Blame Copy Permalink