public function testGetCount() { $categoriesMock = $this->getMock('\\Magento\\Catalog\\Model\\ResourceModel\\Category\\Collection', [], [], '', false); $this->categoriesFactoryMock->expects($this->once())->method('create')->willReturn($categoriesMock); $categoriesMock->expects($this->once())->method('addAttributeToFilter')->with('parent_id', ['gt' => 0])->willReturnSelf(); $categoriesMock->expects($this->once())->method('getSize')->willReturn('expected'); $this->assertEquals('expected', $this->model->getCount()); }
/** * @expectedException \Magento\Framework\Exception\LocalizedException * @expectedExceptionMessage Could not move category */ public function testMoveWithCouldNotMoveException() { $categoryId = 2; $parentId = 1; $afterId = null; $categoryMock = $this->getMock('\\Magento\\Catalog\\Model\\Category', [], [], 'categoryMock', false); $parentCategoryMock = $this->getMock('\\Magento\\Catalog\\Model\\Category', [], [], 'parentCategoryMock', false); $this->categoryRepositoryMock->expects($this->exactly(2))->method('get')->will($this->returnValueMap([[$categoryId, null, $categoryMock], [$parentId, null, $parentCategoryMock]])); $categoryMock->expects($this->once())->method('move')->with($parentId, $afterId)->willThrowException(new \Magento\Framework\Exception\LocalizedException(__('message'))); $this->model->move($categoryId, $parentId, $afterId); }