mirror of
https://github.com/basiliscos/syncspirit.git
synced 2026-06-01 17:47:32 +00:00
737 lines
30 KiB
C++
737 lines
30 KiB
C++
// SPDX-License-Identifier: GPL-3.0-or-later
|
|
// SPDX-FileCopyrightText: 2019-2025 Ivan Baidakou
|
|
|
|
#include "test-utils.h"
|
|
#include "fs/file_actor.h"
|
|
#include "fs/utils.h"
|
|
#include "diff-builder.h"
|
|
#include "net/messages.h"
|
|
#include "test_supervisor.h"
|
|
#include "access.h"
|
|
#include "model/cluster.h"
|
|
#include "model/misc/resolver.h"
|
|
#include "access.h"
|
|
#include <filesystem>
|
|
|
|
using namespace syncspirit;
|
|
using namespace syncspirit::db;
|
|
using namespace syncspirit::test;
|
|
using namespace syncspirit::model;
|
|
using namespace syncspirit::net;
|
|
using namespace syncspirit::fs;
|
|
|
|
namespace bfs = std::filesystem;
|
|
|
|
namespace {
|
|
|
|
struct fixture_t {
|
|
using msg_t = net::message::load_cluster_response_t;
|
|
using msg_ptr_t = r::intrusive_ptr_t<msg_t>;
|
|
using blk_res_t = fs::message::block_response_t;
|
|
using blk_res_ptr_t = r::intrusive_ptr_t<blk_res_t>;
|
|
|
|
fixture_t() noexcept : root_path{unique_path()}, path_guard{root_path} {
|
|
test::init_logging();
|
|
bfs::create_directory(root_path);
|
|
sequencer = make_sequencer(67);
|
|
}
|
|
|
|
virtual supervisor_t::configure_callback_t configure() noexcept {
|
|
return [&](r::plugin::plugin_base_t &plugin) {
|
|
plugin.template with_casted<r::plugin::starter_plugin_t>([&](auto &p) {
|
|
p.subscribe_actor(r::lambda<msg_t>([&](msg_t &msg) { reply = &msg; }));
|
|
p.subscribe_actor(r::lambda<blk_res_t>([&](blk_res_t &msg) { block_reply = &msg; }));
|
|
});
|
|
};
|
|
}
|
|
|
|
virtual void run() noexcept {
|
|
auto my_id =
|
|
device_id_t::from_string("KHQNO2S-5QSILRK-YX4JZZ4-7L77APM-QNVGZJT-EKU7IFI-PNEPBMY-4MXFMQD").value();
|
|
my_device = device_t::create(my_id, "my-device").value();
|
|
|
|
cluster = new cluster_t(my_device, 1);
|
|
auto peer_id =
|
|
device_id_t::from_string("VUV42CZ-IQD5A37-RPEBPM4-VVQK6E4-6WSKC7B-PVJQHHD-4PZD44V-ENC6WAZ").value();
|
|
peer_device = device_t::create(peer_id, "peer-device").value();
|
|
|
|
cluster->get_devices().put(my_device);
|
|
cluster->get_devices().put(peer_device);
|
|
|
|
r::system_context_t ctx;
|
|
sup = ctx.create_supervisor<supervisor_t>()
|
|
.auto_finish(false)
|
|
.auto_ack_blocks(false)
|
|
.timeout(timeout)
|
|
.create_registry()
|
|
.finish();
|
|
sup->cluster = cluster;
|
|
sup->configure_callback = configure();
|
|
|
|
sup->start();
|
|
sup->do_process();
|
|
CHECK(static_cast<r::actor_base_t *>(sup.get())->access<to::state>() == r::state_t::OPERATIONAL);
|
|
|
|
auto sha256 = peer_device->device_id().get_sha256();
|
|
file_actor = sup->create_actor<fs::file_actor_t>()
|
|
.mru_size(2)
|
|
.cluster(cluster)
|
|
.sequencer(sup->sequencer)
|
|
.timeout(timeout)
|
|
.finish();
|
|
sup->do_process();
|
|
CHECK(static_cast<r::actor_base_t *>(file_actor.get())->access<to::state>() == r::state_t::OPERATIONAL);
|
|
file_addr = file_actor->get_address();
|
|
|
|
auto builder = diff_builder_t(*cluster);
|
|
builder.upsert_folder(folder_id, root_path.string(), "my-label")
|
|
.apply(*sup)
|
|
.update_peer(peer_device->device_id(), "some_name", "some-cn", true)
|
|
.apply(*sup)
|
|
.share_folder(sha256, folder_id)
|
|
.apply(*sup);
|
|
|
|
folder = cluster->get_folders().by_id(folder_id);
|
|
folder_my = folder->get_folder_infos().by_device(*my_device);
|
|
folder_peer = folder->get_folder_infos().by_device(*peer_device);
|
|
|
|
main();
|
|
reply.reset();
|
|
|
|
sup->shutdown();
|
|
sup->do_process();
|
|
|
|
CHECK(static_cast<r::actor_base_t *>(sup.get())->access<to::state>() == r::state_t::SHUT_DOWN);
|
|
}
|
|
|
|
virtual void main() noexcept {}
|
|
|
|
r::address_ptr_t file_addr;
|
|
r::pt::time_duration timeout = r::pt::millisec{10};
|
|
cluster_ptr_t cluster;
|
|
model::sequencer_ptr_t sequencer;
|
|
model::device_ptr_t peer_device;
|
|
model::device_ptr_t my_device;
|
|
model::folder_ptr_t folder;
|
|
model::folder_info_ptr_t folder_my;
|
|
model::folder_info_ptr_t folder_peer;
|
|
r::intrusive_ptr_t<supervisor_t> sup;
|
|
r::intrusive_ptr_t<fs::file_actor_t> file_actor;
|
|
bfs::path root_path;
|
|
path_guard_t path_guard;
|
|
r::system_context_t ctx;
|
|
msg_ptr_t reply;
|
|
blk_res_ptr_t block_reply;
|
|
std::string_view folder_id = "1234-5678";
|
|
};
|
|
} // namespace
|
|
|
|
void test_remote_copy() {
|
|
struct F : fixture_t {
|
|
void main() noexcept override {
|
|
proto::FileInfo pr_fi;
|
|
std::int64_t modified = 1641828421;
|
|
proto::set_name(pr_fi, "q.txt");
|
|
proto::set_modified_s(pr_fi, modified);
|
|
proto::set_sequence(pr_fi, folder_peer->get_max_sequence() + 1);
|
|
|
|
auto &v = proto::get_version(pr_fi);
|
|
proto::add_counters(v, proto::Counter(peer_device->device_id().get_uint(), 1));
|
|
|
|
auto builder = diff_builder_t(*cluster, file_addr);
|
|
|
|
auto make_file = [&]() {
|
|
auto file = file_info_t::create(sequencer->next_uuid(), pr_fi, folder_peer).value();
|
|
REQUIRE(folder_peer->add_strict(file));
|
|
return file;
|
|
};
|
|
|
|
SECTION("empty regular file") {
|
|
auto peer_file = make_file();
|
|
builder.remote_copy(*peer_file).apply(*sup);
|
|
|
|
auto my_file = folder_my->get_file_infos().by_name(peer_file->get_name());
|
|
auto &path = my_file->get_path();
|
|
REQUIRE(bfs::exists(path));
|
|
REQUIRE(bfs::file_size(path) == 0);
|
|
REQUIRE(to_unix(bfs::last_write_time(path)) == 1641828421);
|
|
}
|
|
|
|
SECTION("empty regular file a subdir") {
|
|
auto name = std::string_view("a/b/c/d/e.txt");
|
|
proto::set_name(pr_fi, name);
|
|
auto peer_file = make_file();
|
|
builder.remote_copy(*peer_file).apply(*sup);
|
|
|
|
auto file = folder_my->get_file_infos().by_name(name);
|
|
auto &path = file->get_path();
|
|
REQUIRE(bfs::exists(path));
|
|
REQUIRE(bfs::file_size(path) == 0);
|
|
}
|
|
|
|
SECTION("non-empty regular file") {
|
|
proto::set_size(pr_fi, 5);
|
|
proto::set_block_size(pr_fi, 5);
|
|
|
|
auto block_hash = utils::sha256_digest(as_bytes("12345")).value();
|
|
auto &b = proto::add_blocks(pr_fi);
|
|
proto::set_hash(b, block_hash);
|
|
proto::set_size(b, 5);
|
|
auto bi = block_info_t::create(b).value();
|
|
auto &blocks_map = cluster->get_blocks();
|
|
blocks_map.put(bi);
|
|
|
|
auto peer_file = make_file();
|
|
peer_file->assign_block(bi, 0);
|
|
builder.remote_copy(*peer_file).apply(*sup);
|
|
|
|
auto file = folder_my->get_file_infos().by_name(proto::get_name(pr_fi));
|
|
|
|
auto filename = std::string(file->get_name()) + ".syncspirit-tmp";
|
|
auto path = root_path / filename;
|
|
REQUIRE(bfs::exists(path));
|
|
REQUIRE(bfs::file_size(path) == 5);
|
|
}
|
|
|
|
SECTION("directory") {
|
|
proto::set_type(pr_fi, proto::FileInfoType::DIRECTORY);
|
|
|
|
auto peer_file = make_file();
|
|
builder.remote_copy(*peer_file).apply(*sup);
|
|
|
|
auto file = folder_my->get_file_infos().by_name(proto::get_name(pr_fi));
|
|
|
|
auto &path = file->get_path();
|
|
REQUIRE(bfs::exists(path));
|
|
REQUIRE(bfs::is_directory(path));
|
|
}
|
|
|
|
SECTION("symlink") {
|
|
bfs::path target = root_path / "some-existing-file";
|
|
write_file(target, "zzz");
|
|
|
|
proto::set_type(pr_fi, proto::FileInfoType::SYMLINK);
|
|
proto::set_symlink_target(pr_fi, target.string());
|
|
|
|
auto peer_file = make_file();
|
|
builder.remote_copy(*peer_file).apply(*sup);
|
|
|
|
auto file = folder_my->get_file_infos().by_name(proto::get_name(pr_fi));
|
|
|
|
auto &path = file->get_path();
|
|
CHECK(!bfs::exists(path));
|
|
#ifndef SYNCSPIRIT_WIN
|
|
CHECK(bfs::is_symlink(path));
|
|
CHECK(bfs::read_symlink(path) == target);
|
|
#endif
|
|
}
|
|
|
|
SECTION("deleted file") {
|
|
proto::set_deleted(pr_fi, true);
|
|
bfs::path target = root_path / proto::get_name(pr_fi);
|
|
write_file(target, "zzz");
|
|
REQUIRE(bfs::exists(target));
|
|
|
|
auto peer_file = make_file();
|
|
builder.remote_copy(*peer_file).apply(*sup);
|
|
|
|
auto file = folder_my->get_file_infos().by_name(proto::get_name(pr_fi));
|
|
CHECK(file->is_deleted());
|
|
|
|
REQUIRE(!bfs::exists(target));
|
|
}
|
|
}
|
|
};
|
|
F().run();
|
|
}
|
|
|
|
void test_append_block() {
|
|
struct F : fixture_t {
|
|
void main() noexcept override {
|
|
std::int64_t modified = 1641828421;
|
|
proto::FileInfo pr_source;
|
|
auto next_sequence = 7ul;
|
|
|
|
proto::set_name(pr_source, "q.txt");
|
|
proto::set_modified_s(pr_source, modified);
|
|
proto::set_block_size(pr_source, 5);
|
|
|
|
auto &v = proto::get_version(pr_source);
|
|
proto::add_counters(v, proto::Counter(peer_device->device_id().get_uint(), 1));
|
|
|
|
auto block_hash_1 = utils::sha256_digest(as_bytes("12345")).value();
|
|
auto bi = proto::BlockInfo();
|
|
proto::set_hash(bi, block_hash_1);
|
|
proto::set_size(bi, 5);
|
|
auto b = block_info_t::create(bi).value();
|
|
|
|
auto block_hash_2 = utils::sha256_digest(as_bytes("67890")).value();
|
|
auto bi2 = proto::BlockInfo();
|
|
proto::set_hash(bi2, block_hash_2);
|
|
proto::set_size(bi2, 5);
|
|
auto b2 = block_info_t::create(bi).value();
|
|
|
|
cluster->get_blocks().put(b);
|
|
cluster->get_blocks().put(b2);
|
|
auto blocks = std::vector<block_info_ptr_t>{b, b2};
|
|
|
|
auto make_file = [&](size_t block_count) {
|
|
proto::set_sequence(pr_source, ++next_sequence);
|
|
|
|
auto copy = pr_source;
|
|
auto &v2 = proto::get_version(copy);
|
|
auto version = model::version_t(v2);
|
|
version.update(*peer_device);
|
|
version.to_proto(v2);
|
|
|
|
auto file = file_info_t::create(sequencer->next_uuid(), copy, folder_peer).value();
|
|
for (size_t i = 0; i < block_count; ++i) {
|
|
file->assign_block(blocks[i], i);
|
|
}
|
|
REQUIRE(folder_peer->add_strict(file));
|
|
return file;
|
|
};
|
|
|
|
auto builder = diff_builder_t(*cluster, file_addr);
|
|
|
|
SECTION("file with 1 block") {
|
|
proto::set_size(pr_source, 5);
|
|
auto peer_file = make_file(1);
|
|
builder.append_block(*peer_file, 0, as_owned_bytes("12345"))
|
|
.apply(*sup)
|
|
.finish_file(*peer_file)
|
|
.apply(*sup);
|
|
|
|
auto file = folder_my->get_file_infos().by_name(proto::get_name(pr_source));
|
|
auto path = root_path / std::string(file->get_name());
|
|
REQUIRE(bfs::exists(path));
|
|
REQUIRE(bfs::file_size(path) == 5);
|
|
auto data = read_file(path);
|
|
CHECK(data == "12345");
|
|
CHECK(to_unix(bfs::last_write_time(path)) == 1641828421);
|
|
}
|
|
SECTION("file with 2 different blocks") {
|
|
proto::set_size(pr_source, 10);
|
|
|
|
auto peer_file = make_file(2);
|
|
|
|
builder.append_block(*peer_file, 0, as_owned_bytes("12345")).apply(*sup);
|
|
|
|
auto filename = std::string(peer_file->get_name()) + ".syncspirit-tmp";
|
|
auto path = root_path / filename;
|
|
#ifndef SYNCSPIRIT_WIN
|
|
REQUIRE(bfs::exists(path));
|
|
REQUIRE(bfs::file_size(path) == 10);
|
|
auto data = read_file(path);
|
|
CHECK(data.substr(0, 5) == "12345");
|
|
#endif
|
|
builder.append_block(*peer_file, 1, as_owned_bytes("67890")).apply(*sup);
|
|
|
|
SECTION("add 2nd block") {
|
|
builder.finish_file(*peer_file).apply(*sup);
|
|
|
|
filename = std::string(peer_file->get_name());
|
|
path = root_path / filename;
|
|
REQUIRE(bfs::exists(path));
|
|
REQUIRE(bfs::file_size(path) == 10);
|
|
auto data = read_file(path);
|
|
CHECK(data == "1234567890");
|
|
CHECK(to_unix(bfs::last_write_time(path)) == 1641828421);
|
|
}
|
|
|
|
#ifndef SYNCSPIRIT_WIN
|
|
SECTION("remove folder (simulate err)") {
|
|
bfs::remove_all(root_path);
|
|
builder.finish_file(*peer_file).apply(*sup);
|
|
CHECK(static_cast<r::actor_base_t *>(file_actor.get())->access<to::state>() ==
|
|
r::state_t::SHUT_DOWN);
|
|
}
|
|
#endif
|
|
}
|
|
}
|
|
};
|
|
F().run();
|
|
}
|
|
|
|
void test_clone_block() {
|
|
struct F : fixture_t {
|
|
void main() noexcept override {
|
|
|
|
auto hash_1 = utils::sha256_digest(as_bytes("12345")).value();
|
|
auto bi = proto::BlockInfo();
|
|
proto::set_size(bi, 5);
|
|
proto::set_hash(bi, hash_1);
|
|
auto b = block_info_t::create(bi).value();
|
|
|
|
auto hash_2 = utils::sha256_digest(as_bytes("67890")).value();
|
|
auto bi2 = proto::BlockInfo();
|
|
proto::set_size(bi2, 5);
|
|
proto::set_hash(bi2, hash_2);
|
|
auto b2 = block_info_t::create(bi2).value();
|
|
|
|
cluster->get_blocks().put(b);
|
|
cluster->get_blocks().put(b2);
|
|
auto blocks = std::vector<block_info_ptr_t>{b, b2};
|
|
|
|
std::int64_t modified = 1641828421;
|
|
proto::FileInfo pr_source;
|
|
proto::set_name(pr_source, "a.txt");
|
|
proto::set_modified_s(pr_source, modified);
|
|
proto::set_block_size(pr_source, 5);
|
|
|
|
auto &v = proto::get_version(pr_source);
|
|
proto::add_counters(v, proto::Counter(peer_device->device_id().get_uint(), 1));
|
|
|
|
auto next_sequence = 7ul;
|
|
|
|
auto make_file = [&](const proto::FileInfo &fi, size_t count) {
|
|
auto copy = fi;
|
|
proto::set_sequence(copy, ++next_sequence);
|
|
auto file = file_info_t::create(sequencer->next_uuid(), copy, folder_peer).value();
|
|
for (size_t i = 0; i < count; ++i) {
|
|
file->assign_block(blocks[i], i);
|
|
}
|
|
REQUIRE(folder_peer->add_strict(file));
|
|
return file;
|
|
};
|
|
|
|
auto builder = diff_builder_t(*cluster, file_addr);
|
|
|
|
SECTION("source & target are different files") {
|
|
proto::FileInfo pr_target;
|
|
proto::set_name(pr_target, "b.txt");
|
|
proto::set_block_size(pr_target, 5);
|
|
proto::set_version(pr_target, v);
|
|
|
|
SECTION("single block target file") {
|
|
proto::set_size(pr_source, 5);
|
|
proto::set_size(pr_target, 5);
|
|
proto::set_modified_s(pr_target, modified);
|
|
|
|
auto source = make_file(pr_source, 1);
|
|
auto target = make_file(pr_target, 1);
|
|
|
|
auto source_file = folder_peer->get_file_infos().by_name(proto::get_name(pr_source));
|
|
builder.append_block(*source_file, 0, as_owned_bytes("12345"))
|
|
.apply(*sup)
|
|
.finish_file(*source_file)
|
|
.apply(*sup);
|
|
|
|
auto target_file = folder_peer->get_file_infos().by_name(proto::get_name(pr_target));
|
|
auto block = source_file->get_blocks()[0];
|
|
auto file_block = model::file_block_t(block.get(), target_file.get(), 0);
|
|
builder.clone_block(file_block).apply(*sup).finish_file(*target).apply(*sup);
|
|
|
|
auto path = root_path / std::string(target_file->get_name());
|
|
REQUIRE(bfs::exists(path));
|
|
REQUIRE(bfs::file_size(path) == 5);
|
|
auto data = read_file(path);
|
|
CHECK(data == "12345");
|
|
CHECK(to_unix(bfs::last_write_time(path)) == 1641828421);
|
|
}
|
|
|
|
SECTION("multi block target file") {
|
|
proto::set_size(pr_source, 10);
|
|
proto::set_size(pr_target, 10);
|
|
auto source = make_file(pr_source, 2);
|
|
auto target = make_file(pr_target, 2);
|
|
|
|
auto source_file = folder_peer->get_file_infos().by_name(proto::get_name(pr_source));
|
|
|
|
builder.append_block(*source_file, 0, as_owned_bytes("12345"))
|
|
.append_block(*source_file, 1, as_owned_bytes("67890"))
|
|
.apply(*sup);
|
|
|
|
auto target_file = folder_peer->get_file_infos().by_name(proto::get_name(pr_target));
|
|
auto blocks = source_file->get_blocks();
|
|
auto fb_1 = model::file_block_t(blocks[0].get(), target_file.get(), 0);
|
|
auto fb_2 = model::file_block_t(blocks[1].get(), target_file.get(), 1);
|
|
builder.clone_block(fb_1).clone_block(fb_2).apply(*sup).finish_file(*target).apply(*sup);
|
|
|
|
auto filename = std::string(target_file->get_name());
|
|
auto path = root_path / filename;
|
|
REQUIRE(bfs::exists(path));
|
|
REQUIRE(bfs::file_size(path) == 10);
|
|
auto data = read_file(path);
|
|
CHECK(data == "1234567890");
|
|
}
|
|
|
|
SECTION("source/target different sizes") {
|
|
proto::set_size(pr_source, 5);
|
|
proto::set_size(pr_target, 10);
|
|
auto target = make_file(pr_target, 2);
|
|
|
|
proto::set_sequence(pr_source, folder_peer->get_max_sequence() + 1);
|
|
auto source = file_info_t::create(sequencer->next_uuid(), pr_source, folder_peer).value();
|
|
source->assign_block(blocks[1], 0);
|
|
REQUIRE(folder_peer->add_strict(source));
|
|
|
|
auto source_file = folder_peer->get_file_infos().by_name(proto::get_name(pr_source));
|
|
auto target_file = folder_peer->get_file_infos().by_name(proto::get_name(pr_target));
|
|
|
|
builder.append_block(*source_file, 0, as_owned_bytes("67890"))
|
|
.append_block(*target_file, 0, as_owned_bytes("12345"))
|
|
.apply(*sup);
|
|
|
|
auto blocks = source_file->get_blocks();
|
|
auto fb = model::file_block_t(blocks[0].get(), target_file.get(), 1);
|
|
builder.clone_block(fb).apply(*sup).finish_file(*target).apply(*sup);
|
|
|
|
auto filename = std::string(target_file->get_name());
|
|
auto path = root_path / filename;
|
|
REQUIRE(bfs::exists(path));
|
|
REQUIRE(bfs::file_size(path) == 10);
|
|
auto data = read_file(path);
|
|
CHECK(data == "1234567890");
|
|
}
|
|
}
|
|
|
|
SECTION("source & target are is the same file") {
|
|
proto::set_sequence(pr_source, folder_peer->get_max_sequence() + 1);
|
|
proto::set_size(pr_source, 10);
|
|
|
|
auto source = file_info_t::create(sequencer->next_uuid(), pr_source, folder_peer).value();
|
|
source->assign_block(blocks[0], 0);
|
|
source->assign_block(blocks[0], 1);
|
|
REQUIRE(folder_peer->add_strict(source));
|
|
|
|
auto source_file = folder_peer->get_file_infos().by_name(proto::get_name(pr_source));
|
|
auto target_file = source_file;
|
|
|
|
builder.append_block(*source_file, 0, as_owned_bytes("12345")).apply(*sup);
|
|
|
|
auto block = source_file->get_blocks()[0];
|
|
auto file_block = model::file_block_t(block.get(), target_file.get(), 1);
|
|
builder.clone_block(file_block).apply(*sup).finish_file(*source).apply(*sup);
|
|
|
|
auto path = root_path / std::string(target_file->get_name());
|
|
REQUIRE(bfs::exists(path));
|
|
REQUIRE(bfs::file_size(path) == 10);
|
|
auto data = read_file(path);
|
|
CHECK(data == "1234512345");
|
|
CHECK(to_unix(bfs::last_write_time(path)) == 1641828421);
|
|
}
|
|
}
|
|
};
|
|
F().run();
|
|
}
|
|
|
|
void test_requesting_block() {
|
|
struct F : fixture_t {
|
|
void main() noexcept override {
|
|
|
|
auto hash_1 = utils::sha256_digest(as_bytes("12345")).value();
|
|
auto bi = proto::BlockInfo();
|
|
proto::set_size(bi, 5);
|
|
proto::set_hash(bi, hash_1);
|
|
auto b = block_info_t::create(bi).value();
|
|
|
|
auto hash_2 = utils::sha256_digest(as_bytes("67890")).value();
|
|
auto bi2 = proto::BlockInfo();
|
|
proto::set_size(bi2, 5);
|
|
proto::set_hash(bi2, hash_2);
|
|
auto b2 = block_info_t::create(bi2).value();
|
|
|
|
cluster->get_blocks().put(b);
|
|
cluster->get_blocks().put(b2);
|
|
bfs::path target = root_path / "a.txt";
|
|
|
|
std::int64_t modified = 1641828421;
|
|
proto::FileInfo pr_source;
|
|
|
|
proto::set_name(pr_source, "a.txt");
|
|
proto::set_modified_s(pr_source, modified);
|
|
proto::set_block_size(pr_source, 5);
|
|
proto::set_size(pr_source, 10);
|
|
proto::set_sequence(pr_source, folder_my->get_max_sequence() + 1);
|
|
|
|
auto &v = proto::get_version(pr_source);
|
|
proto::add_counters(v, proto::Counter(peer_device->device_id().get_uint(), 1));
|
|
|
|
proto::add_blocks(pr_source, bi);
|
|
proto::add_blocks(pr_source, bi2);
|
|
|
|
auto file = file_info_t::create(sequencer->next_uuid(), pr_source, folder_my).value();
|
|
file->assign_block(b, 0);
|
|
file->assign_block(b2, 1);
|
|
folder_my->add_relaxed(file);
|
|
|
|
auto req = proto::Request();
|
|
proto::set_folder(req, folder->get_id());
|
|
proto::set_name(req, "a.txt");
|
|
proto::set_size(req, 5);
|
|
|
|
auto msg =
|
|
r::make_message<fs::payload::block_request_t>(file_actor->get_address(), req, sup->get_address());
|
|
|
|
SECTION("error, no file") {
|
|
sup->put(msg);
|
|
sup->do_process();
|
|
REQUIRE(block_reply);
|
|
REQUIRE(block_reply->payload.ec);
|
|
REQUIRE(block_reply->payload.data.empty());
|
|
}
|
|
|
|
SECTION("error, oversized request") {
|
|
write_file(target, "1234");
|
|
sup->put(msg);
|
|
sup->do_process();
|
|
REQUIRE(block_reply);
|
|
REQUIRE(block_reply->payload.ec);
|
|
REQUIRE(block_reply->payload.data.empty());
|
|
}
|
|
|
|
SECTION("successful file reading") {
|
|
write_file(target, "1234567890");
|
|
sup->put(msg);
|
|
sup->do_process();
|
|
REQUIRE(block_reply);
|
|
REQUIRE(!block_reply->payload.ec);
|
|
REQUIRE(block_reply->payload.data == as_bytes("12345"));
|
|
|
|
proto::set_offset(req, 5);
|
|
auto msg =
|
|
r::make_message<fs::payload::block_request_t>(file_actor->get_address(), req, sup->get_address());
|
|
sup->put(msg);
|
|
sup->do_process();
|
|
REQUIRE(block_reply);
|
|
REQUIRE(!block_reply->payload.ec);
|
|
REQUIRE(block_reply->payload.data == as_bytes("67890"));
|
|
}
|
|
}
|
|
};
|
|
F().run();
|
|
}
|
|
|
|
void test_conflicts() {
|
|
struct F : fixture_t {
|
|
void main() noexcept override {
|
|
auto builder = diff_builder_t(*cluster, file_addr);
|
|
auto &blocks_map = cluster->get_blocks();
|
|
|
|
proto::FileInfo pr_fi;
|
|
std::int64_t modified = 1641828421;
|
|
proto::set_name(pr_fi, "q.txt");
|
|
proto::set_modified_s(pr_fi, modified);
|
|
proto::set_sequence(pr_fi, folder_peer->get_max_sequence() + 1);
|
|
|
|
SECTION("non-empty (via file_finish)") {
|
|
proto::set_block_size(pr_fi, 5);
|
|
proto::set_size(pr_fi, 5);
|
|
|
|
auto peer_block = []() {
|
|
auto block = proto::BlockInfo();
|
|
auto hash = utils::sha256_digest(as_bytes("12345")).value();
|
|
proto::set_hash(block, hash);
|
|
proto::set_size(block, 5);
|
|
auto bi = block_info_t::create(block).value();
|
|
return bi;
|
|
}();
|
|
blocks_map.put(peer_block);
|
|
|
|
auto my_block = []() {
|
|
auto block = proto::BlockInfo();
|
|
auto hash = utils::sha256_digest(as_bytes("67890")).value();
|
|
proto::set_hash(block, hash);
|
|
proto::set_size(block, 5);
|
|
auto bi = block_info_t::create(block).value();
|
|
return bi;
|
|
}();
|
|
blocks_map.put(my_block);
|
|
|
|
SECTION("remote win") {
|
|
auto peer_file = [&]() {
|
|
auto &v = proto::get_version(pr_fi);
|
|
auto &c = proto::add_counters(v);
|
|
proto::set_id(c, peer_device->device_id().get_uint());
|
|
proto::set_value(c, 10);
|
|
proto::add_blocks(pr_fi, peer_block->as_bep(0));
|
|
proto::set_modified_s(pr_fi, 1734690000);
|
|
|
|
auto file = file_info_t::create(sequencer->next_uuid(), pr_fi, folder_peer).value();
|
|
REQUIRE(folder_peer->add_strict(file));
|
|
auto vv = &proto::get_version(pr_fi);
|
|
proto::clear_counters(*vv);
|
|
proto::clear_blocks(pr_fi);
|
|
file->assign_block(peer_block, 0);
|
|
return file;
|
|
}();
|
|
|
|
auto my_file = [&]() {
|
|
proto:
|
|
auto &v = proto::get_version(pr_fi);
|
|
proto::add_counters(v, proto::Counter(my_device->device_id().get_uint(), 5));
|
|
proto::set_modified_s(pr_fi, 1734600000);
|
|
proto::add_blocks(pr_fi, my_block->as_bep(0));
|
|
auto file = file_info_t::create(sequencer->next_uuid(), pr_fi, folder_my).value();
|
|
REQUIRE(folder_my->add_strict(file));
|
|
file->mark_local();
|
|
|
|
file->assign_block(my_block, 0);
|
|
return file;
|
|
}();
|
|
|
|
bfs::path kept_file = root_path / proto::get_name(pr_fi);
|
|
write_file(kept_file, "12345");
|
|
|
|
REQUIRE(model::resolve(*peer_file) == advance_action_t::resolve_remote_win);
|
|
builder.append_block(*peer_file, 0, as_owned_bytes("67890"))
|
|
.apply(*sup)
|
|
.finish_file(*peer_file)
|
|
.apply(*sup);
|
|
|
|
auto conflict_file = root_path / my_file->make_conflicting_name();
|
|
CHECK(read_file(kept_file) == "67890");
|
|
CHECK(bfs::exists(conflict_file));
|
|
CHECK(read_file(conflict_file) == "12345");
|
|
}
|
|
}
|
|
SECTION("remote win emtpy (file vs directory)") {
|
|
auto peer_file = [&]() {
|
|
auto &v = proto::get_version(pr_fi);
|
|
proto::add_counters(v, proto::Counter(peer_device->device_id().get_uint(), 10));
|
|
proto::set_modified_s(pr_fi, 1734690000);
|
|
auto file = file_info_t::create(sequencer->next_uuid(), pr_fi, folder_peer).value();
|
|
REQUIRE(folder_peer->add_strict(file));
|
|
auto vv = &proto::get_version(pr_fi);
|
|
proto::clear_counters(*vv);
|
|
return file;
|
|
}();
|
|
|
|
auto my_file = [&]() {
|
|
auto &v = proto::get_version(pr_fi);
|
|
proto::add_counters(v, proto::Counter(my_device->device_id().get_uint(), 5));
|
|
proto::set_modified_s(pr_fi, 1734600000);
|
|
auto file = file_info_t::create(sequencer->next_uuid(), pr_fi, folder_my).value();
|
|
REQUIRE(folder_my->add_strict(file));
|
|
file->mark_local();
|
|
return file;
|
|
}();
|
|
|
|
bfs::path kept_file = root_path / proto::get_name(pr_fi);
|
|
bfs::create_directories(kept_file);
|
|
|
|
builder.advance(*peer_file).apply(*sup);
|
|
|
|
auto conflict_file = root_path / my_file->make_conflicting_name();
|
|
CHECK(bfs::exists(conflict_file));
|
|
CHECK(bfs::exists(kept_file));
|
|
CHECK(bfs::is_directory(conflict_file));
|
|
CHECK(bfs::is_regular_file(kept_file));
|
|
}
|
|
}
|
|
};
|
|
F().run();
|
|
}
|
|
|
|
int _init() {
|
|
REGISTER_TEST_CASE(test_remote_copy, "test_remote_copy", "[fs]");
|
|
REGISTER_TEST_CASE(test_append_block, "test_append_block", "[fs]");
|
|
REGISTER_TEST_CASE(test_clone_block, "test_clone_block", "[fs]");
|
|
REGISTER_TEST_CASE(test_requesting_block, "test_requesting_block", "[fs]");
|
|
REGISTER_TEST_CASE(test_conflicts, "test_conflicts", "[fs]");
|
|
return 1;
|
|
}
|
|
|
|
static int v = _init();
|