Add findActionGroupById

lotgd · Sep 17, 2016 · 798490f · 798490f
1 parent 5b0dcb7
commit 798490f
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 0 deletions.
diff --git a/src/Models/CharacterViewpoint.php b/src/Models/CharacterViewpoint.php
@@ -106,6 +106,22 @@ public function setActionGroups(array $actionGroups)
         $this->actionGroups = $actionGroups;
     }
 
+    /**
+     * Finds an action group by id.
+     * @param $actionGroupId
+     * @return ActionGroup|null
+     */
+    public function findActionGroupById(string $actionGroupId)
+    {
+        $groups = $this->getActionGroups();
+        foreach ($groups as $g) {
+            if ($g->getId() == $actionGroupId) {
+                return $g;
+            }
+        }
+        return null;
+    }
+
     /**
      * Returns all attachments.
      * @return array

diff --git a/tests/Models/CharacterViewpointTest.php b/tests/Models/CharacterViewpointTest.php
@@ -103,6 +103,9 @@ public function testActions()
 
         $output = $em->getRepository(CharacterViewpoint::class)->find(2);
         $this->assertEquals($actionGroups, $output->getActionGroups());
+
+        $this->assertEquals($ag2, $input->findActionGroupById('id2'));
+        $this->assertNull($input->findActionGroupById('not-there'));
     }
 
     public function testAttachments()