storeMock = $this->getMockBuilder(Store::class)
->disableOriginalConstructor()
->getMock();
}
/**
* @return Visibility
*/
protected function getModel()
{
return $this->objectManager->getObject(Visibility::class, [
'context' => $this->contextMock,
'uiComponentFactory' => $this->uiComponentFactoryMock,
'components' => [],
'data' => [],
'store' => $this->storeMock,
]);
}
public function testPrepareDataSource()
{
$dataSource = [
'data' => [
'items' => [
[
'stores' => [1]
]
],
],
];
$expectedVisibility =
"Test Website
Test group
Test store
";
$expectedDataSource = [
'data' => [
'items' => [
[
'stores' => [1],
'visibility' => $expectedVisibility,
]
],
],
];
$this->storeMock->expects($this->once())
->method('getStoresStructure')
->willReturn([
[
'label' => 'Test Website',
'children' => [
[
'label' => 'Test group',
'children' => [
[
'label' => 'Test store',
]
],
]
],
],
]);
$this->assertEquals($expectedDataSource, $this->getModel()->prepareDataSource($dataSource));
}
}