shaka-packager/packager/media/test/test_data_util.cc

47 lines
1.4 KiB
C++

// Copyright (c) 2012 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "packager/media/test/test_data_util.h"
#include "glog/logging.h"
namespace shaka {
namespace media {
// Returns a file path for a file in the media/test/data directory.
std::filesystem::path GetTestDataFilePath(const std::string& name) {
auto data_dir = std::filesystem::u8path(TEST_DATA_DIR);
return data_dir / name;
}
// Returns a file path for a file in the media/app/test/testdata directory.
std::filesystem::path GetAppTestDataFilePath(const std::string& name) {
auto data_dir = std::filesystem::u8path(TEST_DATA_DIR);
auto app_data_dir =
data_dir.parent_path().parent_path() / "app" / "test" / "testdata";
return app_data_dir / name;
}
// Reads a test file from media/test/data directory and returns its content.
std::vector<uint8_t> ReadTestDataFile(const std::string& name) {
auto path = GetTestDataFilePath(name);
FILE* f = fopen(path.string().c_str(), "rb");
if (!f) {
LOG(ERROR) << "Failed to read test data from " << path;
return std::vector<uint8_t>();
}
std::vector<uint8_t> data;
data.resize(std::filesystem::file_size(path));
size_t size = fread(data.data(), 1, data.size(), f);
data.resize(size);
fclose(f);
return data;
}
} // namespace media
} // namespace shaka