newsupernova/vendor/magento/module-indexer/Test/Unit/Model/ResourceModel/AbstractResourceStub.php

<?php
/**
 * Copyright © Magento, Inc. All rights reserved.
 * See COPYING.txt for license details.
 */
namespace Magento\Indexer\Test\Unit\Model\ResourceModel;

class AbstractResourceStub extends \Magento\Indexer\Model\ResourceModel\AbstractResource
{
    /**
     * New DB Adapter
     *
     * @var bool
     */
    protected $_newIndexAdapter = false;

    /**
     * Resource initializations
     *
     * @return void
     */
    protected function _construct()
    {
        $this->_init('catalog_category_flat', 'entity_id');
    }

    /**
     * Retrieve connection for write data
     *
     * @return \Magento\Framework\DB\Adapter\AdapterInterface|false
     */
    protected function _getIndexAdapter()
    {
        if (!$this->_newIndexAdapter) {
            return parent::_getIndexAdapter();
        } else {
            return $this->_getConnection('new');
        }
    }

    /**
     * Change write adapter
     *
     * @param bool $newIndexAdapter
     */
    public function newIndexAdapter($newIndexAdapter = true)
    {
        $this->_newIndexAdapter = $newIndexAdapter;
    }
}