185 lines
5.7 KiB
PHP
185 lines
5.7 KiB
PHP
<?php
|
|
/**
|
|
* Copyright (c) 2015 Joas Schilling <nickvergessen@owncloud.com>
|
|
* This file is licensed under the Affero General Public License version 3 or
|
|
* later.
|
|
* See the COPYING-README file.
|
|
*/
|
|
|
|
namespace Test\Repair;
|
|
|
|
/**
|
|
* Tests for the cleaning the tags tables
|
|
*
|
|
* @see \OC\Repair\CleanTags
|
|
*/
|
|
class CleanTags extends \Test\TestCase {
|
|
|
|
/** @var \OC\RepairStep */
|
|
protected $repair;
|
|
|
|
/** @var \Doctrine\DBAL\Connection */
|
|
protected $connection;
|
|
|
|
/** @var int */
|
|
protected $createdFile;
|
|
|
|
protected function setUp() {
|
|
parent::setUp();
|
|
|
|
$this->connection = \OC::$server->getDatabaseConnection();
|
|
$this->repair = new \OC\Repair\CleanTags($this->connection);
|
|
$this->cleanUpTables();
|
|
}
|
|
|
|
protected function tearDown() {
|
|
$this->cleanUpTables();
|
|
|
|
parent::tearDown();
|
|
}
|
|
|
|
protected function cleanUpTables() {
|
|
$qb = $this->connection->createQueryBuilder();
|
|
$qb->delete('`*PREFIX*vcategory`')
|
|
->execute();
|
|
|
|
$qb->delete('`*PREFIX*vcategory_to_object`')
|
|
->execute();
|
|
|
|
$qb->delete('`*PREFIX*filecache`')
|
|
->execute();
|
|
}
|
|
|
|
public function testRun() {
|
|
$cat1 = $this->addTagCategory('TestRepairCleanTags', 'files'); // Retained
|
|
$cat2 = $this->addTagCategory('TestRepairCleanTags2', 'files'); // Deleted: Category will be empty
|
|
$this->addTagCategory('TestRepairCleanTags3', 'files'); // Deleted: Category is empty
|
|
$cat3 = $this->addTagCategory('TestRepairCleanTags', 'contacts'); // Retained
|
|
|
|
$this->addTagEntry($this->getFileID(), $cat2, 'files'); // Retained
|
|
$this->addTagEntry($this->getFileID() + 1, $cat1, 'files'); // Deleted: File is NULL
|
|
$this->addTagEntry(9999999, $cat3, 'contacts'); // Retained
|
|
$this->addTagEntry($this->getFileID(), $cat3 + 1, 'files'); // Deleted: Category is NULL
|
|
|
|
$this->assertEntryCount('*PREFIX*vcategory_to_object', 4, 'Assert tag entries count before repair step');
|
|
$this->assertEntryCount('*PREFIX*vcategory', 4, 'Assert tag categories count before repair step');
|
|
|
|
\Test_Helper::invokePrivate($this->repair, 'deleteOrphanFileEntries');
|
|
$this->assertEntryCount('*PREFIX*vcategory_to_object', 3, 'Assert tag entries count after cleaning file entries');
|
|
$this->assertEntryCount('*PREFIX*vcategory', 4, 'Assert tag categories count after cleaning file entries');
|
|
|
|
\Test_Helper::invokePrivate($this->repair, 'deleteOrphanTagEntries');
|
|
$this->assertEntryCount('*PREFIX*vcategory_to_object', 2, 'Assert tag entries count after cleaning tag entries');
|
|
$this->assertEntryCount('*PREFIX*vcategory', 4, 'Assert tag categories count after cleaning tag entries');
|
|
|
|
\Test_Helper::invokePrivate($this->repair, 'deleteOrphanCategoryEntries');
|
|
$this->assertEntryCount('*PREFIX*vcategory_to_object', 2, 'Assert tag entries count after cleaning category entries');
|
|
$this->assertEntryCount('*PREFIX*vcategory', 2, 'Assert tag categories count after cleaning category entries');
|
|
}
|
|
|
|
/**
|
|
* @param string $tableName
|
|
* @param int $expected
|
|
* @param string $message
|
|
*/
|
|
protected function assertEntryCount($tableName, $expected, $message = '') {
|
|
$qb = $this->connection->createQueryBuilder();
|
|
$result = $qb->select('COUNT(*)')
|
|
->from('`' . $tableName . '`')
|
|
->execute();
|
|
|
|
$this->assertEquals($expected, $result->fetchColumn(), $message);
|
|
}
|
|
|
|
/**
|
|
* Adds a new tag category to the database
|
|
*
|
|
* @param string $category
|
|
* @param string $type
|
|
* @return int
|
|
*/
|
|
protected function addTagCategory($category, $type) {
|
|
$qb = $this->connection->createQueryBuilder();
|
|
$qb->insert('`*PREFIX*vcategory`')
|
|
->values([
|
|
'`uid`' => $qb->createNamedParameter('TestRepairCleanTags'),
|
|
'`category`' => $qb->createNamedParameter($category),
|
|
'`type`' => $qb->createNamedParameter($type),
|
|
])
|
|
->execute();
|
|
|
|
return (int) $this->getLastInsertID('`*PREFIX*vcategory`', '`id`');
|
|
}
|
|
|
|
/**
|
|
* Adds a new tag entry to the database
|
|
* @param int $objectId
|
|
* @param int $category
|
|
* @param string $type
|
|
*/
|
|
protected function addTagEntry($objectId, $category, $type) {
|
|
$qb = $this->connection->createQueryBuilder();
|
|
$qb->insert('`*PREFIX*vcategory_to_object`')
|
|
->values([
|
|
'`objid`' => $qb->createNamedParameter($objectId, \PDO::PARAM_INT),
|
|
'`categoryid`' => $qb->createNamedParameter($category, \PDO::PARAM_INT),
|
|
'`type`' => $qb->createNamedParameter($type),
|
|
])
|
|
->execute();
|
|
}
|
|
|
|
/**
|
|
* Gets the last fileid from the file cache
|
|
* @return int
|
|
*/
|
|
protected function getFileID() {
|
|
if ($this->createdFile) {
|
|
return $this->createdFile;
|
|
}
|
|
|
|
$qb = $this->connection->createQueryBuilder();
|
|
|
|
// We create a new file entry and delete it after the test again
|
|
$fileName = $this->getUniqueID('TestRepairCleanTags', 12);
|
|
$qb->insert('`*PREFIX*filecache`')
|
|
->values([
|
|
'`path`' => $qb->createNamedParameter($fileName),
|
|
'`path_hash`' => $qb->createNamedParameter(md5($fileName)),
|
|
])
|
|
->execute();
|
|
$fileName = $this->getUniqueID('TestRepairCleanTags', 12);
|
|
$qb->insert('`*PREFIX*filecache`')
|
|
->values([
|
|
'`path`' => $qb->createNamedParameter($fileName),
|
|
'`path_hash`' => $qb->createNamedParameter(md5($fileName)),
|
|
])
|
|
->execute();
|
|
|
|
$this->createdFile = (int) $this->getLastInsertID('`*PREFIX*filecache`', '`fileid`');
|
|
return $this->createdFile;
|
|
}
|
|
|
|
/**
|
|
* @param $tableName
|
|
* @param $idName
|
|
* @return int
|
|
*/
|
|
protected function getLastInsertID($tableName, $idName) {
|
|
$id = $this->connection->lastInsertId();
|
|
if ($id) {
|
|
return $id;
|
|
}
|
|
|
|
// FIXME !!!! ORACLE WORKAROUND DO NOT COPY
|
|
// FIXME INSTEAD HELP FIXING DOCTRINE
|
|
// FIXME https://github.com/owncloud/core/issues/13303
|
|
// FIXME ALSO FIX https://github.com/owncloud/core/commit/2dd85ec984c12d3be401518f22c90d2327bec07a
|
|
$qb = $this->connection->createQueryBuilder();
|
|
$result = $qb->select("MAX($idName)")
|
|
->from($tableName)
|
|
->execute();
|
|
|
|
return (int) $result->fetchColumn();
|
|
}
|
|
}
|