Signed-off-by: Georg Ehrke <developer@georgehrke.com>pull/11832/head
Georg Ehrke
6 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