_setModelClass($name); } } /** * ModelAwareTrait test case */ class ModelAwareTraitTest extends TestCase { /** * Test set modelClass * * @return void */ public function testSetModelClass() { $stub = new Stub(); $this->assertNull($stub->modelClass); $stub->setProps('StubArticles'); $this->assertEquals('StubArticles', $stub->modelClass); } /** * test loadModel() * * @return void */ public function testLoadModel() { $stub = new Stub(); $stub->setProps('Articles'); $stub->modelFactory('Table', ['\Cake\ORM\TableRegistry', 'get']); $this->assertTrue($stub->loadModel()); $this->assertInstanceOf('Cake\ORM\Table', $stub->Articles); $this->assertTrue($stub->loadModel('Comments')); $this->assertInstanceOf('Cake\ORM\Table', $stub->Comments); } /** * test alternate model factories. * * @return void */ public function testModelFactory() { $stub = new Stub(); $stub->setProps('Articles'); $stub->modelFactory('Test', function($name) { $mock = new \StdClass(); $mock->name = $name; return $mock; }); $result = $stub->loadModel('Magic', 'Test'); $this->assertTrue($result); $this->assertInstanceOf('\StdClass', $stub->Magic); $this->assertEquals('Magic', $stub->Magic->name); } /** * test MissingModelException being thrown * * @return void * @expectedException Cake\Model\Error\MissingModelException * @expectedExceptionMessage Model class "Magic" of type "Test" could not be found. */ public function testMissingModelException() { $stub = new Stub(); $stub->modelFactory('Test', function($name) { return false; }); $stub->loadModel('Magic', 'Test'); } }