public function testPrepareDataSource() { $dataSource = ['data' => ['items' => [[ProductInterface::STATUS => self::STATUS_ID]]]]; $expectedDataSource = ['data' => ['items' => [[ProductInterface::STATUS => self::STATUS_ID, '' => self::STATUS_TEXT]]]]; $this->statusMock->expects($this->once())->method('getOptionText')->with(self::STATUS_ID)->willReturn(self::STATUS_TEXT); $this->assertEquals($expectedDataSource, $this->getModel()->prepareDataSource($dataSource)); }