mirror of
https://github.com/OPSnet/Gazelle.git
synced 2026-01-17 03:04:47 -05:00
392 lines
15 KiB
PHP
392 lines
15 KiB
PHP
<?php
|
||
|
||
namespace Gazelle;
|
||
|
||
use PHPUnit\Framework\TestCase;
|
||
use GazelleUnitTest\Helper;
|
||
use Gazelle\Enum\DownloadStatus;
|
||
use Gazelle\Enum\TorrentFlag;
|
||
use Gazelle\Enum\UserTorrentSearch;
|
||
use OrpheusNET\Logchecker\Logchecker;
|
||
|
||
class TorrentTest extends TestCase {
|
||
protected Torrent $torrent;
|
||
protected User $user;
|
||
protected array $userList;
|
||
|
||
public function setUp(): void {
|
||
$this->user = Helper::makeUser('torrent.' . randomString(10), 'rent', clearInbox: true);
|
||
$this->user->requestContext()->setViewer($this->user);
|
||
$this->torrent = Helper::makeTorrentMusic(
|
||
tgroup: Helper::makeTGroupMusic(
|
||
name: 'phpunit torrent ' . randomString(6),
|
||
artistName: [[ARTIST_MAIN], ['phpunit torrent ' . randomString(12)]],
|
||
tagName: ['jazz'],
|
||
user: $this->user,
|
||
),
|
||
user: $this->user,
|
||
title: randomString(10),
|
||
);
|
||
}
|
||
|
||
public function tearDown(): void {
|
||
Helper::removeTGroup($this->torrent->group(), $this->user);
|
||
$this->user->remove();
|
||
if (isset($this->userList)) {
|
||
foreach ($this->userList as $user) {
|
||
$user->remove();
|
||
}
|
||
}
|
||
}
|
||
|
||
public function testFlag(): void {
|
||
$this->assertFalse(
|
||
$this->torrent->hasFlag(TorrentFlag::badFile),
|
||
'torrent-no-bad-file-flag',
|
||
);
|
||
$this->assertEquals(
|
||
1,
|
||
$this->torrent->addFlag(TorrentFlag::badFile, $this->user),
|
||
'torrent-add-bad-file-flag',
|
||
);
|
||
$this->assertTrue(
|
||
$this->torrent->hasFlag(TorrentFlag::badFile),
|
||
'torrent-has-bad-file-flag',
|
||
);
|
||
|
||
$this->assertEquals(
|
||
1,
|
||
$this->torrent->removeFlag(TorrentFlag::badFile, $this->user),
|
||
'torrent-remove-bad-file-flag',
|
||
);
|
||
$this->assertEquals(
|
||
0,
|
||
$this->torrent->removeFlag(TorrentFlag::badFolder, $this->user),
|
||
'torrent-remove-no-flag',
|
||
);
|
||
$this->assertFalse(
|
||
$this->torrent->hasFlag(TorrentFlag::badFile),
|
||
'torrent-no-more-bad-file-flag',
|
||
);
|
||
|
||
$log = new Manager\SiteLog()->tgroupLogList($this->torrent->groupId());
|
||
$this->assertCount(3, $log, 'torrent-flag-log-count');
|
||
// reverse chronological, remove then add
|
||
$this->assertEquals(
|
||
"\"Bad Files\" flag removed from torrent {$this->torrent->id}",
|
||
$log[0]['info'],
|
||
'torrent-flag-log-add',
|
||
);
|
||
$this->assertEquals(
|
||
"\"Bad Files\" flag added to torrent {$this->torrent->id}",
|
||
$log[1]['info'],
|
||
'torrent-flag-log-add',
|
||
);
|
||
}
|
||
|
||
public function testContents(): void {
|
||
$bencoder = new \OrpheusNET\BencodeTorrent\BencodeTorrent();
|
||
$bencoder->decodeFile(__DIR__ . '/../fixture/valid_torrent.torrent');
|
||
$info = $bencoder->getData();
|
||
$this->assertIsArray($info, 'torrent-file-data-array');
|
||
$torrentFile = new File\Torrent($this->torrent->id);
|
||
$this->assertTrue(
|
||
$torrentFile->put($bencoder->getEncode()),
|
||
'torrent-file-put'
|
||
);
|
||
$this->assertTrue(
|
||
$torrentFile->exists(),
|
||
'torrent-file-exists'
|
||
);
|
||
|
||
['total_size' => $totalSize, 'files' => $fileList] = $bencoder->getFileList();
|
||
$torMan = new Manager\Torrent();
|
||
['path' => $filename, 'size' => $size] = $fileList[0];
|
||
$this->assertEquals(
|
||
".flac s2056192s 1 track1.flac ÷",
|
||
$torMan->metaFilename($filename, $size),
|
||
'torrent-file-meta-filename',
|
||
);
|
||
|
||
$dbFileList = [];
|
||
foreach ($fileList as ['path' => $filename, 'size' => $size]) {
|
||
$dbFileList[] = $torMan->metaFilename($filename, $size);
|
||
}
|
||
$this->torrent->setField('FileList', implode("\n", $dbFileList))
|
||
->modify();
|
||
|
||
$this->assertEquals(
|
||
['flac' => 2],
|
||
$this->torrent->fileListPrimaryMap(),
|
||
'torrent-file-primary-map'
|
||
);
|
||
$this->assertEquals(
|
||
2,
|
||
$this->torrent->fileListPrimaryTotal(),
|
||
'torrent-file-primary-total'
|
||
);
|
||
$this->assertEquals(
|
||
7228,
|
||
$this->torrent->fileListNonPrimarySize(),
|
||
'torrent-file-non-primary-size'
|
||
);
|
||
$this->assertEquals(
|
||
"1 track1.flac{{{2056192}}}|||2 track2.flac{{{6152192}}}|||Test Album.log{{{7228}}}",
|
||
$this->torrent->fileListLegacyAPI(),
|
||
'torrent-file-legacy-list'
|
||
);
|
||
|
||
$this->assertEquals(8215612, $totalSize, 'torrent-file-total-size');
|
||
$this->assertCount(3, $fileList, 'torrent-file-list');
|
||
$this->assertEquals(
|
||
1,
|
||
$torrentFile->remove(),
|
||
'torrent-file-remove'
|
||
);
|
||
}
|
||
|
||
public function testRemovalPm(): void {
|
||
$torrent = Helper::makeTorrentMusic(
|
||
tgroup: $this->torrent->group(),
|
||
user: $this->user,
|
||
title: randomString(10),
|
||
);
|
||
|
||
// a downloader
|
||
$this->userList['downloader'] = Helper::makeUser('torrent.' . randomString(10), 'rent', clearInbox: true);
|
||
$status = new Download($torrent, new User\UserclassRateLimit($this->userList['downloader']), false);
|
||
$this->assertEquals(DownloadStatus::ok, $status->status(), 'torrent-removal-download');
|
||
|
||
// a snatcher
|
||
$this->userList['snatcher'] = Helper::makeUser('torrent.' . randomString(10), 'rent', clearInbox: true);
|
||
Helper::generateTorrentSnatch($torrent, $this->userList['snatcher']);
|
||
|
||
// check recent snatchlist
|
||
$this->assertEquals(
|
||
[$torrent->groupId()],
|
||
$this->userList['snatcher']->snatch()->recentSnatchList(),
|
||
'utest-recent-snatch'
|
||
);
|
||
// exercise the alternate version of the SQL query
|
||
DB::DB()->prepared_query("
|
||
INSERT INTO user_summary
|
||
(user_id, snatch_total)
|
||
VALUES (?, ?)
|
||
ON DUPLICATE KEY UPDATE
|
||
snatch_total = VALUES(snatch_total)
|
||
", $this->userList['snatcher']->id, FAST_LATEST_SNATCH_THRESHOLD + 1
|
||
);
|
||
$this->userList['snatcher']->stats()->flush();
|
||
$this->assertEquals(
|
||
[$torrent->groupId()],
|
||
$this->userList['snatcher']->snatch()->flush()->recentSnatchList(),
|
||
'utest-recent-snatch'
|
||
);
|
||
|
||
// a seeder
|
||
$this->userList['seeder'] = Helper::makeUser('torrent.' . randomString(10), 'rent', clearInbox: true);
|
||
$useragent = 'uahist-' . randomString(10);
|
||
Helper::generateTorrentSeed(
|
||
$torrent,
|
||
$this->userList['seeder'],
|
||
ipAddr: '127.1.1.1',
|
||
useragent: $useragent
|
||
);
|
||
|
||
// This is as good a place as any to test this
|
||
new Stats\Users()->refreshUseragentTracker();
|
||
$this->assertEquals(
|
||
[[
|
||
"useragent" => $useragent,
|
||
"total" => 1,
|
||
]],
|
||
$this->userList['seeder']->stats()->historyUseragentTracker(),
|
||
'user-status-history-useragent-tracker',
|
||
);
|
||
|
||
$name = $torrent->fullName();
|
||
$path = $torrent->path();
|
||
$this->assertEquals(
|
||
4,
|
||
new Manager\User()->sendRemovalPm(
|
||
$this->user,
|
||
$torrent->id,
|
||
$name,
|
||
$path,
|
||
log: 'phpunit removal test',
|
||
replacementId: 0,
|
||
pmUploader: true
|
||
),
|
||
'torrent-removal-pm'
|
||
);
|
||
|
||
// uploader
|
||
$inbox = $this->user->inbox();
|
||
$this->assertEquals(1, $inbox->messageTotal(), 'torrent-removal-uploader-message-count');
|
||
$list = $inbox->messageList(new Manager\PM($this->user), 2, 0);
|
||
$pm = $list[0];
|
||
$this->assertStringStartsWith(
|
||
"Uploaded torrent deleted: $name",
|
||
$pm->subject(),
|
||
'torrent-removal-pm-subject'
|
||
);
|
||
$postList = $pm->postList(1, 0);
|
||
$this->assertStringStartsWith(
|
||
"A torrent that you uploaded has been deleted.",
|
||
$postList[0]['body'],
|
||
'torrent-removal-uploader-pm-recv-body'
|
||
);
|
||
|
||
// downloader
|
||
$inbox = $this->userList['downloader']->inbox();
|
||
$this->assertEquals(1, $inbox->messageTotal(), 'torrent-removal-downloader-message-count');
|
||
$list = $inbox->messageList(new Manager\PM($this->userList['downloader']), 2, 0);
|
||
$pm = $list[0];
|
||
$this->assertStringStartsWith(
|
||
"Downloaded torrent deleted: $name",
|
||
$pm->subject(),
|
||
'torrent-removal-downloader-pm-subject'
|
||
);
|
||
$postList = $pm->postList(1, 0);
|
||
$this->assertStringStartsWith(
|
||
"A torrent that you have downloaded has been deleted.",
|
||
$postList[0]['body'],
|
||
'torrent-removal-downloader-pm-recv-body'
|
||
);
|
||
|
||
// snatcher
|
||
$inbox = $this->userList['snatcher']->inbox();
|
||
$this->assertEquals(1, $inbox->messageTotal(), 'torrent-removal-snatcher-message-count');
|
||
$list = $inbox->messageList(new Manager\PM($this->userList['snatcher']), 2, 0);
|
||
$pm = $list[0];
|
||
$this->assertStringStartsWith(
|
||
"Snatched torrent deleted: $name",
|
||
$pm->subject(),
|
||
'torrent-removal-snatcher-pm-subject'
|
||
);
|
||
$postList = $pm->postList(1, 0);
|
||
$this->assertStringStartsWith(
|
||
"A torrent that you have snatched has been deleted.",
|
||
$postList[0]['body'],
|
||
'torrent-removal-snatcher-pm-recv-body'
|
||
);
|
||
|
||
// seeder
|
||
$inbox = $this->userList['seeder']->inbox();
|
||
$this->assertEquals(1, $inbox->messageTotal(), 'torrent-removal-seeder-message-count');
|
||
$list = $inbox->messageList(new Manager\PM($this->userList['seeder']), 2, 0);
|
||
$pm = $list[0];
|
||
$this->assertStringStartsWith(
|
||
"Seeded torrent deleted: $name",
|
||
$pm->subject(),
|
||
'torrent-removal-seeder-pm-subject'
|
||
);
|
||
$postList = $pm->postList(1, 0);
|
||
$this->assertStringStartsWith(
|
||
"A torrent that you were seeding has been deleted.",
|
||
$postList[0]['body'],
|
||
'torrent-removal-pm-seeder-recv-body'
|
||
);
|
||
}
|
||
|
||
public function testFileList(): void {
|
||
// create a list of files larger than 1 MiB (30 * 34 - 1 > 1024 ** 2)
|
||
$count = 34;
|
||
$fileList = implode(
|
||
"\n",
|
||
array_fill(0, $count, ".flac s1234s ABCDEFGH.flac " . FILELIST_DELIM)
|
||
);
|
||
$this->torrent->setField('FileList', $fileList)->modify();
|
||
$list = $this->torrent->fileList();
|
||
$this->assertCount($count, $list, 'torrent-large-filelist');
|
||
// if things do not line up then there is an off-by-one somewhere
|
||
for ($n = 0; $n < $count; $n++) {
|
||
$this->assertEquals('ABCDEFGH.flac', $list[$n]['name'], "torrent-filelist-name-$n");
|
||
$this->assertEquals('.flac', $list[$n]['ext'], "torrent-filelist-ext-$n");
|
||
$this->assertEquals(1234, $list[$n]['size'], "torrent-filelist-size-$n");
|
||
}
|
||
}
|
||
|
||
public function testRemoveAllLogs(): void {
|
||
// there are no logs but at least test the SQL
|
||
$this->assertEquals(
|
||
[],
|
||
new Torrent\Log($this->torrent->id)->logDetails(),
|
||
'torrent-log-details'
|
||
);
|
||
$this->assertEquals(
|
||
0,
|
||
$this->torrent->removeAllLogs($this->user),
|
||
'torrent-remove-all-logs'
|
||
);
|
||
}
|
||
|
||
public function testTorrentBBCode(): void {
|
||
$torrentId = $this->torrent->id;
|
||
$tgroupId = $this->torrent->group()->id;
|
||
|
||
$torrentRegexp = "^<a href=\"artist\.php\?id=\d+\" dir=\"ltr\">.*</a> – <a title=\".*?\" href=\"/torrents\.php\?id={$tgroupId}&torrentid={$torrentId}#torrent{$torrentId}\">.* \[\d+ .*?\]</a>";
|
||
$this->assertMatchesRegularExpression("@{$torrentRegexp} .*$@",
|
||
\Text::full_format("[pl]{$torrentId}[/pl]"),
|
||
'text-pl'
|
||
);
|
||
|
||
// FIXME: we generate torrent urls in two different ways
|
||
$torrentRegexp = "^<a href=\"artist\.php\?id=\d+\" dir=\"ltr\">.*</a> – <a href=\"torrents\.php\?id={$tgroupId}&torrentid={$torrentId}#torrent{$torrentId}\" dir=\"ltr\">.*</a> \[\d+ .*?\]";
|
||
$this->assertMatchesRegularExpression("@{$torrentRegexp}@",
|
||
\Text::full_format(SITE_URL . "/{$this->torrent->location()}"),
|
||
"text-torrent-url tg={$tgroupId} t={$torrentId}"
|
||
);
|
||
|
||
$tgroupRegexp = "<a href=\"artist\.php\?id=\d+\" dir=\"ltr\">.*?</a> – <a href=\"torrents.php\?id={$tgroupId}\" title=\".*?\" dir=\"ltr\">.*?</a> \[\d+ \S+\]";
|
||
$this->assertMatchesRegularExpression("@{$tgroupRegexp}@",
|
||
\Text::full_format("[torrent]{$tgroupId}[/torrent]"),
|
||
'text-tgroup-id'
|
||
);
|
||
$this->assertMatchesRegularExpression("@{$tgroupRegexp}@",
|
||
\Text::full_format(SITE_URL . "/{$this->torrent->group()->location()}"),
|
||
'text-tgroup-url'
|
||
);
|
||
}
|
||
|
||
public function testTorrentCollector(): void {
|
||
$title = 'phpunit-zip';
|
||
$collector = new Collector\TList($this->user, new Manager\Torrent(), $title, 0);
|
||
$userTorrent = new Search\UserTorrent($this->user, UserTorrentSearch::uploaded);
|
||
$this->assertCount(1, $userTorrent->idList(), 'torrent-search-upload');
|
||
$collector->setList($userTorrent->idList());
|
||
$this->assertTrue($collector->prepare([]), 'collect-tlist-prepare');
|
||
|
||
$zip = Util\Zip::make($title);
|
||
$this->assertInstanceOf(\ZipStream\ZipStream::class, $zip, 'collect-zipper');
|
||
$this->assertEquals(1, $collector->fillZip($zip), 'collect-tlist-fill');
|
||
$this->assertStringContainsString(
|
||
"Torrent groups scanned: 1\n",
|
||
$collector->summary(),
|
||
"collector-tlist-summary",
|
||
);
|
||
}
|
||
|
||
public function testLogfileHashList(): void {
|
||
try {
|
||
$logfileSummary = new LogfileSummary([
|
||
'error' => [UPLOAD_ERR_OK],
|
||
'name' => ['valid_log_eac.log'],
|
||
'tmp_name' => [__DIR__ . '/../fixture/valid_log_eac.log'],
|
||
]);
|
||
$torrentLogManager = new Manager\TorrentLog();
|
||
$checkerVersion = Logchecker::getLogcheckerVersion();
|
||
foreach ($logfileSummary->all() as $logfile) {
|
||
$torrentLog = $torrentLogManager->create($this->torrent, $logfile, $checkerVersion);
|
||
// Because RipLog::put relies on move_uploaded_file, the create method above fails to put the log file
|
||
// into place, so we do this copy afterwards.
|
||
$ripLog = new File\RipLog($torrentLog->torrentId(), $torrentLog->id());
|
||
copy(__DIR__ . '/../fixture/valid_log_eac.log', $ripLog->path());
|
||
}
|
||
$this->assertEquals([hash_file('sha256', __DIR__ . '/../fixture/valid_log_eac.log')], $this->torrent->logfileHashList());
|
||
} finally {
|
||
new File\RipLog($this->torrent->id, '*')->remove();
|
||
}
|
||
}
|
||
}
|