1 files changed, 19 insertions, 17 deletions

diff --git a/vendor/ruflin/elastica/test/lib/Elastica/Test/Aggregation/ValueCountTest.php b/vendor/ruflin/elastica/test/lib/Elastica/Test/Aggregation/ValueCountTest.php
index 5eba9ea4..21b63cbe 100644
--- a/vendor/ruflin/elastica/test/lib/Elastica/Test/Aggregation/ValueCountTest.php
+++ b/vendor/ruflin/elastica/test/lib/Elastica/Test/Aggregation/ValueCountTest.php
@@ -1,38 +1,40 @@
 <?php
-
 namespace Elastica\Test\Aggregation;
 
-
 use Elastica\Aggregation\ValueCount;
 use Elastica\Document;
 use Elastica\Query;
 
 class ValueCountTest extends BaseAggregationTest
 {
-    protected function setUp()
+    protected function _getIndexForTest()
     {
-        parent::setUp();
-        $this->_index = $this->_createIndex('value_count');
-        $docs = array(
-            new Document('1', array('price' => 5)),
-            new Document('2', array('price' => 8)),
-            new Document('3', array('price' => 1)),
-            new Document('4', array('price' => 3)),
-            new Document('5', array('price' => 3)),
-        );
-        $this->_index->getType('test')->addDocuments($docs);
-        $this->_index->refresh();
+        $index = $this->_createIndex();
+
+        $index->getType('test')->addDocuments(array(
+            new Document(1, array('price' => 5)),
+            new Document(2, array('price' => 8)),
+            new Document(3, array('price' => 1)),
+            new Document(4, array('price' => 3)),
+            new Document(5, array('price' => 3)),
+        ));
+
+        $index->refresh();
+
+        return $index;
     }
 
+    /**
+     * @group functional
+     */
     public function testValueCountAggregation()
     {
-        $agg = new ValueCount("count", "price");
+        $agg = new ValueCount('count', 'price');
 
         $query = new Query();
         $query->addAggregation($agg);
-        $results = $this->_index->search($query)->getAggregation("count");
+        $results = $this->_getIndexForTest()->search($query)->getAggregation('count');
 
         $this->assertEquals(5, $results['value']);
     }
 }
- 
\ No newline at end of file