adding tests for the "create" method in user permission collection

1 files changed, 39 insertions, 0 deletions

diff --git a/tests/Psecio/Gatekeeper/UserPermissionCollectionTest.php b/tests/Psecio/Gatekeeper/UserPermissionCollectionTest.php
index d2a306f..35c3a20 100644
--- a/tests/Psecio/Gatekeeper/UserPermissionCollectionTest.php
+++ b/tests/Psecio/Gatekeeper/UserPermissionCollectionTest.php
@@ -21,4 +21,43 @@ class UserPermissionCollectionTest extends \Psecio\Gatekeeper\Base
         $permissions->findByUserId($userId);
         $this->assertCount(2, $permissions);
     }
+
+    /**
+     * Test the creation of a user permission relation by permission ID
+     */
+    public function testCreateUserPermissionById()
+    {
+        $return = array(
+            array('id' => 1, 'name' => 'perm5')
+        );
+
+        $ds = $this->buildMock($return, 'fetch');
+        $permissions = new UserPermissionCollection($ds);
+        $model = new UserPermissionModel($ds);
+
+        $data = array(1, 2, 3);
+        $permissions->create($model, $data);
+    }
+
+    /**
+     * Test the creation of a user permission relation by permission model instance
+     */
+    public function testCreateUserPermissionByModel()
+    {
+        $return = array(
+            array('id' => 1, 'name' => 'perm5'),
+            array('id' => 2, 'name' => 'perm6')
+        );
+
+        $ds = $this->buildMock($return, 'fetch');
+        $permissions = new UserPermissionCollection($ds);
+        $model = new UserPermissionModel($ds);
+
+        $data = array(
+            new PermissionModel($ds, array('id' => 1)),
+            new PermissionModel($ds, array('id' => 2)),
+            new PermissionModel($ds, array('id' => 3))
+        );
+        $permissions->create($model, $data);
+    }
 }
\ No newline at end of file