public function testSetMainTable()
{
$setMainTableMethod = new \ReflectionMethod($this->_model, '_setMainTable');
$setMainTableMethod->setAccessible(true);
$tableName = $this->_model->getTable('store_website');
$idFieldName = 'website_id';
$setMainTableMethod->invoke($this->_model, $tableName);
$this->assertEquals($tableName, $this->_model->getMainTable());
$setMainTableMethod->invoke($this->_model, $tableName, $idFieldName);
$this->assertEquals($tableName, $this->_model->getMainTable());
$this->assertEquals($idFieldName, $this->_model->getIdFieldName());
}
/**
* @dataProvider getTableDataProvider
* @param $tableName
* @param $expectedResult
*/
public function testGetMainTable($tableName, $expectedResult)
{
$mainTableProperty = new \ReflectionProperty('Magento\\Framework\\Model\\ResourceModel\\Db\\AbstractDb', '_mainTable');
$mainTableProperty->setAccessible(true);
$mainTableProperty->setValue($this->_model, $tableName);
$this->_resourcesMock->expects($this->once())->method('getTableName')->with($expectedResult)->will($this->returnValue($expectedResult));
$this->assertEquals($expectedResult, $this->_model->getMainTable());
}
