mirror of
https://github.com/mhx/dwarfs.git
synced 2025-08-04 02:06:22 -04:00
123 lines
3.5 KiB
C++
123 lines
3.5 KiB
C++
/* vim:set ts=2 sw=2 sts=2 et: */
|
|
/**
|
|
* \author Marcus Holland-Moritz (github@mhxnet.de)
|
|
* \copyright Copyright (c) Marcus Holland-Moritz
|
|
*
|
|
* This file is part of dwarfs.
|
|
*
|
|
* dwarfs is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation, either version 3 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* dwarfs is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with dwarfs. If not, see <https://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
#include <exception>
|
|
#include <filesystem>
|
|
|
|
#include <gmock/gmock.h>
|
|
#include <gtest/gtest.h>
|
|
|
|
#include <boost/program_options.hpp>
|
|
|
|
#include <folly/String.h>
|
|
|
|
#include "dwarfs/categorizer.h"
|
|
#include "dwarfs/mmap.h"
|
|
|
|
#include "test_logger.h"
|
|
|
|
using namespace dwarfs;
|
|
using testing::MatchesRegex;
|
|
|
|
namespace fs = std::filesystem;
|
|
|
|
auto test_dir = fs::path(TEST_DATA_DIR);
|
|
|
|
TEST(pcmaudio_categorizer, requirements) {
|
|
test::test_logger logger(logger::INFO);
|
|
boost::program_options::variables_map vm;
|
|
auto& catreg = categorizer_registry::instance();
|
|
auto catmgr = categorizer_manager(logger);
|
|
|
|
catmgr.add(catreg.create(logger, "pcmaudio", vm));
|
|
|
|
try {
|
|
catmgr.set_metadata_requirements(
|
|
catmgr.category_value("pcmaudio/metadata").value(),
|
|
R"({"endianness": ["set", ["big"]], "bytes_per_sample": ["range", 2, 3]})");
|
|
FAIL() << "expected std::runtime_error";
|
|
} catch (std::runtime_error const& e) {
|
|
EXPECT_STREQ(
|
|
"unsupported metadata requirements: bytes_per_sample, endianness",
|
|
e.what());
|
|
} catch (...) {
|
|
FAIL() << "unexpected exception: "
|
|
<< folly::exceptionStr(std::current_exception());
|
|
}
|
|
|
|
catmgr.set_metadata_requirements(
|
|
catmgr.category_value("pcmaudio/waveform").value(),
|
|
R"({"endianness": ["set", ["mixed"]], "bytes_per_sample": ["range", 2, 3]})");
|
|
|
|
auto wav = test_dir / "pcmaudio" / "test16.wav";
|
|
auto mm = mmap(wav);
|
|
|
|
{
|
|
auto job = catmgr.job(wav);
|
|
job.set_total_size(mm.size());
|
|
|
|
EXPECT_TRUE(logger.empty());
|
|
|
|
job.categorize_random_access(mm.span());
|
|
auto frag = job.result();
|
|
|
|
ASSERT_EQ(1, logger.get_log().size());
|
|
auto const& ent = logger.get_log().front();
|
|
EXPECT_EQ(logger::WARN, ent.level);
|
|
EXPECT_THAT(
|
|
ent.output,
|
|
MatchesRegex(
|
|
R"(^\[WAV\] ".*": endianness 'little' does not meet requirements$)"));
|
|
|
|
EXPECT_TRUE(frag.empty());
|
|
|
|
logger.clear();
|
|
}
|
|
|
|
catmgr.set_metadata_requirements(
|
|
catmgr.category_value("pcmaudio/waveform").value(),
|
|
R"({"endianness": ["set", ["big", "little"]], "bytes_per_sample": ["range", 1, 4]})");
|
|
|
|
{
|
|
auto job = catmgr.job(wav);
|
|
job.set_total_size(mm.size());
|
|
|
|
EXPECT_TRUE(logger.empty());
|
|
|
|
job.categorize_random_access(mm.span());
|
|
auto frag = job.result();
|
|
|
|
EXPECT_TRUE(logger.empty());
|
|
|
|
EXPECT_EQ(2, frag.size());
|
|
|
|
auto const& first = frag.span()[0];
|
|
auto const& second = frag.span()[1];
|
|
EXPECT_EQ("pcmaudio/metadata",
|
|
catmgr.category_name(first.category().value()));
|
|
EXPECT_EQ(44, first.size());
|
|
EXPECT_EQ("pcmaudio/waveform",
|
|
catmgr.category_name(second.category().value()));
|
|
EXPECT_EQ(14, second.size());
|
|
EXPECT_EQ(mm.size(), first.size() + second.size());
|
|
}
|
|
}
|