306 lines
9.2 KiB
PHP
306 lines
9.2 KiB
PHP
<?php
|
|
|
|
namespace Appwrite\Tests;
|
|
|
|
use Appwrite\Auth\Auth;
|
|
use Utopia\Database\Document;
|
|
use Appwrite\Messaging\Adapter\Realtime;
|
|
use PHPUnit\Framework\TestCase;
|
|
|
|
class MessagingChannelsTest extends TestCase
|
|
{
|
|
/**
|
|
* Configures how many Connections the Test should Mock.
|
|
*/
|
|
public $connectionsPerChannel = 10;
|
|
|
|
public Realtime $realtime;
|
|
public $connectionsCount = 0;
|
|
public $connectionsAuthenticated = 0;
|
|
public $connectionsGuest = 0;
|
|
public $connectionsTotal = 0;
|
|
public $allChannels = [
|
|
'files',
|
|
'files.1',
|
|
'collections',
|
|
'collections.1',
|
|
'collections.1.documents',
|
|
'documents',
|
|
'documents.1',
|
|
'executions',
|
|
'executions.1',
|
|
'functions.1',
|
|
];
|
|
|
|
public function setUp(): void
|
|
{
|
|
/**
|
|
* Setup global Counts
|
|
*/
|
|
$this->connectionsAuthenticated = count($this->allChannels) * $this->connectionsPerChannel;
|
|
$this->connectionsGuest = count($this->allChannels) * $this->connectionsPerChannel;
|
|
$this->connectionsTotal = $this->connectionsAuthenticated + $this->connectionsGuest;
|
|
|
|
$this->realtime = new Realtime();
|
|
|
|
/**
|
|
* Add Authenticated Clients
|
|
*/
|
|
for ($i = 0; $i < $this->connectionsPerChannel; $i++) {
|
|
foreach ($this->allChannels as $index => $channel) {
|
|
$user = new Document([
|
|
'$id' => 'user' . $this->connectionsCount,
|
|
'memberships' => [
|
|
[
|
|
'teamId' => 'team' . $i,
|
|
'roles' => [
|
|
empty($index % 2) ? 'admin' : 'member'
|
|
]
|
|
]
|
|
]
|
|
]);
|
|
|
|
$roles = Auth::getRoles($user);
|
|
|
|
$parsedChannels = Realtime::convertChannels([0 => $channel], $user->getId());
|
|
|
|
$this->realtime->subscribe(
|
|
'1',
|
|
$this->connectionsCount,
|
|
$roles,
|
|
$parsedChannels
|
|
);
|
|
|
|
$this->connectionsCount++;
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Add Guest Clients
|
|
*/
|
|
for ($i = 0; $i < $this->connectionsPerChannel; $i++) {
|
|
foreach ($this->allChannels as $index => $channel) {
|
|
$user = new Document([
|
|
'$id' => ''
|
|
]);
|
|
|
|
$roles = Auth::getRoles($user);
|
|
|
|
$parsedChannels = Realtime::convertChannels([0 => $channel], $user->getId());
|
|
|
|
$this->realtime->subscribe(
|
|
'1',
|
|
$this->connectionsCount,
|
|
$roles,
|
|
$parsedChannels
|
|
);
|
|
|
|
$this->connectionsCount++;
|
|
}
|
|
}
|
|
}
|
|
|
|
public function tearDown(): void
|
|
{
|
|
unset($this->realtime);
|
|
$this->connectionsCount = 0;
|
|
}
|
|
|
|
public function testSubscriptions()
|
|
{
|
|
/**
|
|
* Check for 1 project.
|
|
*/
|
|
$this->assertCount(1, $this->realtime->subscriptions);
|
|
|
|
/**
|
|
* Check for correct amount of subscriptions:
|
|
* - XXX users
|
|
* - XXX teams
|
|
* - XXX team roles (2 roles per team)
|
|
* - 1 role:guest
|
|
* - 1 role:member
|
|
*/
|
|
$this->assertCount(($this->connectionsAuthenticated + (3 * $this->connectionsPerChannel) + 2), $this->realtime->subscriptions['1']);
|
|
|
|
/**
|
|
* Check for connections
|
|
* - Authenticated
|
|
* - Guests
|
|
*/
|
|
$this->assertCount($this->connectionsTotal, $this->realtime->connections);
|
|
|
|
$this->realtime->unsubscribe(-1);
|
|
|
|
$this->assertCount($this->connectionsTotal, $this->realtime->connections);
|
|
$this->assertCount(($this->connectionsAuthenticated + (3 * $this->connectionsPerChannel) + 2), $this->realtime->subscriptions['1']);
|
|
|
|
for ($i = 0; $i < $this->connectionsCount; $i++) {
|
|
$this->realtime->unsubscribe($i);
|
|
|
|
$this->assertCount(($this->connectionsCount - $i - 1), $this->realtime->connections);
|
|
}
|
|
|
|
$this->assertEmpty($this->realtime->connections);
|
|
$this->assertEmpty($this->realtime->subscriptions);
|
|
}
|
|
|
|
/**
|
|
* Tests Wildcard (role:all) Permissions on every channel.
|
|
*/
|
|
public function testWildcardPermission()
|
|
{
|
|
foreach ($this->allChannels as $index => $channel) {
|
|
$event = [
|
|
'project' => '1',
|
|
'roles' => ['role:all'],
|
|
'data' => [
|
|
'channels' => [
|
|
0 => $channel,
|
|
]
|
|
]
|
|
];
|
|
|
|
$receivers = $this->realtime->getSubscribers($event);
|
|
|
|
/**
|
|
* Every Client subscribed to the Wildcard should receive this event.
|
|
*/
|
|
$this->assertCount($this->connectionsTotal / count($this->allChannels), $receivers, $channel);
|
|
|
|
foreach ($receivers as $receiver) {
|
|
/**
|
|
* Making sure the right clients receive the event.
|
|
*/
|
|
$this->assertStringEndsWith($index, $receiver);
|
|
}
|
|
}
|
|
}
|
|
|
|
public function testRolePermissions()
|
|
{
|
|
$roles = ['role:guest', 'role:member'];
|
|
foreach ($this->allChannels as $index => $channel) {
|
|
foreach ($roles as $role) {
|
|
$permissions = [$role];
|
|
|
|
$event = [
|
|
'project' => '1',
|
|
'roles' => $permissions,
|
|
'data' => [
|
|
'channels' => [
|
|
0 => $channel,
|
|
]
|
|
]
|
|
];
|
|
|
|
$receivers = $this->realtime->getSubscribers($event);
|
|
|
|
/**
|
|
* Every Role subscribed to a Channel should receive this event.
|
|
*/
|
|
$this->assertCount($this->connectionsPerChannel, $receivers, $channel);
|
|
|
|
foreach ($receivers as $receiver) {
|
|
/**
|
|
* Making sure the right clients receive the event.
|
|
*/
|
|
$this->assertStringEndsWith($index, $receiver);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
public function testUserPermissions()
|
|
{
|
|
foreach ($this->allChannels as $index => $channel) {
|
|
$permissions = [];
|
|
for ($i = 0; $i < $this->connectionsPerChannel; $i++) {
|
|
$permissions[] = 'user:user' . (!empty($i) ? $i : '') . $index;
|
|
}
|
|
$event = [
|
|
'project' => '1',
|
|
'roles' => $permissions,
|
|
'data' => [
|
|
'channels' => [
|
|
0 => $channel,
|
|
]
|
|
]
|
|
];
|
|
|
|
$receivers = $this->realtime->getSubscribers($event);
|
|
|
|
/**
|
|
* Every Client subscribed to a Channel should receive this event.
|
|
*/
|
|
$this->assertCount($this->connectionsAuthenticated / count($this->allChannels), $receivers, $channel);
|
|
|
|
foreach ($receivers as $receiver) {
|
|
/**
|
|
* Making sure the right clients receive the event.
|
|
*/
|
|
$this->assertStringEndsWith($index, $receiver);
|
|
}
|
|
}
|
|
}
|
|
|
|
public function testTeamPermissions()
|
|
{
|
|
foreach ($this->allChannels as $index => $channel) {
|
|
$permissions = [];
|
|
|
|
for ($i = 0; $i < $this->connectionsPerChannel; $i++) {
|
|
$permissions[] = 'team:team' . $i;
|
|
}
|
|
$event = [
|
|
'project' => '1',
|
|
'roles' => $permissions,
|
|
'data' => [
|
|
'channels' => [
|
|
0 => $channel,
|
|
]
|
|
]
|
|
];
|
|
|
|
$receivers = $this->realtime->getSubscribers($event);
|
|
|
|
/**
|
|
* Every Team Member should receive this event.
|
|
*/
|
|
$this->assertCount($this->connectionsAuthenticated / count($this->allChannels), $receivers, $channel);
|
|
|
|
foreach ($receivers as $receiver) {
|
|
/**
|
|
* Making sure the right clients receive the event.
|
|
*/
|
|
$this->assertStringEndsWith($index, $receiver);
|
|
}
|
|
|
|
$permissions = ['team:team' . $index . '/' . (empty($index % 2) ? 'admin' : 'member')];
|
|
|
|
$event = [
|
|
'project' => '1',
|
|
'roles' => $permissions,
|
|
'data' => [
|
|
'channels' => [
|
|
0 => $channel,
|
|
]
|
|
]
|
|
];
|
|
|
|
$receivers = $this->realtime->getSubscribers($event);
|
|
|
|
/**
|
|
* Only 1 Team Member of a role should have access to a specific channel.
|
|
*/
|
|
$this->assertCount(1, $receivers, $channel);
|
|
|
|
foreach ($receivers as $receiver) {
|
|
/**
|
|
* Making sure the right clients receive the event.
|
|
*/
|
|
$this->assertStringEndsWith($index, $receiver);
|
|
}
|
|
}
|
|
}
|
|
}
|