123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131 |
- <?php
- namespace Test\Repair;
- use OC\Repair\OldGroupMembershipShares;
- use OCP\IDBConnection;
- use OCP\IGroupManager;
- use OCP\Migration\IOutput;
- use OCP\Share\IShare;
- use PHPUnit\Framework\MockObject\MockObject;
- class OldGroupMembershipSharesTest extends \Test\TestCase {
- private IDBConnection $connection;
- private IGroupManager&MockObject $groupManager;
- protected function setUp(): void {
- parent::setUp();
- $this->groupManager = $this->getMockBuilder(IGroupManager::class)
- ->disableOriginalConstructor()
- ->getMock();
- $this->connection = \OCP\Server::get(IDBConnection::class);
- $this->deleteAllShares();
- }
- protected function tearDown(): void {
- $this->deleteAllShares();
- parent::tearDown();
- }
- protected function deleteAllShares() {
- $qb = $this->connection->getQueryBuilder();
- $qb->delete('share')->executeStatement();
- }
- public function testRun(): void {
- $repair = new OldGroupMembershipShares(
- $this->connection,
- $this->groupManager
- );
- $this->groupManager->expects($this->exactly(2))
- ->method('isInGroup')
- ->willReturnMap([
- ['member', 'group', true],
- ['not-a-member', 'group', false],
- ]);
- $parent = $this->createShare(IShare::TYPE_GROUP, 'group', null);
- $group2 = $this->createShare(IShare::TYPE_GROUP, 'group2', $parent);
- $user1 = $this->createShare(IShare::TYPE_USER, 'user1', $parent);
-
- $member = $this->createShare(2, 'member', $parent);
- $notAMember = $this->createShare(2, 'not-a-member', $parent);
- $query = $this->connection->getQueryBuilder();
- $result = $query->select('id')
- ->from('share')
- ->orderBy('id', 'ASC')
- ->executeQuery();
- $rows = $result->fetchAll();
- $this->assertEquals([['id' => $parent], ['id' => $group2], ['id' => $user1], ['id' => $member], ['id' => $notAMember]], $rows);
- $result->closeCursor();
-
- $outputMock = $this->getMockBuilder('\OCP\Migration\IOutput')
- ->disableOriginalConstructor()
- ->getMock();
- $repair->run($outputMock);
- $query = $this->connection->getQueryBuilder();
- $result = $query->select('id')
- ->from('share')
- ->orderBy('id', 'ASC')
- ->executeQuery();
- $rows = $result->fetchAll();
- $this->assertEquals([['id' => $parent], ['id' => $group2], ['id' => $user1], ['id' => $member]], $rows);
- $result->closeCursor();
- }
-
- protected function createShare($shareType, $shareWith, $parent) {
- $qb = $this->connection->getQueryBuilder();
- $shareValues = [
- 'share_type' => $qb->expr()->literal($shareType),
- 'share_with' => $qb->expr()->literal($shareWith),
- 'uid_owner' => $qb->expr()->literal('user1'),
- 'item_type' => $qb->expr()->literal('folder'),
- 'item_source' => $qb->expr()->literal(123),
- 'item_target' => $qb->expr()->literal('/123'),
- 'file_source' => $qb->expr()->literal(123),
- 'file_target' => $qb->expr()->literal('/test'),
- 'permissions' => $qb->expr()->literal(1),
- 'stime' => $qb->expr()->literal(time()),
- 'expiration' => $qb->expr()->literal('2015-09-25 00:00:00'),
- ];
- if ($parent) {
- $shareValues['parent'] = $qb->expr()->literal($parent);
- }
- $qb = $this->connection->getQueryBuilder();
- $qb->insert('share')
- ->values($shareValues)
- ->executeStatement();
- return $qb->getLastInsertId();
- }
- }
|