yii2-swiftmailer/tests/unit/framework/base/BehaviorTest.php

<?php

namespace yiiunit\framework\base;

use yii\base\Behavior;
use yii\base\Component;
use yiiunit\TestCase;

class BarClass extends Component
{
}

class FooClass extends Component
{
	public function behaviors()
	{
		return array(
			'foo' => __NAMESPACE__ . '\BarBehavior',
		);
	}
}

class BarBehavior extends Behavior
{
	public $behaviorProperty = 'behavior property';

	public function behaviorMethod()
	{
		return 'behavior method';
	}

	public function __call($name, $params)
	{
		if ($name == 'magicBehaviorMethod') {
			return 'Magic Behavior Method Result!';
		}
		return parent::__call($name, $params);
	}

	public function hasMethod($name)
	{
		if ($name == 'magicBehaviorMethod') {
			return true;
		}
		return parent::hasMethod($name);
	}
}

class BehaviorTest extends TestCase
{
	protected function setUp()
	{
		parent::setUp();
		$this->mockApplication();
	}

	public function testAttachAndAccessing()
	{
		$bar = new BarClass();
		$behavior = new BarBehavior();
		$bar->attachBehavior('bar', $behavior);
		$this->assertEquals('behavior property', $bar->behaviorProperty);
		$this->assertEquals('behavior method', $bar->behaviorMethod());
		$this->assertEquals('behavior property', $bar->getBehavior('bar')->behaviorProperty);
		$this->assertEquals('behavior method', $bar->getBehavior('bar')->behaviorMethod());

		$behavior = new BarBehavior(array('behaviorProperty' => 'reattached'));
		$bar->attachBehavior('bar', $behavior);
		$this->assertEquals('reattached', $bar->behaviorProperty);
	}

	public function testAutomaticAttach()
	{
		$foo = new FooClass();
		$this->assertEquals('behavior property', $foo->behaviorProperty);
		$this->assertEquals('behavior method', $foo->behaviorMethod());
	}

	public function testMagicMethods()
	{
		$bar = new BarClass();
		$behavior = new BarBehavior();

		$this->assertFalse($bar->hasMethod('magicBehaviorMethod'));
		$bar->attachBehavior('bar', $behavior);
		$this->assertFalse($bar->hasMethod('magicBehaviorMethod', false));
		$this->assertTrue($bar->hasMethod('magicBehaviorMethod'));

		$this->assertEquals('Magic Behavior Method Result!', $bar->magicBehaviorMethod());
	}

	public function testCallUnknownMethod()
	{
		$bar = new BarClass();
		$behavior = new BarBehavior();
		$this->setExpectedException('yii\base\UnknownMethodException');

		$this->assertFalse($bar->hasMethod('nomagicBehaviorMethod'));
		$bar->attachBehavior('bar', $behavior);
		$bar->nomagicBehaviorMethod();
	}

}