public function testGetList() { $criteriaMock = $this->getMockBuilder('Magento\\CatalogInventory\\Api\\StockStatusCriteriaInterface')->getMock(); $queryBuilderMock = $this->getMockBuilder('Magento\\Framework\\DB\\QueryBuilder')->disableOriginalConstructor()->setMethods(['setCriteria', 'setResource', 'create'])->getMock(); $queryMock = $this->getMockBuilder('Magento\\Framework\\DB\\QueryInterface')->getMock(); $queryCollectionMock = $this->getMockBuilder('Magento\\CatalogInventory\\Api\\Data\\StockStatusCollectionInterface')->getMock(); $this->queryBuilderFactoryMock->expects($this->once())->method('create')->willReturn($queryBuilderMock); $queryBuilderMock->expects($this->once())->method('setCriteria')->with($criteriaMock)->willReturnSelf(); $queryBuilderMock->expects($this->once())->method('setResource')->with($this->stockStatusResourceMock)->willReturnSelf(); $queryBuilderMock->expects($this->once())->method('create')->willReturn($queryMock); $this->stockStatusCollectionMock->expects($this->once())->method('create')->willReturn($queryCollectionMock); $this->assertEquals($queryCollectionMock, $this->model->getList($criteriaMock)); }
/** * @inheritdoc */ public function getList(\Magento\CatalogInventory\Api\StockItemCriteriaInterface $criteria) { $queryBuilder = $this->queryBuilderFactory->create(); $queryBuilder->setCriteria($criteria); $queryBuilder->setResource($this->resource); $query = $queryBuilder->create(); $collection = $this->stockItemCollectionFactory->create(['query' => $query]); return $collection; }
/** * Run test getList method * * @return void */ public function testGetList() { $criteriaMock = $this->getMock('Magento\\Cms\\Model\\Resource\\PageCriteria', [], [], '', false); $queryBuilderMock = $this->getMock('Magento\\Framework\\DB\\QueryBuilder', ['setCriteria', 'setResource', 'create'], [], '', false); $queryMock = $this->getMockForAbstractClass('Magento\\Framework\\DB\\QueryInterface', [], '', false); $collectionMock = $this->getMock('Magento\\Cms\\Model\\Resource\\Page\\Collection', [], [], '', false); $this->queryBuilderFactoryMock->expects($this->once())->method('create')->will($this->returnValue($queryBuilderMock)); $queryBuilderMock->expects($this->once())->method('setCriteria')->with($criteriaMock); $queryBuilderMock->expects($this->once())->method('setResource')->with($this->resourceMock); $queryBuilderMock->expects($this->once())->method('create')->will($this->returnValue($queryMock)); $this->pageCollectionFactoryMock->expects($this->once())->method('create')->with(['query' => $queryMock])->will($this->returnValue($collectionMock)); $this->assertEquals($collectionMock, $this->pageRepository->getList($criteriaMock)); }