implemented BranchRestrictions::update (API 2.0)

Author: vimishor

docs/repositories/branch-restrictions.md

@@ -33,6 +33,17 @@ $restrictions->create($account_name, $repo_slug, array(
 $restrictions->get($account_name, $repo_slug, $restrictionID);
 ```
 
+### Update a specific restriction: (API 2.0)
+```php
+$restrictions->update($account_name, $repo_slug, $restrictionID, array(
+    'users' => array(
+        array('username' => 'joe'),
+        array('username' => 'mary'),
+        array('username' => 'joe-work')
+    )
+));
+```
+
 ----
 
 #### Related:

lib/Bitbucket/API/Repositories/BranchRestrictions.php

@@ -89,4 +89,34 @@ public function get($account, $repo, $id)
             sprintf('repositories/%s/%s/branch-restrictions/%d', $account, $repo, $id)
         );
     }
+
+    /**
+     * Updates a specific branch restriction.
+     *
+     * @access public
+     * @param  string           $account The team or individual account owning the repository.
+     * @param  string           $repo    The repository identifier.
+     * @param  int              $id      The restriction's identifier.
+     * @param  array            $params  Additional parameters
+     * @return MessageInterface
+     *
+     * @throws \InvalidArgumentException
+     */
+    public function update($account, $repo, $id, $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['kind'])) {
+            throw new \InvalidArgumentException('You cannot change the "kind" value.');
+        }
+
+        return $this->getClient()->setApiVersion('2.0')->put(
+            sprintf('repositories/%s/%s/branch-restrictions/%d', $account, $repo, $id),
+            json_encode($params),
+            array('Content-Type' => 'application/json')
+        );
+    }
 }

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

@@ -106,4 +106,60 @@ public function testGetSpecificRestriction()
 
         $this->assertEquals($expectedResult, $actual);
     }
+
+    public function testUpdateRestrictionFromArray()
+    {
+        $endpoint       = 'repositories/gentle/eof/branch-restrictions/1';
+        $params         = array(
+            'users' => array(
+                array('username' => 'vimishor'),
+                array('username' => 'gtl_test1')
+            )
+        );
+
+        $client = $this->getHttpClientMock();
+        $client->expects($this->once())
+            ->method('put')
+            ->with($endpoint, json_encode($params));
+
+        /** @var \Bitbucket\API\Repositories\BranchRestrictions $restriction */
+        $restriction = $this->getClassMock('Bitbucket\API\Repositories\BranchRestrictions', $client);
+
+        $restriction->update('gentle', 'eof', 1, $params);
+    }
+
+    public function testUpdateRestrictionFromJSON()
+    {
+        $endpoint       = 'repositories/gentle/eof/branch-restrictions/1';
+        $params         = json_encode(array(
+            'dummy' => array(
+                array('username' => 'vimishor'),
+                array('username' => 'gtl_test1')
+            )
+        ));
+
+        $client = $this->getHttpClientMock();
+        $client->expects($this->once())
+            ->method('put')
+            ->with($endpoint, $params);
+
+        /** @var \Bitbucket\API\Repositories\BranchRestrictions $restriction */
+        $restriction = $this->getClassMock('Bitbucket\API\Repositories\BranchRestrictions', $client);
+
+        $restriction->update('gentle', 'eof', 1, $params);
+    }
+
+    /**
+     * @expectedException \InvalidArgumentException
+     */
+    public function testCreateRestrictionShouldFailIfKindIsSpecified()
+    {
+        $params         = array(
+                'kind' => 'invalid'
+            );
+
+        $restrictions   = new \Bitbucket\API\Repositories\BranchRestrictions;
+
+        $restrictions->update('gentle', 'eof', 1, $params);
+    }
 }