transmission/tests/libtransmission/remove-test.cc

344 lines
17 KiB
C++

// This file Copyright (C) 2022 Mnemosyne LLC.
// It may be used under GPLv2 (SPDX: GPL-2.0-only), GPLv3 (SPDX: GPL-3.0-only),
// or any future license endorsed by Mnemosyne LLC.
// License text can be found in the licenses/ folder.
#include <array>
#include <cstdint> // uint64_t
#include <set>
#include <string>
#include <string_view>
#include <utility>
#include <libtransmission/transmission.h>
#include <libtransmission/file.h>
#include <libtransmission/torrent-files.h>
#include <libtransmission/tr-strbuf.h>
#include "gtest/gtest.h"
#include "test-fixtures.h"
using namespace std::literals;
using SubpathAndSize = std::pair<std::string_view, uint64_t>;
class RemoveTest : public libtransmission::test::SandboxedTest
{
protected:
static constexpr std::string_view Content = "Hello, World!"sv;
static constexpr std::string_view JunkBasename = ".DS_Store"sv;
static constexpr std::string_view NonJunkBasename = "passwords.txt"sv;
static void sysPathRemove(char const* filename)
{
tr_sys_path_remove(filename, nullptr);
}
static auto aliceFiles()
{
static constexpr std::array<SubpathAndSize, 106> AliceFiles = { {
{ "alice_in_wonderland_librivox/AliceInWonderland_librivox.m4b", 87525736ULL },
{ "alice_in_wonderland_librivox/Alice_in_Wonderland.jpg", 81464ULL },
{ "alice_in_wonderland_librivox/Alice_in_Wonderland.pdf", 185367ULL },
{ "alice_in_wonderland_librivox/Alice_in_Wonderland_abbyy.gz", 24582ULL },
{ "alice_in_wonderland_librivox/Alice_in_Wonderland_chocr.html.gz", 22527ULL },
{ "alice_in_wonderland_librivox/Alice_in_Wonderland_djvu.txt", 2039ULL },
{ "alice_in_wonderland_librivox/Alice_in_Wonderland_djvu.xml", 28144ULL },
{ "alice_in_wonderland_librivox/Alice_in_Wonderland_hocr.html", 56942ULL },
{ "alice_in_wonderland_librivox/Alice_in_Wonderland_hocr_pageindex.json.gz", 40ULL },
{ "alice_in_wonderland_librivox/Alice_in_Wonderland_hocr_searchtext.txt.gz", 943ULL },
{ "alice_in_wonderland_librivox/Alice_in_Wonderland_jp2.zip", 1499986ULL },
{ "alice_in_wonderland_librivox/Alice_in_Wonderland_page_numbers.json", 136ULL },
{ "alice_in_wonderland_librivox/Alice_in_Wonderland_scandata.xml", 538ULL },
{ "alice_in_wonderland_librivox/Alice_in_Wonderland_thumb.jpg", 26987ULL },
{ "alice_in_wonderland_librivox/__ia_thumb.jpg", 16557ULL },
{ "alice_in_wonderland_librivox/alice_in_wonderland_librivox.json", 13740ULL },
{ "alice_in_wonderland_librivox/alice_in_wonderland_librivox.storj-store.trigger", 0ULL },
{ "alice_in_wonderland_librivox/alice_in_wonderland_librivox_128kb.m3u", 984ULL },
{ "alice_in_wonderland_librivox/alice_in_wonderland_librivox_64kb.m3u", 1044ULL },
{ "alice_in_wonderland_librivox/alice_in_wonderland_librivox_meta.sqlite", 20480ULL },
{ "alice_in_wonderland_librivox/alice_in_wonderland_librivox_meta.xml", 2805ULL },
{ "alice_in_wonderland_librivox/wonderland_ch_01.mp3", 10249859ULL },
{ "alice_in_wonderland_librivox/wonderland_ch_01.ogg", 7509828ULL },
{ "alice_in_wonderland_librivox/wonderland_ch_01.png", 10779ULL },
{ "alice_in_wonderland_librivox/wonderland_ch_01_64kb.mp3", 5124992ULL },
{ "alice_in_wonderland_librivox/wonderland_ch_01_esshigh.json.gz", 1977ULL },
{ "alice_in_wonderland_librivox/wonderland_ch_01_esslow.json.gz", 29258ULL },
{ "alice_in_wonderland_librivox/wonderland_ch_01_spectrogram.png", 234022ULL },
{ "alice_in_wonderland_librivox/wonderland_ch_02.mp3", 11772312ULL },
{ "alice_in_wonderland_librivox/wonderland_ch_02.ogg", 5148365ULL },
{ "alice_in_wonderland_librivox/wonderland_ch_02.png", 10962ULL },
{ "alice_in_wonderland_librivox/wonderland_ch_02_64kb.mp3", 5886455ULL },
{ "alice_in_wonderland_librivox/wonderland_ch_02_esshigh.json.gz", 1980ULL },
{ "alice_in_wonderland_librivox/wonderland_ch_02_esslow.json.gz", 30287ULL },
{ "alice_in_wonderland_librivox/wonderland_ch_02_spectrogram.png", 326161ULL },
{ "alice_in_wonderland_librivox/wonderland_ch_03.mp3", 17024560ULL },
{ "alice_in_wonderland_librivox/wonderland_ch_03.ogg", 12046177ULL },
{ "alice_in_wonderland_librivox/wonderland_ch_03.png", 8725ULL },
{ "alice_in_wonderland_librivox/wonderland_ch_03_64kb.mp3", 8512448ULL },
{ "alice_in_wonderland_librivox/wonderland_ch_03_esshigh.json.gz", 1966ULL },
{ "alice_in_wonderland_librivox/wonderland_ch_03_esslow.json.gz", 34110ULL },
{ "alice_in_wonderland_librivox/wonderland_ch_03_spectrogram.png", 218264ULL },
{ "alice_in_wonderland_librivox/wonderland_ch_04.mp3", 19087768ULL },
{ "alice_in_wonderland_librivox/wonderland_ch_04.ogg", 9880920ULL },
{ "alice_in_wonderland_librivox/wonderland_ch_04.png", 6055ULL },
{ "alice_in_wonderland_librivox/wonderland_ch_04_64kb.mp3", 9544016ULL },
{ "alice_in_wonderland_librivox/wonderland_ch_04_esshigh.json.gz", 1967ULL },
{ "alice_in_wonderland_librivox/wonderland_ch_04_esslow.json.gz", 36154ULL },
{ "alice_in_wonderland_librivox/wonderland_ch_04_spectrogram.png", 282145ULL },
{ "alice_in_wonderland_librivox/wonderland_ch_05.mp3", 12946949ULL },
{ "alice_in_wonderland_librivox/wonderland_ch_05.ogg", 7470734ULL },
{ "alice_in_wonderland_librivox/wonderland_ch_05.png", 12061ULL },
{ "alice_in_wonderland_librivox/wonderland_ch_05_64kb.mp3", 6473687ULL },
{ "alice_in_wonderland_librivox/wonderland_ch_05_esshigh.json.gz", 1963ULL },
{ "alice_in_wonderland_librivox/wonderland_ch_05_esslow.json.gz", 31048ULL },
{ "alice_in_wonderland_librivox/wonderland_ch_05_spectrogram.png", 304150ULL },
{ "alice_in_wonderland_librivox/wonderland_ch_06.mp3", 12413304ULL },
{ "alice_in_wonderland_librivox/wonderland_ch_06.ogg", 7154040ULL },
{ "alice_in_wonderland_librivox/wonderland_ch_06.png", 11383ULL },
{ "alice_in_wonderland_librivox/wonderland_ch_06_64kb.mp3", 6206820ULL },
{ "alice_in_wonderland_librivox/wonderland_ch_06_esshigh.json.gz", 1942ULL },
{ "alice_in_wonderland_librivox/wonderland_ch_06_esslow.json.gz", 30295ULL },
{ "alice_in_wonderland_librivox/wonderland_ch_06_spectrogram.png", 288601ULL },
{ "alice_in_wonderland_librivox/wonderland_ch_07.mp3", 16742808ULL },
{ "alice_in_wonderland_librivox/wonderland_ch_07.ogg", 10513847ULL },
{ "alice_in_wonderland_librivox/wonderland_ch_07.png", 8180ULL },
{ "alice_in_wonderland_librivox/wonderland_ch_07_64kb.mp3", 8371136ULL },
{ "alice_in_wonderland_librivox/wonderland_ch_07_esshigh.json.gz", 1963ULL },
{ "alice_in_wonderland_librivox/wonderland_ch_07_esslow.json.gz", 33992ULL },
{ "alice_in_wonderland_librivox/wonderland_ch_07_spectrogram.png", 233725ULL },
{ "alice_in_wonderland_librivox/wonderland_ch_08.mp3", 12784781ULL },
{ "alice_in_wonderland_librivox/wonderland_ch_08.ogg", 9306961ULL },
{ "alice_in_wonderland_librivox/wonderland_ch_08.png", 11470ULL },
{ "alice_in_wonderland_librivox/wonderland_ch_08_64kb.mp3", 6392576ULL },
{ "alice_in_wonderland_librivox/wonderland_ch_08_esshigh.json.gz", 1973ULL },
{ "alice_in_wonderland_librivox/wonderland_ch_08_esslow.json.gz", 31964ULL },
{ "alice_in_wonderland_librivox/wonderland_ch_08_spectrogram.png", 233626ULL },
{ "alice_in_wonderland_librivox/wonderland_ch_09.mp3", 14528920ULL },
{ "alice_in_wonderland_librivox/wonderland_ch_09.ogg", 8062952ULL },
{ "alice_in_wonderland_librivox/wonderland_ch_09.png", 9439ULL },
{ "alice_in_wonderland_librivox/wonderland_ch_09_64kb.mp3", 7264466ULL },
{ "alice_in_wonderland_librivox/wonderland_ch_09_esshigh.json.gz", 1946ULL },
{ "alice_in_wonderland_librivox/wonderland_ch_09_esslow.json.gz", 32295ULL },
{ "alice_in_wonderland_librivox/wonderland_ch_09_spectrogram.png", 282898ULL },
{ "alice_in_wonderland_librivox/wonderland_ch_10.mp3", 21894203ULL },
{ "alice_in_wonderland_librivox/wonderland_ch_10.ogg", 15226220ULL },
{ "alice_in_wonderland_librivox/wonderland_ch_10.png", 8796ULL },
{ "alice_in_wonderland_librivox/wonderland_ch_10_64kb.mp3", 10947200ULL },
{ "alice_in_wonderland_librivox/wonderland_ch_10_esshigh.json.gz", 1970ULL },
{ "alice_in_wonderland_librivox/wonderland_ch_10_esslow.json.gz", 37062ULL },
{ "alice_in_wonderland_librivox/wonderland_ch_10_spectrogram.png", 221277ULL },
{ "alice_in_wonderland_librivox/wonderland_ch_11.mp3", 9919894ULL },
{ "alice_in_wonderland_librivox/wonderland_ch_11.ogg", 7676067ULL },
{ "alice_in_wonderland_librivox/wonderland_ch_11.png", 7140ULL },
{ "alice_in_wonderland_librivox/wonderland_ch_11_64kb.mp3", 4959296ULL },
{ "alice_in_wonderland_librivox/wonderland_ch_11_esshigh.json.gz", 1959ULL },
{ "alice_in_wonderland_librivox/wonderland_ch_11_esslow.json.gz", 28694ULL },
{ "alice_in_wonderland_librivox/wonderland_ch_11_spectrogram.png", 229471ULL },
{ "alice_in_wonderland_librivox/wonderland_ch_12.mp3", 12359368ULL },
{ "alice_in_wonderland_librivox/wonderland_ch_12.ogg", 8065179ULL },
{ "alice_in_wonderland_librivox/wonderland_ch_12.png", 11074ULL },
{ "alice_in_wonderland_librivox/wonderland_ch_12_64kb.mp3", 6179840ULL },
{ "alice_in_wonderland_librivox/wonderland_ch_12_esshigh.json.gz", 1981ULL },
{ "alice_in_wonderland_librivox/wonderland_ch_12_esslow.json.gz", 30975ULL },
{ "alice_in_wonderland_librivox/wonderland_ch_12_spectrogram.png", 235568ULL },
{ "alice_in_wonderland_librivox/history/files/alice_in_wonderland_librivox.storj-store.trigger.~1~", 0ULL },
} };
auto files = tr_torrent_files{};
for (auto const& file : AliceFiles)
{
auto const& [filename, size] = file;
files.add(filename, size);
}
return files;
}
static auto ubuntuFiles()
{
static auto constexpr Files = std::array<SubpathAndSize, 1>{ { { "ubuntu-20.04.4-desktop-amd64.iso"sv,
3379068928ULL } } };
auto files = tr_torrent_files{};
for (auto const& file : Files)
{
auto const& [filename, size] = file;
files.add(filename, size);
}
return files;
}
auto createFiles(tr_torrent_files const& files, char const* parent)
{
auto paths = std::set<std::string>{};
for (tr_file_index_t i = 0, n = files.fileCount(); i < n; ++i)
{
auto walk = tr_pathbuf{ parent, '/', files.path(i) };
createFileWithContents(walk, std::data(Content), std::size(Content));
paths.emplace(walk);
while (!tr_sys_path_is_same(parent, walk))
{
walk.popdir();
paths.emplace(walk);
}
}
return paths;
}
static auto getSubtreeContents(std::string_view parent_dir)
{
auto filenames = std::set<std::string>{};
auto file_func = [&filenames](char const* filename)
{
filenames.emplace(filename);
};
libtransmission::test::depthFirstWalk(tr_pathbuf{ parent_dir }, file_func);
return filenames;
}
};
TEST_F(RemoveTest, RemovesSingleFile)
{
auto const parent = sandboxDir();
auto expected_tree = std::set<std::string>{ parent };
EXPECT_EQ(expected_tree, getSubtreeContents(parent));
auto const files = ubuntuFiles();
expected_tree = createFiles(files, parent.c_str());
EXPECT_EQ(expected_tree, getSubtreeContents(parent));
files.remove(parent, "tmpdir_prefix"sv, sysPathRemove);
expected_tree = { parent };
EXPECT_EQ(expected_tree, getSubtreeContents(parent));
}
TEST_F(RemoveTest, RemovesSubtree)
{
auto const parent = sandboxDir();
auto expected_tree = std::set<std::string>{ parent };
EXPECT_EQ(expected_tree, getSubtreeContents(parent));
auto const files = aliceFiles();
expected_tree = createFiles(files, parent.c_str());
EXPECT_EQ(expected_tree, getSubtreeContents(parent));
files.remove(parent, "tmpdir_prefix"sv, sysPathRemove);
expected_tree = { parent };
EXPECT_EQ(expected_tree, getSubtreeContents(parent));
}
TEST_F(RemoveTest, RemovesLeftoverJunk)
{
auto const parent = sandboxDir();
auto expected_tree = std::set<std::string>{ parent };
EXPECT_EQ(expected_tree, getSubtreeContents(parent));
auto const files = aliceFiles();
expected_tree = createFiles(files, parent.c_str());
EXPECT_EQ(expected_tree, getSubtreeContents(parent));
// add a junk file *inside of* the torrent's top directory.
auto const junk_file = tr_pathbuf{ parent, "/alice_in_wonderland_librivox/"sv, JunkBasename };
createFileWithContents(junk_file, std::data(Content), std::size(Content));
expected_tree.emplace(junk_file);
EXPECT_EQ(expected_tree, getSubtreeContents(parent));
files.remove(parent, "tmpdir_prefix"sv, sysPathRemove);
expected_tree = { parent };
EXPECT_EQ(expected_tree, getSubtreeContents(parent));
}
TEST_F(RemoveTest, LeavesSiblingsAlone)
{
auto const parent = sandboxDir();
auto expected_tree = std::set<std::string>{ parent };
EXPECT_EQ(expected_tree, getSubtreeContents(parent));
auto const files = aliceFiles();
expected_tree = createFiles(files, parent.c_str());
EXPECT_GT(std::size(expected_tree), 100U);
EXPECT_EQ(expected_tree, getSubtreeContents(parent));
// add a junk file *as a sibling of* the torrent's top directory.
auto const junk_file = tr_pathbuf{ parent, '/', JunkBasename };
createFileWithContents(junk_file, std::data(Content), std::size(Content));
expected_tree.emplace(junk_file);
EXPECT_EQ(expected_tree, getSubtreeContents(parent));
// add a non-junk file *as a sibling of* the torrent's top directory.
auto const non_junk_file = tr_pathbuf{ parent, '/', NonJunkBasename };
createFileWithContents(non_junk_file, std::data(Content), std::size(Content));
expected_tree.emplace(non_junk_file);
EXPECT_EQ(expected_tree, getSubtreeContents(parent));
files.remove(parent, "tmpdir_prefix"sv, sysPathRemove);
expected_tree = { parent, junk_file.c_str(), non_junk_file.c_str() };
EXPECT_EQ(expected_tree, getSubtreeContents(parent));
}
TEST_F(RemoveTest, LeavesNonJunkAlone)
{
auto const parent = sandboxDir();
auto expected_tree = std::set<std::string>{ parent };
EXPECT_EQ(expected_tree, getSubtreeContents(parent));
auto const files = aliceFiles();
expected_tree = createFiles(files, parent.c_str());
EXPECT_EQ(expected_tree, getSubtreeContents(parent));
// add a non-junk file.
auto const nonjunk_file = tr_pathbuf{ parent, "/alice_in_wonderland_librivox/"sv, NonJunkBasename };
createFileWithContents(nonjunk_file, std::data(Content), std::size(Content));
expected_tree.emplace(nonjunk_file);
EXPECT_EQ(expected_tree, getSubtreeContents(parent));
files.remove(parent, "tmpdir_prefix"sv, sysPathRemove);
expected_tree = { parent, std::string{ tr_sys_path_dirname(nonjunk_file) }, nonjunk_file.c_str() };
EXPECT_EQ(expected_tree, getSubtreeContents(parent));
}
TEST_F(RemoveTest, PreservesDirectoryHierarchyIfPossible)
{
auto const parent = sandboxDir();
auto expected_tree = std::set<std::string>{ parent };
EXPECT_EQ(expected_tree, getSubtreeContents(parent));
// add a recycle bin
auto const recycle_bin = tr_pathbuf{ parent, "/Trash"sv };
tr_sys_dir_create(recycle_bin, TR_SYS_DIR_CREATE_PARENTS, 0777);
expected_tree.emplace(recycle_bin);
EXPECT_EQ(expected_tree, getSubtreeContents(parent));
auto const files = aliceFiles();
expected_tree = createFiles(files, parent.c_str());
expected_tree.emplace(recycle_bin);
EXPECT_EQ(expected_tree, getSubtreeContents(parent));
auto const recycle_func = [&recycle_bin](char const* filename)
{
tr_sys_path_rename(filename, tr_pathbuf{ recycle_bin, '/', tr_sys_path_basename(filename) });
};
files.remove(parent, "tmpdir_prefix"sv, recycle_func);
// after remove, the subtree should be:
expected_tree = { parent, recycle_bin.c_str() };
for (tr_file_index_t i = 0, n = files.fileCount(); i < n; ++i)
{
expected_tree.emplace(tr_pathbuf{ recycle_bin, '/', files.path(i) });
}
expected_tree.emplace(tr_pathbuf{ recycle_bin, "/alice_in_wonderland_librivox"sv });
expected_tree.emplace(tr_pathbuf{ recycle_bin, "/alice_in_wonderland_librivox/history"sv });
expected_tree.emplace(tr_pathbuf{ recycle_bin, "/alice_in_wonderland_librivox/history/files"sv });
EXPECT_EQ(expected_tree, getSubtreeContents(parent));
}