dwarfs/test/test_helpers.h
Marcus Holland-Moritz a0e70d5aef Build fix
2023-07-07 00:28:50 +02:00

139 lines
4.1 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/>.
*/
#pragma once
#include <filesystem>
#include <functional>
#include <map>
#include <memory>
#include <optional>
#include <set>
#include <span>
#include <string>
#include <string_view>
#include <variant>
#include <vector>
#include "dwarfs/file_stat.h"
#include "dwarfs/os_access.h"
#include "dwarfs/script.h"
namespace dwarfs::test {
struct simplestat {
file_stat::ino_type ino;
file_stat::mode_type mode;
file_stat::nlink_type nlink;
file_stat::uid_type uid;
file_stat::gid_type gid;
file_stat::off_type size;
file_stat::dev_type rdev;
file_stat::time_type atime;
file_stat::time_type mtime;
file_stat::time_type ctime;
posix_file_type::value type() const {
return static_cast<posix_file_type::value>(mode & posix_file_type::mask);
}
};
class os_access_mock : public os_access {
private:
struct mock_directory;
struct mock_dirent;
public:
using value_variant_type =
std::variant<std::monostate, std::string, std::function<std::string()>,
std::unique_ptr<mock_directory>>;
os_access_mock();
~os_access_mock();
static std::shared_ptr<os_access_mock> create_test_instance();
size_t size() const;
void add(std::filesystem::path const& path, simplestat const& st);
void add(std::filesystem::path const& path, simplestat const& st,
std::string const& contents);
void add(std::filesystem::path const& path, simplestat const& st,
std::function<std::string()> generator);
void add_dir(std::filesystem::path const& path);
void add_file(std::filesystem::path const& path, size_t size);
void add_file(std::filesystem::path const& path, std::string const& contents);
void set_access_fail(std::filesystem::path const& path);
std::shared_ptr<dir_reader>
opendir(std::filesystem::path const& path) const override;
file_stat symlink_info(std::filesystem::path const& path) const override;
std::filesystem::path
read_symlink(std::filesystem::path const& path) const override;
std::shared_ptr<mmif>
map_file(std::filesystem::path const& path, size_t size) const override;
int access(std::filesystem::path const&, int) const override;
private:
static std::vector<std::string> splitpath(std::filesystem::path const& path);
struct mock_dirent* find(std::filesystem::path const& path) const;
struct mock_dirent* find(std::vector<std::string> parts) const;
void add_internal(std::filesystem::path const& path, simplestat const& st,
value_variant_type var);
std::unique_ptr<mock_dirent> root_;
size_t ino_{1000000};
std::set<std::filesystem::path> access_fail_set_;
};
class script_mock : public script {
public:
bool has_configure() const override { return true; }
bool has_filter() const override { return true; }
bool has_transform() const override { return true; }
bool has_order() const override { return true; }
void configure(options_interface const& /*oi*/) override {}
bool filter(entry_interface const& /*ei*/) override { return true; }
void transform(entry_interface& /*ei*/) override {
// do nothing
}
void order(inode_vector& /*iv*/) override {
// do nothing
}
};
extern std::map<std::string, simplestat> statmap;
std::optional<std::filesystem::path> find_binary(std::string_view name);
std::span<std::pair<simplestat, std::string_view> const> test_dirtree();
} // namespace dwarfs::test