mirror of https://github.com/nextcloud/server.git
Add Repair step to regenerate birthday calendar
Signed-off-by: Georg Ehrke <developer@georgehrke.com>pull/11832/head
Georg Ehrke
5 years ago
parent
58520209be
commit
3acde071f3
@ -0,0 +1,85 @@
|
||||
<?php
|
||||
/**
|
||||
* @copyright 2019 Georg Ehrke <oc.list@georgehrke.com>
|
||||
*
|
||||
* @author Georg Ehrke <oc.list@georgehrke.com>
|
||||
*
|
||||
* @license GNU AGPL version 3 or any later version
|
||||
*
|
||||
* This program is free software: you can redistribute it and/or modify
|
||||
* it under the terms of the GNU Affero General Public License as
|
||||
* published by the Free Software Foundation, either version 3 of the
|
||||
* License, or (at your option) any later version.
|
||||
*
|
||||
* 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
|
||||
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||
*
|
||||
*/
|
||||
namespace OCA\DAV\Migration;
|
||||
|
||||
use OCA\DAV\BackgroundJob\GenerateBirthdayCalendarBackgroundJob;
|
||||
use OCP\BackgroundJob\IJobList;
|
||||
use OCP\IConfig;
|
||||
use OCP\IUser;
|
||||
use OCP\IUserManager;
|
||||
use OCP\Migration\IOutput;
|
||||
use OCP\Migration\IRepairStep;
|
||||
|
||||
class RegenerateBirthdayCalendars implements IRepairStep {
|
||||
|
||||
/** @var IUserManager */
|
||||
private $userManager;
|
||||
|
||||
/** @var IJobList */
|
||||
private $jobList;
|
||||
|
||||
/** @var IConfig */
|
||||
private $config;
|
||||
|
||||
/**
|
||||
* @param IUserManager $userManager,
|
||||
* @param IJobList $jobList
|
||||
* @param IConfig $config
|
||||
*/
|
||||
public function __construct(IUserManager $userManager,
|
||||
IJobList $jobList,
|
||||
IConfig $config) {
|
||||
$this->userManager = $userManager;
|
||||
$this->jobList = $jobList;
|
||||
$this->config = $config;
|
||||
}
|
||||
|
||||
/**
|
||||
* @return string
|
||||
*/
|
||||
public function getName() {
|
||||
return 'Regenerating birthday calendars to use new icons and fix old birthday events without year';
|
||||
}
|
||||
|
||||
/**
|
||||
* @param IOutput $output
|
||||
*/
|
||||
public function run(IOutput $output) {
|
||||
// only run once
|
||||
if ($this->config->getAppValue('dav', 'regeneratedBirthdayCalendarsForYearFix') === 'yes') {
|
||||
$output->info('Repair step already executed');
|
||||
return;
|
||||
}
|
||||
|
||||
$output->info('Adding background jobs to regenerate birthday calendar');
|
||||
$this->userManager->callForAllUsers(function(IUser $user) {
|
||||
$this->jobList->add(GenerateBirthdayCalendarBackgroundJob::class, [
|
||||
'userId' => $user->getUID(),
|
||||
'purgeBeforeGenerating' => true
|
||||
]);
|
||||
});
|
||||
|
||||
// if all were done, no need to redo the repair during next upgrade
|
||||
$this->config->setAppValue('dav', 'regeneratedBirthdayCalendarsForYearFix', 'yes');
|
||||
}
|
||||
}
|
||||
@ -0,0 +1,137 @@
|
||||
<?php
|
||||
/**
|
||||
* @copyright Copyright (c) 2018, Georg Ehrke
|
||||
*
|
||||
* @author Georg Ehrke <oc.list@georgehrke.com>
|
||||
*
|
||||
* @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 OCA\DAV\Tests\unit\DAV\Migration;
|
||||
|
||||
use OCA\DAV\BackgroundJob\GenerateBirthdayCalendarBackgroundJob;
|
||||
use OCA\DAV\Migration\RefreshWebcalJobRegistrar;
|
||||
use OCA\DAV\Migration\RegenerateBirthdayCalendars;
|
||||
use OCP\BackgroundJob\IJobList;
|
||||
use OCP\IConfig;
|
||||
use OCP\IUser;
|
||||
use OCP\IUserManager;
|
||||
use OCP\Migration\IOutput;
|
||||
use Test\TestCase;
|
||||
|
||||
class RegenerateBirthdayCalendarsTest extends TestCase {
|
||||
|
||||
/** @var IUserManager | \PHPUnit_Framework_MockObject_MockObject */
|
||||
private $userManager;
|
||||
|
||||
/** @var IJobList | \PHPUnit_Framework_MockObject_MockObject */
|
||||
private $jobList;
|
||||
|
||||
/** @var IConfig | \PHPUnit_Framework_MockObject_MockObject */
|
||||
private $config;
|
||||
|
||||
/** @var RefreshWebcalJobRegistrar */
|
||||
private $migration;
|
||||
|
||||
protected function setUp() {
|
||||
parent::setUp();
|
||||
|
||||
$this->userManager = $this->createMock(IUserManager::class);
|
||||
$this->jobList = $this->createMock(IJobList::class);
|
||||
$this->config = $this->createMock(IConfig::class);
|
||||
|
||||
$this->migration = new RegenerateBirthdayCalendars($this->userManager,
|
||||
$this->jobList, $this->config);
|
||||
}
|
||||
|
||||
public function testGetName() {
|
||||
$this->assertEquals(
|
||||
'Regenerating birthday calendars to use new icons and fix old birthday events without year',
|
||||
$this->migration->getName()
|
||||
);
|
||||
}
|
||||
|
||||
public function testRun() {
|
||||
$this->config->expects($this->at(0))
|
||||
->method('getAppValue')
|
||||
->with('dav', 'regeneratedBirthdayCalendarsForYearFix')
|
||||
->willReturn(null);
|
||||
|
||||
$output = $this->createMock(IOutput::class);
|
||||
$output->expects($this->once())
|
||||
->method('info')
|
||||
->with('Adding background jobs to regenerate birthday calendar');
|
||||
|
||||
$this->userManager->expects($this->once())
|
||||
->method('callForAllUsers')
|
||||
->will($this->returnCallback(function($closure) {
|
||||
$user1 = $this->createMock(IUser::class);
|
||||
$user1->method('getUID')->will($this->returnValue('uid1'));
|
||||
$user2 = $this->createMock(IUser::class);
|
||||
$user2->method('getUID')->will($this->returnValue('uid2'));
|
||||
$user3 = $this->createMock(IUser::class);
|
||||
$user3->method('getUID')->will($this->returnValue('uid3'));
|
||||
|
||||
$closure($user1);
|
||||
$closure($user2);
|
||||
$closure($user3);
|
||||
}));
|
||||
|
||||
$this->jobList->expects($this->at(0))
|
||||
->method('add')
|
||||
->with(GenerateBirthdayCalendarBackgroundJob::class, [
|
||||
'userId' => 'uid1',
|
||||
'purgeBeforeGenerating' => true
|
||||
]);
|
||||
$this->jobList->expects($this->at(1))
|
||||
->method('add')
|
||||
->with(GenerateBirthdayCalendarBackgroundJob::class, [
|
||||
'userId' => 'uid2',
|
||||
'purgeBeforeGenerating' => true
|
||||
]);
|
||||
$this->jobList->expects($this->at(2))
|
||||
->method('add')
|
||||
->with(GenerateBirthdayCalendarBackgroundJob::class, [
|
||||
'userId' => 'uid3',
|
||||
'purgeBeforeGenerating' => true
|
||||
]);
|
||||
|
||||
$this->config->expects($this->at(1))
|
||||
->method('setAppValue')
|
||||
->with('dav', 'regeneratedBirthdayCalendarsForYearFix', 'yes');
|
||||
|
||||
$this->migration->run($output);
|
||||
}
|
||||
|
||||
public function testRunSecondTime() {
|
||||
$this->config->expects($this->once())
|
||||
->method('getAppValue')
|
||||
->with('dav', 'regeneratedBirthdayCalendarsForYearFix')
|
||||
->willReturn('yes');
|
||||
|
||||
$output = $this->createMock(IOutput::class);
|
||||
$output->expects($this->once())
|
||||
->method('info')
|
||||
->with('Repair step already executed');
|
||||
|
||||
$this->userManager->expects($this->never())
|
||||
->method('callForAllUsers');
|
||||
|
||||
$this->migration->run($output);
|
||||
}
|
||||
|
||||
|
||||
}
|
||||
Loading…
Reference in New Issue