public function testCorrectDecoratorCreated() { $company = new Company(); $decorator = DataDecorator::getDecoratorForModel($company); $this->assertInstanceOf("Rhubarb\\Stem\\Tests\\Fixtures\\CompanyDecorator", $decorator); $decorator = DataDecorator::getDecoratorForModel(new Category()); $this->assertFalse($decorator, "If no decorator exists false should be returned."); $example = new Example(); $decorator = DataDecorator::getDecoratorForModel($example); $this->assertInstanceOf("Rhubarb\\Stem\\Tests\\Fixtures\\ExampleDecorator", $decorator); DataDecorator::registerDecoratorClass("Rhubarb\\Stem\\Tests\\Fixtures\\ModelDecorator", "Rhubarb\\Stem\\Models\\Model"); $user = new User(); $decorator = DataDecorator::getDecoratorForModel($user); $this->assertInstanceOf("Rhubarb\\Stem\\Tests\\Fixtures\\ModelDecorator", $decorator); }
protected function initialise() { SolutionSchema::registerSchema('CommunicationsSolutionSchema', CommunicationsSolutionSchema::class); DataDecorator::registerDecoratorClass(CommunicationDecorator::class, CommunicationItem::class); DataDecorator::registerDecoratorClass(CommunicationDecorator::class, Communication::class); }