mirrors/aria2
1
0
Fork 0
mirror of https://github.com/aria2/aria2.git synced 2025-04-05 21:47:37 +03:00
Code Issues Projects Releases Packages Wiki Activity

Use request URI as referer if --referer="*" is given

Browse source
This commit is contained in:
Tatsuhiro Tsujikawa 2013-01-08 21:42:42 +09:00
parent 58c047a4a8
commit 87af7c32c6
3 changed files with 24 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

15
test/FileEntryTest.cc
View file

@ -15,6 +15,7 @@ class FileEntryTest : public CppUnit::TestFixture {
CPPUNIT_TEST(testGetRequest);
CPPUNIT_TEST(testGetRequest_withoutUriReuse);
CPPUNIT_TEST(testGetRequest_withUniqueProtocol);
CPPUNIT_TEST(testGetRequest_withReferer);
CPPUNIT_TEST(testReuseUri);
CPPUNIT_TEST(testAddUri);
CPPUNIT_TEST(testAddUris);
@ -30,6 +31,7 @@ public:
void testGetRequest();
void testGetRequest_withoutUriReuse();
void testGetRequest_withUniqueProtocol();
void testGetRequest_withReferer();
void testReuseUri();
void testAddUri();
void testAddUris();
@ -184,6 +186,19 @@ void FileEntryTest::testGetRequest_withUniqueProtocol()
fileEntry->getRemainingUris()[1]);
}
void FileEntryTest::testGetRequest_withReferer()
{
SharedHandle<FileEntry> fileEntry = createFileEntry();
SharedHandle<InorderURISelector> selector(new InorderURISelector());
std::vector<std::pair<size_t, std::string> > usedHosts;
SharedHandle<Request> req =
fileEntry->getRequest(selector, true, usedHosts, "http://referer");
CPPUNIT_ASSERT_EQUAL(std::string("http://referer"), req->getReferer());
// URI is used as referer if "*" is given.
req = fileEntry->getRequest(selector, true, usedHosts, "*");
CPPUNIT_ASSERT_EQUAL(req->getUri(), req->getReferer());
}
void FileEntryTest::testReuseUri()
{
SharedHandle<InorderURISelector> selector(new InorderURISelector());