diff options
| -rw-r--r-- | tests/Psecio/Gatekeeper/GroupModelTest.php | 50 |
1 files changed, 50 insertions, 0 deletions
diff --git a/tests/Psecio/Gatekeeper/GroupModelTest.php b/tests/Psecio/Gatekeeper/GroupModelTest.php index c71a296..ef25e90 100644 --- a/tests/Psecio/Gatekeeper/GroupModelTest.php +++ b/tests/Psecio/Gatekeeper/GroupModelTest.php @@ -128,4 +128,54 @@ class GroupModelTest extends \Psecio\Gatekeeper\Base $this->assertFalse($group1->addChild($group2)); } + + /** + * Remove a valid child of the group by ID + */ + public function testRemoveChildByIdValid() + { + $ds = $this->getMockBuilder('\Psecio\Gatekeeper\DataSource\Stub') + ->disableOriginalConstructor() + ->setMethods(array('find', 'delete')) + ->getMock(); + + $group = new GroupModel($ds, array('id' => 1234)); + $ds->method('find')->willReturn($group); + $ds->method('delete')->willReturn(true); + + $this->assertTrue($group->removeChild(1)); + } + + /** + * Remove a valid child of the group by model instance + */ + public function testRemoveChildByModelValid() + { + $ds = $this->getMockBuilder('\Psecio\Gatekeeper\DataSource\Stub') + ->disableOriginalConstructor() + ->setMethods(array('find', 'delete')) + ->getMock(); + + $group1 = new GroupModel($ds, array('id' => 1234)); + $group2 = new GroupModel($ds, array('id' => 4321)); + + $ds->method('find')->willReturn($group1); + $ds->method('delete')->willReturn(true); + + $this->assertTrue($group1->removeChild($group2)); + } + + /** + * Test the false return of tyring to remove a child when no ID + * is set on the parent group + */ + public function testRemoveChildNoId() + { + $ds = $this->getMockBuilder('\Psecio\Gatekeeper\DataSource\Mysql') + ->disableOriginalConstructor() + ->getMock(); + + $group = new GroupModel($ds); + $this->assertFalse($group->removeChild(1)); + } } |