97 lines
2.6 KiB
PHP
97 lines
2.6 KiB
PHP
<?php
|
|
/**
|
|
* @author Robin Appelman <icewind@owncloud.com>
|
|
*
|
|
* @copyright Copyright (c) 2015, ownCloud, Inc.
|
|
* @license AGPL-3.0
|
|
*
|
|
* This code is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU Affero General Public License, version 3,
|
|
* as published by the Free Software Foundation.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU Affero General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU Affero General Public License, version 3,
|
|
* along with this program. If not, see <http://www.gnu.org/licenses/>
|
|
*
|
|
*/
|
|
|
|
namespace Test\Lock;
|
|
|
|
use OCP\Lock\ILockingProvider;
|
|
|
|
class DBLockingProvider extends LockingProvider {
|
|
/**
|
|
* @var \OC\Lock\DBLockingProvider
|
|
*/
|
|
protected $instance;
|
|
|
|
/**
|
|
* @var \OCP\IDBConnection
|
|
*/
|
|
private $connection;
|
|
|
|
/**
|
|
* @var \OCP\AppFramework\Utility\ITimeFactory
|
|
*/
|
|
private $timeFactory;
|
|
|
|
private $currentTime;
|
|
|
|
public function setUp() {
|
|
$this->currentTime = time();
|
|
$this->timeFactory = $this->getMock('\OCP\AppFramework\Utility\ITimeFactory');
|
|
$this->timeFactory->expects($this->any())
|
|
->method('getTime')
|
|
->will($this->returnCallback(function () {
|
|
return $this->currentTime;
|
|
}));
|
|
parent::setUp();
|
|
}
|
|
|
|
/**
|
|
* @return \OCP\Lock\ILockingProvider
|
|
*/
|
|
protected function getInstance() {
|
|
$this->connection = \OC::$server->getDatabaseConnection();
|
|
return new \OC\Lock\DBLockingProvider($this->connection, \OC::$server->getLogger(), $this->timeFactory);
|
|
}
|
|
|
|
public function tearDown() {
|
|
$this->connection->executeQuery('DELETE FROM `*PREFIX*file_locks`');
|
|
parent::tearDown();
|
|
}
|
|
|
|
public function testCleanEmptyLocks() {
|
|
$this->currentTime = 100;
|
|
$this->instance->acquireLock('foo', ILockingProvider::LOCK_EXCLUSIVE);
|
|
$this->instance->acquireLock('asd', ILockingProvider::LOCK_EXCLUSIVE);
|
|
|
|
$this->currentTime = 200;
|
|
$this->instance->acquireLock('bar', ILockingProvider::LOCK_EXCLUSIVE);
|
|
$this->instance->changeLock('asd', ILockingProvider::LOCK_SHARED);
|
|
|
|
$this->currentTime = 150 + \OC\Lock\DBLockingProvider::TTL;
|
|
|
|
$this->assertEquals(3, $this->getLockEntryCount());
|
|
|
|
$this->instance->cleanEmptyLocks();
|
|
|
|
$this->assertEquals(3, $this->getLockEntryCount());
|
|
|
|
$this->instance->releaseAll();
|
|
|
|
$this->instance->cleanEmptyLocks();
|
|
|
|
$this->assertEquals(2, $this->getLockEntryCount());
|
|
}
|
|
|
|
private function getLockEntryCount() {
|
|
$query = $this->connection->prepare('SELECT count(*) FROM `*PREFIX*file_locks`');
|
|
$query->execute();
|
|
return $query->fetchColumn();
|
|
}
|
|
}
|