Fix overwriteService() for apps

Signed-off-by: Joas Schilling <coding@schilljs.com>
This commit is contained in:
Joas Schilling 2020-07-28 09:38:43 +02:00
parent 4cac0f6895
commit bab4fb98eb
No known key found for this signature in database
GPG Key ID: 7076EA9751AACDDA
2 changed files with 17 additions and 5 deletions

View File

@ -156,7 +156,12 @@ class ServerContainer extends SimpleContainer {
return parent::query($name, $autoload);
}
private function getAppContainerForService(string $id): ?DIContainer {
/**
* @internal
* @param string $id
* @return DIContainer|null
*/
public function getAppContainerForService(string $id): ?DIContainer {
if (strpos($id, 'OCA\\') !== 0 || substr_count($id, '\\') < 2) {
return null;
}

View File

@ -51,13 +51,16 @@ abstract class TestCase extends \PHPUnit\Framework\TestCase {
* @param mixed $newService
* @return bool
*/
public function overwriteService($name, $newService) {
public function overwriteService(string $name, $newService): bool {
if (isset($this->services[$name])) {
return false;
}
$this->services[$name] = \OC::$server->query($name);
\OC::$server->registerService($name, function () use ($newService) {
$container = \OC::$server->getAppContainerForService($name);
$container = $container ?? \OC::$server;
$container->registerService($name, function () use ($newService) {
return $newService;
});
@ -68,10 +71,14 @@ abstract class TestCase extends \PHPUnit\Framework\TestCase {
* @param string $name
* @return bool
*/
public function restoreService($name) {
public function restoreService(string $name): bool {
if (isset($this->services[$name])) {
$oldService = $this->services[$name];
\OC::$server->registerService($name, function () use ($oldService) {
$container = \OC::$server->getAppContainerForService($name);
$container = $container ?? \OC::$server;
$container->registerService($name, function () use ($oldService) {
return $oldService;
});