implemented BranchRestrictions::create (API 2.0)

Author: vimishor

docs/repositories/branch-restrictions.md

@@ -14,6 +14,20 @@ $restrictions->setCredentials( new Bitbucket\API\Authentication\Basic($bb_user,
 $restrictions->all($account_name, $repo_slug);
 ```
 
+### Creates restrictions: (API 2.0)
+
+Restrict push access to any branches starting with `joe-and-mary-` only to users `joe` and `mary`:
+```php
+$restrictions->create($account_name, $repo_slug, array(
+    'kind'      => 'push',
+    'pattern'   => 'joe-and-mary-*',
+    'users'     => array(
+        array('username' => 'joe'),
+        array('username' => 'mary')
+    )
+));
+```
+
 ----
 
 #### Related:

lib/Bitbucket/API/Repositories/BranchRestrictions.php

@@ -35,4 +35,42 @@ public function all($account, $repo)
             sprintf('repositories/%s/%s/branch-restrictions', $account, $repo)
         );
     }
+
+    /**
+     * Creates restrictions for the specified repository.
+     *
+     * @access public
+     * @param  string           $account The team or individual account owning the repository.
+     * @param  string           $repo    The repository identifier.
+     * @param  array            $params  Additional parameters
+     * @return MessageInterface
+     *
+     * @throws \InvalidArgumentException
+     */
+    public function create($account, $repo, $params = array())
+    {
+        // allow developer to directly specify params as json if (s)he wants.
+        if (!empty($params) && is_string($params)) {
+            $params = $this->decodeJSON($params);
+        }
+
+        if (!empty($params) && is_array($params)) {
+            $params = array_merge(
+                array(
+                    'kind' => 'push'
+                ),
+                $params
+            );
+        }
+
+        if (empty($params['kind']) or !in_array($params['kind'], array('push', 'delete', 'force'))) {
+            throw new \InvalidArgumentException('Invalid restriction kind.');
+        }
+
+        return $this->getClient()->setApiVersion('2.0')->post(
+            sprintf('repositories/%s/%s/branch-restrictions', $account, $repo),
+            json_encode($params),
+            array('Content-Type' => 'application/json')
+        );
+    }
 }

test/Bitbucket/Tests/API/Repositories/BranchRestrictionsTest.php

@@ -24,4 +24,68 @@ public function testGetAllRestrictions()
 
         $this->assertEquals($expectedResult, $actual);
     }
+
+    public function testCreateRestrictionFromArray()
+    {
+        $endpoint       = 'repositories/gentle/eof/branch-restrictions';
+        $params         = array(
+            'kind' => 'push'
+        );
+
+        $client = $this->getHttpClientMock();
+        $client->expects($this->once())
+            ->method('post')
+            ->with($endpoint, json_encode($params));
+
+        /** @var \Bitbucket\API\Repositories\BranchRestrictions $restrictions */
+        $restrictions   = $this->getClassMock('Bitbucket\API\Repositories\BranchRestrictions', $client);
+
+        $restrictions->create('gentle', 'eof', $params);
+    }
+
+    public function testCreateRestrictionFromJSON()
+    {
+        $endpoint       = 'repositories/gentle/eof/branch-restrictions';
+        $params         = json_encode(array(
+            'kind' => 'push'
+        ));
+
+        $client = $this->getHttpClientMock();
+        $client->expects($this->once())
+            ->method('post')
+            ->with($endpoint, $params);
+
+        /** @var \Bitbucket\API\Repositories\BranchRestrictions $restrictions */
+        $restrictions   = $this->getClassMock('Bitbucket\API\Repositories\BranchRestrictions', $client);
+
+        $restrictions->create('gentle', 'eof', $params);
+    }
+
+    /**
+     * @expectedException \InvalidArgumentException
+     */
+    public function testCreateRestrictionFromArrayShouldFailWithInvalidRestrictionKind()
+    {
+        $params         = array(
+            'kind' => 'invalid'
+        );
+
+        $restrictions   = new \Bitbucket\API\Repositories\BranchRestrictions;
+
+        $restrictions->create('gentle', 'eof', $params);
+    }
+
+    /**
+     * @expectedException \InvalidArgumentException
+     */
+    public function testCreateRestrictionFromJSONShouldFailWithInvalidRestrictionKind()
+    {
+        $params         = json_encode(array(
+            'kind' => 'invalid'
+        ));
+
+        $restrictions   = new \Bitbucket\API\Repositories\BranchRestrictions;
+
+        $restrictions->create('gentle', 'eof', $params);
+    }
 }