/**
  * testCountMenu
  *
  * @return void
  */
 public function testCountMenu()
 {
     $conn = $this->getConnectionMock();
     $query = $this->getQueryBuilderMock();
     $keygen = $this->getKeygenMock();
     $menuRepository = new DBMenuRepository($conn, $keygen);
     $conn->shouldReceive('table')->with(self::MENU_TABLE)->andReturn($query);
     $query->shouldReceive('where')->andReturn($query);
     $query->shouldReceive('count')->andReturn(3);
     $count = $menuRepository->countMenu('main');
     $this->assertEquals(3, $count);
 }