public function testSetForm() { $baseUrl = 'base_url'; $attributeKey = 'attribute_key'; $attributeValue = 'attribute_value'; $this->dataForm->expects($this->once())->method('setParent')->with($this->model)->willReturnSelf(); $this->dataForm->expects($this->once())->method('setBaseUrl')->with($baseUrl)->willReturnSelf(); $this->dataForm->expects($this->once())->method('addCustomAttribute')->with($attributeKey, $attributeValue)->willReturnSelf(); $this->urlBuilder->expects($this->once())->method('getBaseUrl')->willReturn($baseUrl); $this->model->setData('custom_attributes', [$attributeKey => $attributeValue]); $this->assertEquals($this->model, $this->model->setForm($this->dataForm)); }