Implemented some tests for the count

Author: 0x6368

Tests/Functional/Service/IndexServiceTest.php

@@ -15,6 +15,7 @@
 use ONGR\App\Entity\DummyDocumentInTheEntityDirectory;
 use ONGR\ElasticsearchBundle\Result\DocumentIterator;
 use ONGR\ElasticsearchBundle\Test\AbstractElasticsearchTestCase;
+use ONGR\ElasticsearchDSL\Query\FullText\SimpleQueryStringQuery;
 
 /**
  * Functional tests for orm manager.
@@ -120,4 +121,37 @@ public function testIndexConfigOverride()
 
         $this->assertEquals(['localhost:9200'], $hosts);
     }
+
+    public function testCount()
+    {
+        $index = $this->getIndex(DummyDocument::class);
+
+        $search = $index->createSearch();
+        $search->addQuery(new SimpleQueryStringQuery('foo'));
+
+        $count = $index->count($search);
+
+        $this->assertEquals(2, $count);
+    }
+
+    public function testRawCount()
+    {
+        $index = $this->getIndex(DummyDocument::class);
+
+        $search = $index->createSearch();
+        $search->addQuery(new SimpleQueryStringQuery('bar'));
+
+        $result = $index->count($search, [], true);
+
+        $this->assertEquals(1, $result['count']);
+    }
+
+    public function testIndexDocumentCount()
+    {
+        $index = $this->getIndex(DummyDocument::class);
+
+        $count = $index->getIndexDocumentCount();
+
+        $this->assertEquals(3, $count);
+    }
 }