037-diff-update_folder.cpp 5.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135
  1. // SPDX-License-Identifier: GPL-3.0-or-later
  2. // SPDX-FileCopyrightText: 2019-2023 Ivan Baidakou
  3. #include "test-utils.h"
  4. #include "access.h"
  5. #include "model/cluster.h"
  6. #include "model/misc/error_code.h"
  7. #include "model/diff/modify/create_folder.h"
  8. #include "model/diff/modify/share_folder.h"
  9. #include "model/diff/peer/update_folder.h"
  10. using namespace syncspirit;
  11. using namespace syncspirit::model;
  12. using namespace syncspirit::proto;
  13. using namespace syncspirit::test;
  14. TEST_CASE("update folder (via Index)", "[model]") {
  15. auto my_id = device_id_t::from_string("KHQNO2S-5QSILRK-YX4JZZ4-7L77APM-QNVGZJT-EKU7IFI-PNEPBMY-4MXFMQD").value();
  16. auto my_device = device_t::create(my_id, "my-device").value();
  17. auto peer_id = device_id_t::from_string("VUV42CZ-IQD5A37-RPEBPM4-VVQK6E4-6WSKC7B-PVJQHHD-4PZD44V-ENC6WAZ").value();
  18. auto peer_device = device_t::create(peer_id, "peer-device").value();
  19. auto cluster = cluster_ptr_t(new cluster_t(my_device, 1, 1));
  20. cluster->get_devices().put(my_device);
  21. cluster->get_devices().put(peer_device);
  22. auto &folders = cluster->get_folders();
  23. db::Folder db_folder_1;
  24. db_folder_1.set_id("1234-5678");
  25. db_folder_1.set_label("my-label");
  26. db_folder_1.set_path("/my/path");
  27. db::Folder db_folder_2;
  28. db_folder_2.set_id("5555-4444");
  29. db_folder_2.set_label("my-l2");
  30. db_folder_2.set_path("/my/path/2");
  31. auto diff = diff::cluster_diff_ptr_t(new diff::modify::create_folder_t(db_folder_1));
  32. REQUIRE(diff->apply(*cluster));
  33. diff = diff::cluster_diff_ptr_t(new diff::modify::create_folder_t(db_folder_2));
  34. REQUIRE(diff->apply(*cluster));
  35. auto folder = folders.by_id(db_folder_1.id());
  36. diff = diff::cluster_diff_ptr_t(new diff::modify::share_folder_t(peer_id.get_sha256(), db_folder_1.id()));
  37. REQUIRE(diff->apply(*cluster));
  38. auto pr_index = proto::Index();
  39. pr_index.set_folder(db_folder_1.id());
  40. SECTION("successful case") {
  41. auto peer_folder_info = folder->get_folder_infos().by_device(*peer_device);
  42. auto file = pr_index.add_files();
  43. file->set_name("a.txt");
  44. file->set_sequence(10ul);
  45. file->set_size(5ul);
  46. file->set_block_size(5ul);
  47. file->set_modified_s(1);
  48. auto b = file->add_blocks();
  49. b->set_hash("123");
  50. b->set_size(5ul);
  51. peer_folder_info->set_max_sequence(10ul);
  52. diff = diff::peer::update_folder_t::create(*cluster, *peer_device, pr_index).value();
  53. REQUIRE(diff->apply(*cluster));
  54. auto &peer_files = peer_folder_info->get_file_infos();
  55. REQUIRE(peer_files.size() == 1);
  56. CHECK(peer_folder_info->is_actual());
  57. auto f = peer_files.by_name(file->name());
  58. REQUIRE(f);
  59. auto key = std::string(f->get_key());
  60. SECTION("when a file with existing name is added, key is kept") {
  61. auto index_update = proto::IndexUpdate{};
  62. index_update.set_folder(pr_index.folder());
  63. auto same_file = index_update.add_files();
  64. *same_file = *file;
  65. same_file->set_modified_s(2);
  66. same_file->set_sequence(11ul);
  67. peer_folder_info->set_max_sequence(11ul);
  68. diff = diff::peer::update_folder_t::create(*cluster, *peer_device, index_update).value();
  69. REQUIRE(diff->apply(*cluster));
  70. REQUIRE(peer_files.size() == 1);
  71. auto same_f = peer_files.by_name(file->name());
  72. CHECK(same_f.get() != f.get());
  73. CHECK(same_f->get_key() == f->get_key());
  74. }
  75. }
  76. SECTION("folder does not exists") {
  77. pr_index.set_folder(db_folder_1.id() + "xxx");
  78. auto opt = diff::peer::update_folder_t::create(*cluster, *peer_device, pr_index);
  79. REQUIRE(!opt);
  80. CHECK(opt.error() == model::make_error_code(model::error_code_t::folder_does_not_exist));
  81. }
  82. SECTION("folder is not shared") {
  83. pr_index.set_folder(db_folder_2.id());
  84. auto opt = diff::peer::update_folder_t::create(*cluster, *peer_device, pr_index);
  85. REQUIRE(!opt);
  86. CHECK(opt.error() == model::make_error_code(model::error_code_t::folder_is_not_shared));
  87. }
  88. #if 0
  89. SECTION("exceed max sequence") {
  90. pr_index.set_folder(db_folder_1.id());
  91. auto f = pr_index.add_files();
  92. f->set_sequence(999);
  93. auto opt = diff::peer::update_folder_t::create(*cluster, *peer_device, pr_index);
  94. REQUIRE(!opt);
  95. CHECK(opt.error() == model::make_error_code(model::error_code_t::exceed_max_sequence));
  96. }
  97. #endif
  98. SECTION("blocks are not expected") {
  99. auto file = pr_index.add_files();
  100. file->set_name("a.txt");
  101. file->set_sequence(10ul);
  102. file->set_size(5ul);
  103. file->set_block_size(5ul);
  104. file->set_deleted(true);
  105. auto b = file->add_blocks();
  106. b->set_hash("123");
  107. auto opt = diff::peer::update_folder_t::create(*cluster, *peer_device, pr_index);
  108. REQUIRE(!opt);
  109. CHECK(opt.error() == model::make_error_code(model::error_code_t::unexpected_blocks));
  110. }
  111. }