Files
nzbget/tests/postprocess/DupeMatcherTest.cpp
2015-11-21 00:02:22 +01:00

99 lines
3.7 KiB
C++

/*
* This file is part of nzbget
*
* Copyright (C) 2015 Andrey Prygunkov <hugbug@users.sourceforge.net>
*
* This program 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 2 of the License, or
* (at your option) any later version.
*
* This program 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 this program; if not, write to the Free Software
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
*
* $Revision$
* $Date$
*
*/
#include "nzbget.h"
#include "catch.h"
#include "Options.h"
#include "DupeMatcher.h"
#include "TestUtil.h"
TEST_CASE("Disk matcher", "[Par][DupeMatcher][Slow][TestData]")
{
Options options(NULL, NULL);
TestUtil::PrepareWorkingDir("DupeMatcher");
char errBuf[256];
// prepare directories
std::string dupe1(TestUtil::WorkingDir() + "/dupe1");
REQUIRE(Util::ForceDirectories(dupe1.c_str(), errBuf, sizeof(errBuf)));
TestUtil::CopyAllFiles(dupe1, TestUtil::TestDataDir() + "/parchecker");
std::string dupe2(TestUtil::WorkingDir() + "/dupe2");
REQUIRE(Util::ForceDirectories(dupe2.c_str(), errBuf, sizeof(errBuf)));
TestUtil::CopyAllFiles(dupe2, TestUtil::TestDataDir() + "/parchecker");
remove((dupe2 + "/testfile.nfo").c_str());
std::string rardupe1(TestUtil::TestDataDir() + "/dupematcher1");
std::string rardupe2(TestUtil::TestDataDir() + "/dupematcher2");
std::string nondupe(TestUtil::WorkingDir() + "/nondupe");
REQUIRE(Util::ForceDirectories(nondupe.c_str(), errBuf, sizeof(errBuf)));
TestUtil::CopyAllFiles(nondupe, TestUtil::TestDataDir() + "/parchecker");
remove((nondupe + "/testfile.dat").c_str());
// now test
int64 expectedSize = Util::FileSize((dupe1 + "/testfile.dat").c_str());
DupeMatcher dupe1Matcher(dupe1.c_str(), expectedSize);
CHECK(dupe1Matcher.Prepare());
CHECK(dupe1Matcher.MatchDupeContent(dupe2.c_str()));
CHECK(dupe1Matcher.MatchDupeContent(rardupe1.c_str()));
CHECK(dupe1Matcher.MatchDupeContent(rardupe2.c_str()));
CHECK_FALSE(dupe1Matcher.MatchDupeContent(nondupe.c_str()));
DupeMatcher dupe2Matcher(dupe2.c_str(), expectedSize);
CHECK(dupe2Matcher.Prepare());
CHECK(dupe2Matcher.MatchDupeContent(dupe1.c_str()));
CHECK(dupe2Matcher.MatchDupeContent(rardupe1.c_str()));
CHECK(dupe2Matcher.MatchDupeContent(rardupe2.c_str()));
CHECK_FALSE(dupe2Matcher.MatchDupeContent(nondupe.c_str()));
DupeMatcher nonDupeMatcher(nondupe.c_str(), expectedSize);
CHECK_FALSE(nonDupeMatcher.Prepare());
CHECK_FALSE(nonDupeMatcher.MatchDupeContent(dupe1.c_str()));
CHECK_FALSE(nonDupeMatcher.MatchDupeContent(dupe2.c_str()));
CHECK_FALSE(nonDupeMatcher.MatchDupeContent(rardupe1.c_str()));
CHECK_FALSE(nonDupeMatcher.MatchDupeContent(rardupe2.c_str()));
DupeMatcher rardupe1matcher(rardupe1.c_str(), expectedSize);
CHECK(rardupe1matcher.Prepare());
CHECK(rardupe1matcher.MatchDupeContent(dupe1.c_str()));
CHECK(rardupe1matcher.MatchDupeContent(dupe2.c_str()));
CHECK(rardupe1matcher.MatchDupeContent(rardupe2.c_str()));
CHECK_FALSE(rardupe1matcher.MatchDupeContent(nondupe.c_str()));
DupeMatcher rardupe2matcher(rardupe2.c_str(), expectedSize);
CHECK(rardupe2matcher.Prepare());
CHECK(rardupe2matcher.MatchDupeContent(rardupe1.c_str()));
CHECK(rardupe2matcher.MatchDupeContent(dupe1.c_str()));
CHECK(rardupe2matcher.MatchDupeContent(dupe2.c_str()));
CHECK_FALSE(rardupe2matcher.MatchDupeContent(nondupe.c_str()));
}