2021-06-30 01:11:14 +12:00
|
|
|
<?php
|
|
|
|
|
|
|
|
namespace Appwrite\Tests;
|
|
|
|
|
2021-12-17 07:12:06 +13:00
|
|
|
use Utopia\Database\Document;
|
2021-06-30 01:11:14 +12:00
|
|
|
use Appwrite\Messaging\Adapter\Realtime;
|
|
|
|
use PHPUnit\Framework\TestCase;
|
|
|
|
|
|
|
|
class MessagingTest extends TestCase
|
|
|
|
{
|
|
|
|
public function setUp(): void
|
|
|
|
{
|
|
|
|
}
|
|
|
|
|
|
|
|
public function tearDown(): void
|
|
|
|
{
|
|
|
|
}
|
|
|
|
|
|
|
|
public function testUser()
|
|
|
|
{
|
|
|
|
$realtime = new Realtime();
|
|
|
|
|
|
|
|
$realtime->subscribe(
|
|
|
|
'1',
|
|
|
|
1,
|
|
|
|
['user:123', 'role:member', 'team:abc', 'team:abc/administrator', 'team:abc/moderator', 'team:def', 'team:def/guest'],
|
|
|
|
['files' => 0, 'documents' => 0, 'documents.789' => 0, 'account.123' => 0]
|
|
|
|
);
|
|
|
|
|
|
|
|
$event = [
|
|
|
|
'project' => '1',
|
2021-10-08 04:35:17 +13:00
|
|
|
'roles' => ['role:all'],
|
2021-06-30 01:11:14 +12:00
|
|
|
'data' => [
|
|
|
|
'channels' => [
|
|
|
|
0 => 'account.123',
|
|
|
|
]
|
|
|
|
]
|
|
|
|
];
|
|
|
|
|
2021-07-14 03:18:02 +12:00
|
|
|
$receivers = $realtime->getSubscribers($event);
|
2021-06-30 01:11:14 +12:00
|
|
|
|
|
|
|
$this->assertCount(1, $receivers);
|
|
|
|
$this->assertEquals(1, $receivers[0]);
|
|
|
|
|
2021-07-14 03:18:02 +12:00
|
|
|
$event['roles'] = ['role:member'];
|
2021-06-30 01:11:14 +12:00
|
|
|
|
2021-07-14 03:18:02 +12:00
|
|
|
$receivers = $realtime->getSubscribers($event);
|
2021-06-30 01:11:14 +12:00
|
|
|
|
|
|
|
$this->assertCount(1, $receivers);
|
|
|
|
$this->assertEquals(1, $receivers[0]);
|
|
|
|
|
2021-07-14 03:18:02 +12:00
|
|
|
$event['roles'] = ['user:123'];
|
2021-06-30 01:11:14 +12:00
|
|
|
|
2021-07-14 03:18:02 +12:00
|
|
|
$receivers = $realtime->getSubscribers($event);
|
2021-06-30 01:11:14 +12:00
|
|
|
|
|
|
|
$this->assertCount(1, $receivers);
|
|
|
|
$this->assertEquals(1, $receivers[0]);
|
|
|
|
|
2021-07-14 03:18:02 +12:00
|
|
|
$event['roles'] = ['team:abc'];
|
2021-06-30 01:11:14 +12:00
|
|
|
|
2021-07-14 03:18:02 +12:00
|
|
|
$receivers = $realtime->getSubscribers($event);
|
2021-06-30 01:11:14 +12:00
|
|
|
|
|
|
|
$this->assertCount(1, $receivers);
|
|
|
|
$this->assertEquals(1, $receivers[0]);
|
|
|
|
|
2021-07-14 03:18:02 +12:00
|
|
|
$event['roles'] = ['team:abc/administrator'];
|
2021-06-30 01:11:14 +12:00
|
|
|
|
2021-07-14 03:18:02 +12:00
|
|
|
$receivers = $realtime->getSubscribers($event);
|
2021-06-30 01:11:14 +12:00
|
|
|
|
|
|
|
$this->assertCount(1, $receivers);
|
|
|
|
$this->assertEquals(1, $receivers[0]);
|
|
|
|
|
2021-07-14 03:18:02 +12:00
|
|
|
$event['roles'] = ['team:abc/moderator'];
|
2021-06-30 01:11:14 +12:00
|
|
|
|
2021-07-14 03:18:02 +12:00
|
|
|
$receivers = $realtime->getSubscribers($event);
|
2021-06-30 01:11:14 +12:00
|
|
|
|
|
|
|
$this->assertCount(1, $receivers);
|
|
|
|
$this->assertEquals(1, $receivers[0]);
|
|
|
|
|
2021-07-14 03:18:02 +12:00
|
|
|
$event['roles'] = ['team:def'];
|
2021-06-30 01:11:14 +12:00
|
|
|
|
2021-07-14 03:18:02 +12:00
|
|
|
$receivers = $realtime->getSubscribers($event);
|
2021-06-30 01:11:14 +12:00
|
|
|
|
|
|
|
$this->assertCount(1, $receivers);
|
|
|
|
$this->assertEquals(1, $receivers[0]);
|
|
|
|
|
2021-07-14 03:18:02 +12:00
|
|
|
$event['roles'] = ['team:def/guest'];
|
2021-06-30 01:11:14 +12:00
|
|
|
|
2021-07-14 03:18:02 +12:00
|
|
|
$receivers = $realtime->getSubscribers($event);
|
2021-06-30 01:11:14 +12:00
|
|
|
|
|
|
|
$this->assertCount(1, $receivers);
|
|
|
|
$this->assertEquals(1, $receivers[0]);
|
|
|
|
|
2021-07-14 03:18:02 +12:00
|
|
|
$event['roles'] = ['user:456'];
|
2021-06-30 01:11:14 +12:00
|
|
|
|
2021-07-14 03:18:02 +12:00
|
|
|
$receivers = $realtime->getSubscribers($event);
|
2021-06-30 01:11:14 +12:00
|
|
|
|
|
|
|
$this->assertEmpty($receivers);
|
|
|
|
|
2021-07-14 03:18:02 +12:00
|
|
|
$event['roles'] = ['team:def/member'];
|
2021-06-30 01:11:14 +12:00
|
|
|
|
2021-07-14 03:18:02 +12:00
|
|
|
$receivers = $realtime->getSubscribers($event);
|
2021-06-30 01:11:14 +12:00
|
|
|
|
|
|
|
$this->assertEmpty($receivers);
|
|
|
|
|
2021-10-08 04:35:17 +13:00
|
|
|
$event['roles'] = ['role:all'];
|
2021-06-30 01:11:14 +12:00
|
|
|
$event['data']['channels'] = ['documents.123'];
|
|
|
|
|
2021-07-14 03:18:02 +12:00
|
|
|
$receivers = $realtime->getSubscribers($event);
|
2021-06-30 01:11:14 +12:00
|
|
|
|
|
|
|
$this->assertEmpty($receivers);
|
|
|
|
|
|
|
|
$event['data']['channels'] = ['documents.789'];
|
|
|
|
|
2021-07-14 03:18:02 +12:00
|
|
|
$receivers = $realtime->getSubscribers($event);
|
2021-06-30 01:11:14 +12:00
|
|
|
|
|
|
|
$this->assertCount(1, $receivers);
|
|
|
|
$this->assertEquals(1, $receivers[0]);
|
|
|
|
|
|
|
|
$event['project'] = '2';
|
|
|
|
|
2021-07-14 03:18:02 +12:00
|
|
|
$receivers = $realtime->getSubscribers($event);
|
2021-06-30 01:11:14 +12:00
|
|
|
|
|
|
|
$this->assertEmpty($receivers);
|
|
|
|
|
|
|
|
$realtime->unsubscribe(2);
|
|
|
|
|
|
|
|
$this->assertCount(1, $realtime->connections);
|
|
|
|
$this->assertCount(7, $realtime->subscriptions['1']);
|
|
|
|
|
|
|
|
$realtime->unsubscribe(1);
|
|
|
|
|
|
|
|
$this->assertEmpty($realtime->connections);
|
|
|
|
$this->assertEmpty($realtime->subscriptions);
|
|
|
|
}
|
2021-06-30 23:36:58 +12:00
|
|
|
|
|
|
|
public function testConvertChannelsGuest()
|
|
|
|
{
|
|
|
|
$user = new Document([
|
|
|
|
'$id' => ''
|
|
|
|
]);
|
|
|
|
|
|
|
|
$channels = [
|
|
|
|
0 => 'files',
|
|
|
|
1 => 'documents',
|
|
|
|
2 => 'documents.789',
|
|
|
|
3 => 'account',
|
|
|
|
4 => 'account.456'
|
|
|
|
];
|
|
|
|
|
2021-07-14 03:18:02 +12:00
|
|
|
$channels = Realtime::convertChannels($channels, $user->getId());
|
2021-08-27 20:20:44 +12:00
|
|
|
$this->assertCount(4, $channels);
|
2021-06-30 23:36:58 +12:00
|
|
|
$this->assertArrayHasKey('files', $channels);
|
|
|
|
$this->assertArrayHasKey('documents', $channels);
|
|
|
|
$this->assertArrayHasKey('documents.789', $channels);
|
2021-08-27 20:20:44 +12:00
|
|
|
$this->assertArrayHasKey('account', $channels);
|
2021-06-30 23:36:58 +12:00
|
|
|
$this->assertArrayNotHasKey('account.456', $channels);
|
|
|
|
}
|
|
|
|
|
|
|
|
public function testConvertChannelsUser()
|
|
|
|
{
|
|
|
|
$user = new Document([
|
|
|
|
'$id' => '123',
|
|
|
|
'memberships' => [
|
|
|
|
[
|
|
|
|
'teamId' => 'abc',
|
|
|
|
'roles' => [
|
|
|
|
'administrator',
|
|
|
|
'moderator'
|
|
|
|
]
|
|
|
|
],
|
|
|
|
[
|
|
|
|
'teamId' => 'def',
|
|
|
|
'roles' => [
|
|
|
|
'guest'
|
|
|
|
]
|
|
|
|
]
|
|
|
|
]
|
|
|
|
]);
|
|
|
|
$channels = [
|
|
|
|
0 => 'files',
|
|
|
|
1 => 'documents',
|
|
|
|
2 => 'documents.789',
|
|
|
|
3 => 'account',
|
|
|
|
4 => 'account.456'
|
|
|
|
];
|
|
|
|
|
2021-07-14 03:18:02 +12:00
|
|
|
$channels = Realtime::convertChannels($channels, $user->getId());
|
2021-06-30 23:36:58 +12:00
|
|
|
|
2021-08-27 20:20:44 +12:00
|
|
|
$this->assertCount(5, $channels);
|
2021-06-30 23:36:58 +12:00
|
|
|
$this->assertArrayHasKey('files', $channels);
|
|
|
|
$this->assertArrayHasKey('documents', $channels);
|
|
|
|
$this->assertArrayHasKey('documents.789', $channels);
|
|
|
|
$this->assertArrayHasKey('account.123', $channels);
|
2021-08-27 20:20:44 +12:00
|
|
|
$this->assertArrayHasKey('account', $channels);
|
2021-06-30 23:36:58 +12:00
|
|
|
$this->assertArrayNotHasKey('account.456', $channels);
|
|
|
|
}
|
2021-12-17 07:12:06 +13:00
|
|
|
|
|
|
|
public function testFromPayloadCollectionLevelPermissions(): void
|
|
|
|
{
|
|
|
|
/**
|
|
|
|
* Test Collection Level Permissions
|
|
|
|
*/
|
|
|
|
$result = Realtime::fromPayload(
|
2022-04-19 04:21:45 +12:00
|
|
|
event: 'collections.collection_id.documents.document_id.create',
|
2021-12-17 07:12:06 +13:00
|
|
|
payload: new Document([
|
|
|
|
'$id' => 'test',
|
|
|
|
'$collection' => 'collection',
|
|
|
|
'$read' => ['role:admin'],
|
|
|
|
'$write' => ['role:admin']
|
|
|
|
]),
|
|
|
|
collection: new Document([
|
|
|
|
'$id' => 'collection',
|
|
|
|
'$read' => ['role:all'],
|
|
|
|
'$write' => ['role:all'],
|
|
|
|
'permission' => 'collection'
|
|
|
|
])
|
|
|
|
);
|
|
|
|
|
|
|
|
$this->assertContains('role:all', $result['roles']);
|
|
|
|
$this->assertNotContains('role:admin', $result['roles']);
|
|
|
|
|
|
|
|
/**
|
|
|
|
* Test Document Level Permissions
|
|
|
|
*/
|
|
|
|
$result = Realtime::fromPayload(
|
2022-04-19 04:21:45 +12:00
|
|
|
event: 'collections.collection_id.documents.document_id.create',
|
2021-12-17 07:12:06 +13:00
|
|
|
payload: new Document([
|
|
|
|
'$id' => 'test',
|
|
|
|
'$collection' => 'collection',
|
|
|
|
'$read' => ['role:all'],
|
|
|
|
'$write' => ['role:all']
|
|
|
|
]),
|
|
|
|
collection: new Document([
|
|
|
|
'$id' => 'collection',
|
|
|
|
'$read' => ['role:admin'],
|
|
|
|
'$write' => ['role:admin'],
|
|
|
|
'permission' => 'document'
|
|
|
|
])
|
|
|
|
);
|
|
|
|
|
|
|
|
$this->assertContains('role:all', $result['roles']);
|
|
|
|
$this->assertNotContains('role:admin', $result['roles']);
|
|
|
|
}
|
2022-04-19 04:21:45 +12:00
|
|
|
|
|
|
|
public function testFromPayloadBucketLevelPermissions(): void
|
|
|
|
{
|
|
|
|
/**
|
|
|
|
* Test Collection Level Permissions
|
|
|
|
*/
|
|
|
|
$result = Realtime::fromPayload(
|
|
|
|
event: 'buckets.bucket_id.files.file_id.create',
|
|
|
|
payload: new Document([
|
|
|
|
'$id' => 'test',
|
|
|
|
'$collection' => 'bucket',
|
|
|
|
'$read' => ['role:admin'],
|
|
|
|
'$write' => ['role:admin']
|
|
|
|
]),
|
|
|
|
bucket: new Document([
|
|
|
|
'$id' => 'bucket',
|
|
|
|
'$read' => ['role:all'],
|
|
|
|
'$write' => ['role:all'],
|
|
|
|
'permission' => 'bucket'
|
|
|
|
])
|
|
|
|
);
|
|
|
|
|
|
|
|
$this->assertContains('role:all', $result['roles']);
|
|
|
|
$this->assertNotContains('role:admin', $result['roles']);
|
|
|
|
|
|
|
|
/**
|
|
|
|
* Test Document Level Permissions
|
|
|
|
*/
|
|
|
|
$result = Realtime::fromPayload(
|
|
|
|
event: 'buckets.bucket_id.files.file_id.create',
|
|
|
|
payload: new Document([
|
|
|
|
'$id' => 'test',
|
|
|
|
'$collection' => 'bucket',
|
|
|
|
'$read' => ['role:all'],
|
|
|
|
'$write' => ['role:all']
|
|
|
|
]),
|
|
|
|
bucket: new Document([
|
|
|
|
'$id' => 'bucket',
|
|
|
|
'$read' => ['role:admin'],
|
|
|
|
'$write' => ['role:admin'],
|
|
|
|
'permission' => 'file'
|
|
|
|
])
|
|
|
|
);
|
|
|
|
|
|
|
|
$this->assertContains('role:all', $result['roles']);
|
|
|
|
$this->assertNotContains('role:admin', $result['roles']);
|
|
|
|
}
|
2021-06-30 01:11:14 +12:00
|
|
|
}
|