Browse Source

Even more fluent eloquent model via magic setters

Now it is possible to use Eloquent's magic setters in chains. For example:

    $model->set_foo('foo')->take(10)->set_bar(some_function());
    // ...even though setters 'foo' and 'bar' are not defined on the model
Pavel Puchkin 11 years ago
parent
commit
3a46721eca
2 changed files with 15 additions and 1 deletions
  1. 1 0
      laravel/database/eloquent/model.php
  2. 14 1
      laravel/tests/cases/eloquent.test.php

+ 1 - 0
laravel/database/eloquent/model.php

@@ -770,6 +770,7 @@ abstract class Model {
 		elseif (starts_with($method, 'set_'))
 		elseif (starts_with($method, 'set_'))
 		{
 		{
 			$this->set_attribute(substr($method, 4), $parameters[0]);
 			$this->set_attribute(substr($method, 4), $parameters[0]);
+			return $this;
 		}
 		}
 
 
 		// Finally we will assume that the method is actually the beginning of a
 		// Finally we will assume that the method is actually the beginning of a

+ 14 - 1
laravel/tests/cases/eloquent.test.php

@@ -133,6 +133,19 @@ class EloquentTest extends PHPUnit_Framework_TestCase {
 		Model::$accessible = null;
 		Model::$accessible = null;
 	}
 	}
 
 
+	/**
+	 * Test the Model::__set method allows chaining.
+	 *
+	 * @group laravel
+	 */
+	public function testAttributeMagicSetterMethodAllowsChaining()
+	{
+		$model = new Model;
+		$this->assertInstanceOf('Model', $model->set_foo('foo'));
+		$model->set_bar('bar')->set_baz('baz');
+		$this->assertEquals(array('foo' => 'foo', 'bar' => 'bar', 'baz' => 'baz'), $model->to_array());
+	}
+
 	/**
 	/**
 	 * Test the Model::__get method.
 	 * Test the Model::__get method.
 	 *
 	 *
@@ -288,4 +301,4 @@ class EloquentTest extends PHPUnit_Framework_TestCase {
 
 
 	}
 	}
 
 
-}
+}