[EAV] Value::setAttribute add return this

ftwbzhao · ftwbzhao · commit 115dd3bdb327 · 2015-11-16T13:52:07.000+08:00
diff --git a/More/EAV/Tests/ValueTest.php b/More/EAV/Tests/ValueTest.php
@@ -40,7 +40,6 @@ public function testSetGetAttribute()
         $value->setName('Silver');
         $this->assertSame($attribute, $value->getAttribute());
 
-        $value->setAttribute($attribute);
-        $this->assertSame($attribute, $value->getAttribute());
+        $this->assertSame($attribute, $value->setAttribute($attribute)->getAttribute());
     }
 }
diff --git a/More/EAV/Value.php b/More/EAV/Value.php
@@ -28,12 +28,15 @@ public function __construct(Attribute $attribute)
 
     /**
      * @param Attribute $attribute
+     * @return $this
      */
     public function setAttribute(Attribute $attribute)
     {
         $this->attribute->removeValue($this); // Remove value from current attribute
         $attribute->addValue($this); // Add value to new attribute
         $this->attribute = $attribute;
+
+        return $this;
     }
 
     /**

Original file line number	Diff line number	Diff line change
`@@ -40,7 +40,6 @@ public function testSetGetAttribute()`
`40`	`40`	`$value->setName('Silver');`
`41`	`41`	`$this->assertSame($attribute, $value->getAttribute());`
`42`	`42`
`43`		`- $value->setAttribute($attribute);`
`44`		`- $this->assertSame($attribute, $value->getAttribute());`
	`43`	`+ $this->assertSame($attribute, $value->setAttribute($attribute)->getAttribute());`
`45`	`44`	`}`
`46`	`45`	`}`
Original file line number	Diff line number	Diff line change
`@@ -28,12 +28,15 @@ public function __construct(Attribute $attribute)`
`28`	`28`
`29`	`29`	`/**`
`30`	`30`	`* @param Attribute $attribute`
	`31`	`+ * @return $this`
`31`	`32`	`*/`
`32`	`33`	`public function setAttribute(Attribute $attribute)`
`33`	`34`	`{`
`34`	`35`	`$this->attribute->removeValue($this); // Remove value from current attribute`
`35`	`36`	`$attribute->addValue($this); // Add value to new attribute`
`36`	`37`	`$this->attribute = $attribute;`
	`38`	`+`
	`39`	`+ return $this;`
`37`	`40`	`}`
`38`	`41`
`39`	`42`	`/**`