/** * @param $attributeCode * @param $data * @param $expected * @dataProvider afterLoadDataProvider */ public function testAfterLoad($attributeCode, $data, $expected) { $this->_attribute->expects($this->any())->method('getName')->will($this->returnValue($attributeCode)); $object = new \Magento\Framework\DataObject($data); $this->_model->afterLoad($object); $this->assertTrue($object->hasData($attributeCode)); $this->assertSame($expected, $object->getData($attributeCode)); }