fix(State): add roles to state draft model

Author: Jens Schulze

src/Model/State/StateDraft.php

@@ -24,6 +24,8 @@
  * @method StateDraft setInitial(bool $initial = null)
  * @method StateReferenceCollection getTransitions()
  * @method StateDraft setTransitions(StateReferenceCollection $transitions = null)
+ * @method array getRoles()
+ * @method StateDraft setRoles(array $roles = null)
  */
 class StateDraft extends JsonObject
 {
@@ -37,6 +39,7 @@ public function fieldDefinitions()
             'name' => [static::TYPE => '\Commercetools\Core\Model\Common\LocalizedString'],
             'description' => [static::TYPE => '\Commercetools\Core\Model\Common\LocalizedString'],
             'initial' => [static::TYPE => 'bool'],
+            'roles' => [static::TYPE => 'array'],
             'transitions' => [static::TYPE => '\Commercetools\Core\Model\State\StateReferenceCollection']
         ];
     }

tests/fixtures/models.yaml

@@ -810,6 +810,7 @@ stateDraft:
     - description
     - initial
     - transitions
+    - roles
 
 location:
   domain: zone

tests/integration/State/StateUpdateRequestTest.php

@@ -253,6 +253,16 @@ public function testSetRoles()
         $this->assertInstanceOf('\Commercetools\Core\Model\State\State', $result);
     }
 
+    public function testCreateWithRoles()
+    {
+        $draft = $this->getDraft('set-roles');
+        $draft->setRoles(['ReviewIncludedInStatistics']);
+        $state = $this->createState($draft);
+
+        $this->assertInstanceOf('\Commercetools\Core\Model\State\State', $state);
+        $this->assertSame(['ReviewIncludedInStatistics'], $state->getRoles());
+    }
+
     public function testAddRemoveRoles()
     {
         $draft = $this->getDraft('add-roles');