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);
 }