Stockbase-Connect / magento2-module

    /**

     * testGetReserveForProduct

     */

    public function testGetReserveForProduct()

    {

        $reserveCollection = $this->createMock(StockItemReserveCollection::class);

        $this->objectManager->expects($this->once())->method('create')

            ->with(StockItemReserveCollection::class)

            ->willReturn($reserveCollection);

        $reserve = $this->createMock(StockItemReserve::class);

        $reserveCollection->expects($this->any())->method('addFieldToFilter')

            ->with('product_id', ['in' => [101]]);

        $reserveCollection->expects($this->once())->method('getItems')

            ->willReturn([$reserve]);

        $model = $this->createModel();

        $result = $model->getReserveForProduct(101);

        $this->assertEquals([$reserve], $result);

    }

    /**

     * testGetReserveForQuoteItem

    /**

     * testGetReserveForQuoteItem

     */

    public function testGetReserveForQuoteItem()

    {

        $reserveCollection = $this->createMock(StockItemReserveCollection::class);

        $this->objectManager->expects($this->once())->method('create')

            ->with(StockItemReserveCollection::class)

            ->willReturn($reserveCollection);

        $reserve = $this->createMock(StockItemReserve::class);

        $reserveCollection->expects($this->any())->method('addFieldToFilter')

            ->with('quote_item_id', ['in' => [101]]);

        $reserveCollection->expects($this->once())->method('getItems')

            ->willReturn([$reserve]);

        $model = $this->createModel();

        $result = $model->getReserveForQuoteItem(101);

        $this->assertEquals([$reserve], $result);

    }

    protected function createModel()

    {