IBX-1310: Created slots for assigning/unassigning user to groups (#3125)

ezsystems · Dec 10, 2021 · da90d00 · da90d00
1 parent 17a60f8
commit da90d00
Show file tree

Hide file tree

Showing 3 changed files with 126 additions and 0 deletions.
diff --git a/eZ/Publish/API/Repository/Tests/UserServiceTest.php b/eZ/Publish/API/Repository/Tests/UserServiceTest.php
@@ -2101,6 +2101,39 @@ public function testAssignUserToUserGroup()
         );
     }
 
+    /**
+     * @covers \eZ\Publish\API\Repository\UserService::assignUserToUserGroup
+     */
+    public function testAssignUserToGroupWithLocationsValidation(): void
+    {
+        $repository = $this->getRepository();
+        $userService = $repository->getUserService();
+        $locationService = $repository->getLocationService();
+
+        $administratorGroupId = $this->generateId('group', 12);
+
+        $user = $this->createUserVersion1();
+
+        $group = $userService->loadUserGroup($administratorGroupId);
+        $groupLocation = $locationService->loadLocation($group->contentInfo->mainLocationId);
+
+        // Count number of child locations before assigning user to group
+        $count = $locationService->getLocationChildCount($groupLocation);
+        $expectedCount = $count + 1;
+
+        $userService->assignUserToUserGroup(
+            $user,
+            $group
+        );
+
+        $this->refreshSearch($repository);
+
+        // Count number of child locations after assigning user to group
+        $actualCount = $locationService->getLocationChildCount($groupLocation);
+
+        self::assertEquals($expectedCount, $actualCount);
+    }
+
     /**
      * Test for the assignUserToUserGroup() method.
      *
@@ -2224,6 +2257,48 @@ public function testUnAssignUserFromUserGroupThrowsBadStateArgumentException()
         /* END: Use Case */
     }
 
+    /**
+     * @covers \eZ\Publish\API\Repository\UserService::unAssignUserFromUserGroup
+     */
+    public function testUnAssignUserToGroupWithLocationValidation(): void
+    {
+        $repository = $this->getRepository();
+        $userService = $repository->getUserService();
+        $locationService = $repository->getLocationService();
+
+        $editorsGroupId = $this->generateId('group', 13);
+        $anonymousGroupId = $this->generateId('group', 42);
+
+        $user = $this->createUserVersion1();
+
+        $this->refreshSearch($repository);
+
+        $group = $userService->loadUserGroup($editorsGroupId);
+        $groupLocation = $locationService->loadLocation($group->contentInfo->mainLocationId);
+
+        // Count number of child locations before unassigning user from group
+        $count = $locationService->getLocationChildCount($groupLocation);
+        $expectedCount = $count - 1;
+
+        // Assigning user to different group to avoid removing all groups from user
+        $userService->assignUserToUserGroup(
+            $user,
+            $userService->loadUserGroup($anonymousGroupId)
+        );
+
+        $userService->unAssignUserFromUserGroup(
+            $user,
+            $userService->loadUserGroup($editorsGroupId)
+        );
+
+        $this->refreshSearch($repository);
+
+        // Count number of child locations after unassigning user from group
+        $actualCount = $locationService->getLocationChildCount($groupLocation);
+
+        self::assertEquals($expectedCount, $actualCount);
+    }
+
     /**
      * Test that multi-language logic for the loadUserGroup method respects prioritized language list.
      *

diff --git a/eZ/Publish/Core/Search/Common/Slot/AssignUserToUserGroup.php b/eZ/Publish/Core/Search/Common/Slot/AssignUserToUserGroup.php
@@ -0,0 +1,25 @@
+<?php
+
+/**
+ * @copyright Copyright (C) eZ Systems AS. All rights reserved.
+ * @license For full copyright and license information view LICENSE file distributed with this source code.
+ */
+namespace eZ\Publish\Core\Search\Common\Slot;
+
+use eZ\Publish\Core\SignalSlot\Signal;
+
+/**
+ * A Search Engine slot handling AssignUserToUserGroupSignal.
+ */
+final class AssignUserToUserGroup extends AbstractSubtree
+{
+    public function receive(Signal $signal): void
+    {
+        if (!$signal instanceof Signal\UserService\AssignUserToUserGroupSignal) {
+            return;
+        }
+
+        $content = $this->persistenceHandler->contentHandler()->load($signal->userId);
+        $this->searchHandler->indexContent($content);
+    }
+}
diff --git a/eZ/Publish/Core/Search/Common/Slot/UnAssignUserFromUserGroup.php b/eZ/Publish/Core/Search/Common/Slot/UnAssignUserFromUserGroup.php
@@ -0,0 +1,26 @@
+<?php
+
+/**
+ * @copyright Copyright (C) eZ Systems AS. All rights reserved.
+ * @license For full copyright and license information view LICENSE file distributed with this source code.
+ */
+namespace eZ\Publish\Core\Search\Common\Slot;
+
+use eZ\Publish\Core\SignalSlot\Signal;
+use eZ\Publish\Core\Search\Common\Slot;
+
+/**
+ * A Search Engine slot handling UnAssignUserFromUserGroup.
+ */
+final class UnAssignUserFromUserGroup extends Slot
+{
+    public function receive(Signal $signal): void
+    {
+        if (!$signal instanceof Signal\UserService\UnAssignUserFromUserGroupSignal) {
+            return;
+        }
+
+        $content = $this->persistenceHandler->contentHandler()->load($signal->userId);
+        $this->searchHandler->indexContent($content);
+    }
+}