Merge pull request #138 from clue-labs/remove-attribute

Added AttributeAware::removeAttribute()

Author: clue

src/Attribute/AttributeAware.php

@@ -26,6 +26,13 @@ public function getAttribute($name, $default = null);
      */
     public function setAttribute($name, $value);
 
+    /**
+     * Removes a single attribute with the given $name
+     *
+     * @param string $name
+     */
+    public function removeAttribute($name);
+
     /**
      * get a container for all attributes
      *

src/Attribute/AttributeBag.php

@@ -9,6 +9,7 @@ interface AttributeBag extends AttributeAware
 {
     // public function getAttribute($name, $default);
     // public function setAttribute($name, $value);
+    // public function removeAttribute();
     // public function getAttributeBag();
 
     /**

src/Attribute/AttributeBagContainer.php

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

src/Attribute/AttributeBagNamespaced.php

@@ -68,6 +68,16 @@ public function setAttribute($name, $value)
         $this->bag->setAttribute($this->prefix . $name, $value);
     }
 
+    /**
+     * Removes a single attribute with the given $name
+     *
+     * @param string $name
+     */
+    public function removeAttribute($name)
+    {
+        $this->bag->removeAttribute($this->prefix . $name);
+    }
+
     /**
      * get an array of all attributes
      *

src/Attribute/AttributeBagReference.php

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

src/Edge/Base.php

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

src/Graph.php

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

src/Vertex.php

@@ -382,6 +382,11 @@ public function setAttribute($name, $value)
         $this->attributes[$name] = $value;
     }
 
+    public function removeAttribute($name)
+    {
+        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

tests/Attribute/AttributeBagContainerTest.php

@@ -2,8 +2,13 @@
 
 use Fhaculty\Graph\Attribute\AttributeBagContainer;
 
-class AttributeBagContainerTest extends TestCase
+class AttributeBagContainerTest extends AbstractAttributeAwareTest
 {
+    protected function createAttributeAware()
+    {
+        return new AttributeBagContainer();
+    }
+
     public function testEmpty()
     {
         $bag = new AttributeBagContainer();