mirror of
https://github.com/OPSnet/Gazelle.git
synced 2026-09-26 22:13:26 -04:00
388 lines
16 KiB
PHP
388 lines
16 KiB
PHP
<?php
|
|
|
|
namespace Gazelle;
|
|
|
|
use PHPUnit\Framework\TestCase;
|
|
use GazelleUnitTest\Helper;
|
|
use Gazelle\Enum\UserAuditEvent;
|
|
use Gazelle\Enum\UserStatus;
|
|
|
|
class UserManagerTest extends TestCase {
|
|
use Pg;
|
|
|
|
protected array $userList;
|
|
protected Request $request;
|
|
|
|
public function setUp(): void {
|
|
$this->userList = [
|
|
Helper::makeUser('um1.' . randomString(10), 'userman', enable: true, clearInbox: true),
|
|
Helper::makeUser('um2.' . randomString(10), 'userman', enable: true, clearInbox: true),
|
|
Helper::makeUser('um3.' . randomString(10), 'userman', enable: true, clearInbox: true),
|
|
];
|
|
}
|
|
|
|
public function tearDown(): void {
|
|
if (isset($this->request)) {
|
|
$this->request->remove();
|
|
}
|
|
foreach ($this->userList as $user) {
|
|
$user->remove();
|
|
}
|
|
}
|
|
|
|
public function testCycleAuthKeys(): void {
|
|
$this->assertEquals(
|
|
(int)DB::DB()->scalar("
|
|
SELECT count(*) FROM users_main
|
|
"),
|
|
new Manager\User()->cycleAuthKeys(),
|
|
'uman-auth-keys'
|
|
);
|
|
}
|
|
|
|
public function testDisableUserList(): void {
|
|
$userMan = new Manager\User();
|
|
$idList = array_map(fn ($u) => $u->id, $this->userList);
|
|
|
|
$this->assertEquals(
|
|
3,
|
|
$userMan->disableUserList(
|
|
$idList,
|
|
UserAuditEvent::activity,
|
|
'phpunit mass disable',
|
|
3
|
|
),
|
|
'uman-mass-disable'
|
|
);
|
|
$this->userList[2]->flush();
|
|
$this->assertTrue($this->userList[2]->isDisabled(), 'uman-disabled-user2');
|
|
$this->assertFalse($this->userList[2]->isEnabled(), 'uman-enabled-user2');
|
|
$this->assertFalse($this->userList[2]->isUnconfirmed(), 'uman-unconfirmed-user2');
|
|
}
|
|
|
|
public function testDisplayUsername(): void {
|
|
$userMan = new Manager\User();
|
|
$this->userList[1]->privilege()->addSecondaryClass(FLS_TEAM);
|
|
$this->userList[2]->setField('Enabled', UserStatus::disabled->value)->modify();
|
|
|
|
$this->assertEquals(
|
|
"<a href=\"user.php?id={$this->userList[1]->id}\">{$this->userList[1]->username()}</a>",
|
|
$userMan->displayUsername($this->userList[1]->id, $this->userList[0]),
|
|
'uman-display-user-short'
|
|
);
|
|
$this->assertEquals(
|
|
"<a class=\"username\" href=\"user.php?id={$this->userList[1]->id}\">{$this->userList[1]->username()}</a> <span class=\"tooltip secondary_class\" title=\"First Line Support\">FLS</span> (User)",
|
|
$userMan->displayUsername($this->userList[1]->id, $this->userList[0], true),
|
|
'uman-display-user-secondary'
|
|
);
|
|
$this->assertEquals(
|
|
"<a class=\"username\" href=\"user.php?id={$this->userList[2]->id}\">{$this->userList[2]->username()}</a><a href=\"rules.php\"><img loading=\"lazy\" class=\"tooltip\" src=\""
|
|
. STATIC_SERVER
|
|
. "/common/symbols/disabled.png\" alt=\"Banned\" title=\"Disabled\" /></a> (User)",
|
|
$userMan->displayUsername($this->userList[2]->id, $this->userList[0], true),
|
|
'uman-display-user-disabled'
|
|
);
|
|
}
|
|
|
|
public function testModifyUserAttr(): void {
|
|
$userMan = new Manager\User();
|
|
$idList = array_map(fn ($u) => $u->id, $this->userList);
|
|
$this->assertFalse($this->userList[0]->hasAttr('hide-tags'), 'uman-attr-no-attr');
|
|
|
|
$this->assertEquals(3, $userMan->modifyAttr($idList, 'hide-tags', true), 'uman-attr-modify');
|
|
$this->assertTrue($this->userList[0]->flush()->hasAttr('hide-tags'), 'uman-attr-has-attr');
|
|
}
|
|
|
|
public function testModifyUserMassToken(): void {
|
|
$userMan = new Manager\User();
|
|
array_map(fn ($u) => $u->id, $this->userList);
|
|
$this->assertEquals(0, $this->userList[0]->tokenCount(), 'uman-masstoken-initial');
|
|
|
|
$userMan->addMassTokens(10, allowLeechDisabled: false);
|
|
foreach ($this->userList as $user) {
|
|
$user->flush();
|
|
}
|
|
$this->assertEquals(10, $this->userList[0]->tokenCount(), 'uman-masstoken-add-all-0');
|
|
$this->assertEquals(10, $this->userList[1]->tokenCount(), 'uman-masstoken-add-all-1');
|
|
$this->assertEquals(10, $this->userList[2]->tokenCount(), 'uman-masstoken-add-all-2');
|
|
|
|
$this->userList[0]->setField('can_leech', 0)->modify();
|
|
$userMan->addMassTokens(15, allowLeechDisabled: false);
|
|
foreach ($this->userList as $user) {
|
|
$user->flush();
|
|
}
|
|
$this->assertEquals(10, $this->userList[0]->tokenCount(), 'uman-masstoken-add-more-0');
|
|
$this->assertEquals(25, $this->userList[1]->tokenCount(), 'uman-masstoken-add-more-1');
|
|
$this->assertEquals(25, $this->userList[2]->tokenCount(), 'uman-masstoken-add-more-2');
|
|
|
|
$this->userList[1]->toggleAttr('no-fl-gifts', true);
|
|
$userMan->addMassTokens(12, allowLeechDisabled: false);
|
|
foreach ($this->userList as $user) {
|
|
$user->flush();
|
|
}
|
|
$this->assertEquals(10, $this->userList[0]->tokenCount(), 'uman-masstoken-add-fl-0');
|
|
$this->assertEquals(25, $this->userList[1]->tokenCount(), 'uman-masstoken-add-fl-1');
|
|
$this->assertEquals(37, $this->userList[2]->tokenCount(), 'uman-masstoken-add-fl-2');
|
|
|
|
$userMan->addMassTokens(11, allowLeechDisabled: true);
|
|
foreach ($this->userList as $user) {
|
|
$user->flush();
|
|
}
|
|
$this->assertEquals(21, $this->userList[0]->tokenCount(), 'uman-masstoken-add-force-0');
|
|
$this->assertEquals(25, $this->userList[1]->tokenCount(), 'uman-masstoken-add-force-1');
|
|
$this->assertEquals(48, $this->userList[2]->tokenCount(), 'uman-masstoken-add-force-2');
|
|
|
|
$this->assertEquals(
|
|
1,
|
|
$userMan->disableUserList(
|
|
[$this->userList[2]->id],
|
|
UserAuditEvent::activity,
|
|
'phpunit fltoken',
|
|
\Gazelle\Manager\User::DISABLE_MANUAL,
|
|
),
|
|
'uman-masstoken-disable'
|
|
);
|
|
$userMan->clearMassTokens(18, allowLeechDisabled: false, excludeDisabled: true);
|
|
foreach ($this->userList as $user) {
|
|
$user->flush();
|
|
}
|
|
$this->assertEquals(21, $this->userList[0]->tokenCount(), 'uman-masstoken-clear-0');
|
|
$this->assertEquals(18, $this->userList[1]->tokenCount(), 'uman-masstoken-clear-1');
|
|
$this->assertEquals(48, $this->userList[2]->tokenCount(), 'uman-masstoken-clear-2');
|
|
|
|
$userMan->clearMassTokens(17, allowLeechDisabled: true, excludeDisabled: false);
|
|
foreach ($this->userList as $user) {
|
|
$user->flush();
|
|
}
|
|
$this->assertEquals(17, $this->userList[0]->tokenCount(), 'uman-masstoken-clear2-0');
|
|
$this->assertEquals(17, $this->userList[1]->tokenCount(), 'uman-masstoken-clear2-1');
|
|
$this->assertEquals(48, $this->userList[2]->tokenCount(), 'uman-masstoken-clear2-2');
|
|
|
|
$userMan->clearMassTokens(16, allowLeechDisabled: false, excludeDisabled: false);
|
|
foreach ($this->userList as $user) {
|
|
$user->flush();
|
|
}
|
|
$this->assertEquals(17, $this->userList[0]->tokenCount(), 'uman-masstoken-clear3-0');
|
|
$this->assertEquals(16, $this->userList[1]->tokenCount(), 'uman-masstoken-clear3-1');
|
|
$this->assertEquals(48, $this->userList[2]->tokenCount(), 'uman-masstoken-clear3-2');
|
|
|
|
$userMan->clearMassTokens(15, allowLeechDisabled: true, excludeDisabled: true);
|
|
foreach ($this->userList as $user) {
|
|
$user->flush();
|
|
}
|
|
$this->assertEquals(15, $this->userList[0]->tokenCount(), 'uman-masstoken-clear4-0');
|
|
$this->assertEquals(15, $this->userList[1]->tokenCount(), 'uman-masstoken-clear4-1');
|
|
$this->assertEquals(15, $this->userList[2]->tokenCount(), 'uman-masstoken-clear4-2');
|
|
}
|
|
|
|
public function testUserRatioWatch(): void {
|
|
$userMan = new Manager\User();
|
|
$this->userList[] = Helper::makeUser('um4.' . randomString(10), 'userman', enable: true, clearInbox: true);
|
|
$user = $this->userList[0];
|
|
$this->assertFalse($user->onRatioWatch(), 'utest-personal-on-ratio-watch');
|
|
$this->assertEquals(0.0, $user->requiredRatio(), 'utest-required-ratio');
|
|
$this->assertEquals(0, $user->downloadedOnRatioWatch(), 'utest-download-ratio-watch');
|
|
$idList = array_map(fn ($u) => $u->id, $this->userList);
|
|
|
|
// put users onto ratio watch
|
|
$GiB50 = 50 * 1_105_507_304;
|
|
$db = DB::DB();
|
|
$db->prepared_query("
|
|
UPDATE users_leech_stats SET
|
|
Downloaded = ?
|
|
WHERE UserID IN (" . placeholders($idList) . ")
|
|
", $GiB50, ...$idList
|
|
);
|
|
$db->prepared_query("
|
|
INSERT INTO users_torrent_history
|
|
(UserID, NumTorrents, Date, Time)
|
|
VALUES " . placeholders($idList, '(?, 1, unix_timestamp(now()), 259200)') . "
|
|
", ...$idList
|
|
);
|
|
$result = $userMan->updateRatioRequirements();
|
|
$this->assertCount(29, $result, 'uman-ratiowatch-update');
|
|
$this->assertGreaterThanOrEqual(
|
|
count($idList),
|
|
$result['07-weight'][0],
|
|
'uman-ratiowatch-set-action',
|
|
);
|
|
$this->assertEquals(
|
|
$idList,
|
|
$userMan->ratioWatchSetList(),
|
|
'uman-ratiowatch-set-list',
|
|
);
|
|
$this->assertEquals(
|
|
count($idList),
|
|
$userMan->ratioWatchSet(),
|
|
'uman-ratiowatch-set-action',
|
|
);
|
|
foreach ($this->userList as $user) {
|
|
$user->flush();
|
|
}
|
|
|
|
$receiver = $this->userList[0]->inbox();
|
|
$pmMan = new Manager\PM($receiver->user());
|
|
$list = $receiver->messageList($pmMan, 2, 0);
|
|
$this->assertEquals(1, $receiver->messageTotal(), 'uman-ratiowatch-pm-count');
|
|
$this->assertEquals(
|
|
'You have been put on Ratio Watch',
|
|
$list[0]->subject(),
|
|
'uman-ratiowatch-pm-subject',
|
|
);
|
|
|
|
// nuke the recent pms
|
|
foreach ($this->userList as $user) {
|
|
$user->setField('Enabled', '1')->modify();
|
|
$pmMan = new Manager\PM($user);
|
|
foreach ($user->inbox()->messageList($pmMan, 1, 0) as $pm) {
|
|
$pm->remove();
|
|
}
|
|
}
|
|
|
|
// user[0] doubles down and is blocked
|
|
$db->prepared_query("
|
|
UPDATE users_info ui
|
|
INNER JOIN users_leech_stats uls USING (UserID)
|
|
SET
|
|
ui.RatioWatchDownload = ?,
|
|
uls.Downloaded = uls.Downloaded + ?
|
|
WHERE ui.UserID = ?
|
|
", $GiB50, $GiB50, $this->userList[0]->id
|
|
);
|
|
$this->userList[0]->flush();
|
|
$this->assertEquals([$this->userList[0]->id], $userMan->ratioWatchBlockList(), 'uman-ratiowatch-block-list');
|
|
$this->assertEquals(1, $userMan->ratioWatchBlock(), 'uman-ratiowatch-do-block');
|
|
$this->userList[0]->flush();
|
|
|
|
$receiver = $this->userList[0]->inbox();
|
|
$pmMan = new Manager\PM($receiver->user());
|
|
$this->assertEquals(1, $receiver->messageTotal(), 'uman-ratiowatch-pm-block-count');
|
|
$list = $receiver->messageList($pmMan, 2, 0);
|
|
$this->assertEquals('Your download privileges have been removed', $list[0]->subject(), 'uman-ratiowatch-pm-block-subject');
|
|
$this->assertFalse($this->userList[0]->canLeech(), 'uman-user0-no-canleech');
|
|
|
|
// user[1] makes good and is cleared
|
|
$db->prepared_query("
|
|
UPDATE users_leech_stats SET
|
|
Uploaded = Uploaded + ?
|
|
WHERE UserID = ?
|
|
", $GiB50, $this->userList[1]->id
|
|
);
|
|
|
|
// ratio watch ends
|
|
$db->prepared_query("
|
|
UPDATE users_info SET
|
|
RatioWatchEnds = now()
|
|
WHERE UserID IN (" . placeholders($idList) . ")
|
|
", ...$idList
|
|
);
|
|
foreach ($this->userList as $user) {
|
|
$user->flush();
|
|
}
|
|
|
|
// user[1] is cleared
|
|
$this->assertEquals([$this->userList[1]->id], $userMan->ratioWatchClearList(), 'uman-ratiowatch-clear-list');
|
|
$this->assertEquals(1, $userMan->ratioWatchClear(), 'uman-ratiowatch-do-clear');
|
|
$this->assertEquals(0, $userMan->ratioWatchClear(), 'uman-ratiowatch-reprocess-clear');
|
|
$this->userList[1]->flush();
|
|
|
|
$receiver = $this->userList[1]->inbox();
|
|
$pmMan = new Manager\PM($receiver->user());
|
|
$this->assertEquals(1, $receiver->messageTotal(), 'uman-ratiowatch-pm-clear-count');
|
|
$list = $receiver->messageList($pmMan, 2, 0);
|
|
$this->assertEquals('You have been taken off Ratio Watch', $list[0]->subject(), 'uman-ratiowatch-pm-clear-subject');
|
|
$this->assertTrue($this->userList[1]->canLeech(), 'uman-user1-canleech');
|
|
|
|
// user[3] is cleared
|
|
$this->assertEquals(
|
|
1,
|
|
$this->userList[3]->history()->resetRatioWatch(),
|
|
'uman-ratiowatch-reset'
|
|
);
|
|
|
|
// user[2] did nothing, loses download privileges
|
|
$this->assertEquals([$this->userList[2]->id], $userMan->ratioWatchEngageList(), 'uman-ratiowatch-engage-list');
|
|
$this->assertEquals(1, $userMan->ratioWatchEngage(), 'uman-ratiowatch-do-engage');
|
|
$this->assertEquals(0, $userMan->ratioWatchEngage(), 'uman-ratiowatch-reprocess-engage');
|
|
$this->userList[2]->flush();
|
|
|
|
$receiver = $this->userList[2]->inbox();
|
|
$pmMan = new Manager\PM($receiver->user());
|
|
$this->assertEquals(1, $receiver->messageTotal(), 'uman-ratiowatch-pm-engage-count');
|
|
$list = $receiver->messageList($pmMan, 2, 0);
|
|
$this->assertEquals(
|
|
'Your downloading privileges have been suspended',
|
|
$list[0]->subject(),
|
|
'uman-ratiowatch-pm-engage-subject'
|
|
);
|
|
$this->assertFalse($this->userList[2]->canLeech(), 'uman-user2-no-canleech');
|
|
}
|
|
|
|
public function testSendCustomPM(): void {
|
|
$userMan = new Manager\User();
|
|
$this->assertEquals(
|
|
2,
|
|
$userMan->sendCustomPM(
|
|
$this->userList[0],
|
|
'phpunit sendCustomPMTest',
|
|
'phpunit sendCustomPMTest %USERNAME% message',
|
|
[$this->userList[1]->id, $this->userList[2]->id],
|
|
),
|
|
'uman-send-custom-pm'
|
|
);
|
|
|
|
$receiver = $this->userList[2]->inbox();
|
|
$pmMan = new Manager\PM($receiver->user());
|
|
$this->assertEquals(1, $receiver->messageTotal(), 'uman-custom-pm-count');
|
|
$list = $receiver->messageList($pmMan, 2, 0);
|
|
$this->assertEquals(
|
|
'phpunit sendCustomPMTest',
|
|
$list[0]->subject(),
|
|
'uman-custom-pm-subject'
|
|
);
|
|
$postlist = $list[0]->postlist(10, 0);
|
|
$postId = $postlist[0]['id'];
|
|
$pm = $pmMan->findByPostId($postId);
|
|
$this->assertInstanceOf(PM::class, $pm, 'uman-found-pm');
|
|
$this->assertStringContainsString(
|
|
$this->userList[2]->username(),
|
|
(string)$pm->postBody($postId),
|
|
'uman-custom-pm-body'
|
|
);
|
|
}
|
|
|
|
public function testUserclassFlush(): void {
|
|
$userMan = new Manager\User();
|
|
$administratorId = (int)current(array_filter(
|
|
$userMan->classList(),
|
|
fn ($class) => $class['Name'] == 'Administrator'
|
|
))['ID'];
|
|
$alphaTeamId = (int)current(
|
|
array_filter($userMan->classList(),
|
|
fn ($class) => $class['Name'] == 'Alpha Team'
|
|
))['ID'];
|
|
|
|
$this->userList[0]->addClasses([$administratorId]);
|
|
$this->userList[1]->addClasses([$alphaTeamId]);
|
|
$this->assertEquals(1, $userMan->flushUserclass($administratorId), 'uman-flush-userclass-administrator');
|
|
$this->assertEquals(1, $userMan->flushUserclass($alphaTeamId), 'uman-flush-userclass-alphateam');
|
|
}
|
|
|
|
public function testUserflow(): void {
|
|
$userflow = new Manager\User()->userflow();
|
|
$this->assertGreaterThanOrEqual(0, count($userflow), 'uman-userflow-is-array');
|
|
$recent = end($userflow);
|
|
$this->assertGreaterThanOrEqual(0, count($recent), 'uman-userflow-recent-is-array');
|
|
$this->assertEquals(['Week', 'created', 'disabled'], array_keys($recent), 'userman-recent-keys');
|
|
}
|
|
|
|
public function testUserFind(): void {
|
|
$manager = new Manager\User();
|
|
$this->assertInstanceOf(
|
|
User::class,
|
|
$manager->findByEmail($this->userList[0]->email()),
|
|
'userman-find-by-email'
|
|
);
|
|
}
|
|
}
|