Added AttributeAware::removeAttribute()

Author: johnathanmdell

src/Attribute/AttributeAware.php

@@ -26,6 +26,12 @@ public function getAttribute($name, $default = null);
      */
     public function setAttribute($name, $value);
 
+    /**
+     * @param $name
+     * @return mixed
+     */
+    public function removeAttribute($name);
+
     /**
      * get a container for all attributes
      *

src/Attribute/AttributeBagContainer.php

@@ -41,6 +41,21 @@ public function setAttribute($name, $value)
         return $this;
     }
 
+    /**
+     * remove a single attribute with the given $name
+     *
+     * @param $name
+     * @return $this
+     */
+    public function removeAttribute($name)
+    {
+        if (isset($this->attributes[$name]) === true) {
+            unset($this->attributes[$name]);
+        }
+
+        return $this;
+    }
+
     /**
      * get an array of all attributes
      *

src/Attribute/AttributeBagNamespaced.php

@@ -68,6 +68,21 @@ public function setAttribute($name, $value)
         $this->bag->setAttribute($this->prefix . $name, $value);
     }
 
+    /**
+     * remove a single attribute with the given $name
+     *
+     * @param $name
+     * @return $this
+     */
+    public function removeAttribute($name)
+    {
+        if (isset($this->attributes[$name]) === true) {
+            unset($this->attributes[$name]);
+        }
+
+        return $this;
+    }
+
     /**
      * get an array of all attributes
      *

src/Attribute/AttributeBagReference.php

@@ -55,6 +55,21 @@ public function setAttribute($name, $value)
         return $this;
     }
 
+    /**
+     * remove a single attribute with the given $name
+     *
+     * @param $name
+     * @return $this
+     */
+    public function removeAttribute($name)
+    {
+        if (isset($this->attributes[$name]) === true) {
+            unset($this->attributes[$name]);
+        }
+
+        return $this;
+    }
+
     /**
      * get an array of all attributes
      *

src/Edge/Base.php

@@ -303,6 +303,13 @@ public function setAttribute($name, $value)
         $this->attributes[$name] = $value;
     }
 
+    public function removeAttribute($name)
+    {
+        if (isset($this->attributes[$name]) === true) {
+            unset($this->attributes[$name]);
+        }
+    }
+
     public function getAttributeBag()
     {
         return new AttributeBagReference($this->attributes);

src/Graph.php

@@ -462,6 +462,13 @@ public function setAttribute($name, $value)
         $this->attributes[$name] = $value;
     }
 
+    public function removeAttribute($name)
+    {
+        if (isset($this->attributes[$name]) === true) {
+            unset($this->attributes[$name]);
+        }
+    }
+
     public function getAttributeBag()
     {
         return new AttributeBagReference($this->attributes);

src/Vertex.php

@@ -382,6 +382,13 @@ public function setAttribute($name, $value)
         $this->attributes[$name] = $value;
     }
 
+    public function removeAttribute($name)
+    {
+        if (isset($this->attributes[$name]) === true) {
+            unset($this->attributes[$name]);
+        }
+    }
+
     public function getAttributeBag()
     {
         return new AttributeBagReference($this->attributes);

tests/Attribute/AbstractAttributeAwareTest.php

@@ -27,6 +27,19 @@ public function testAttributeSetGetDefault(AttributeAware $entity)
         $this->assertEquals('default', $entity->getAttribute('unknown', 'default'));
     }
 
+    /**
+     * @depends testAttributeAwareInterface
+     * @param AttributeAware $entity
+     */
+    public function testAttributeSetRemoveGet(AttributeAware $entity)
+    {
+        $entity->setAttribute('test', 'value');
+        $this->assertEquals('value', $entity->getAttribute('test'));
+
+        $entity->removeAttribute('test');
+        $this->assertEquals(null, $entity->getAttribute('test'));
+    }
+
     /**
      * @depends testAttributeAwareInterface
      * @param AttributeAware $entity