implemented FriendlyBuilderBag

Aggregation/FilterAggregationTest.php

@@ -69,7 +69,7 @@ class FilterAggregationTest extends \PHPUnit_Framework_TestCase
             ->method('getName')
             ->willReturn('agg_test_agg2');
 
-        $aggregation->aggregations->addAggregation($aggregation2);
+        $aggregation->aggregations->add($aggregation2);
 
         $result = [
             'agg_test_agg' => [

Aggregation/GlobalAggregationTest.php

@@ -41,7 +41,7 @@ class GlobalAggregationTest extends \PHPUnit_Framework_TestCase
         // Case #1 nested global aggregation.
         $aggregation = new GlobalAggregation('test_agg');
         $aggregation2 = new GlobalAggregation('test_agg_2');
-        $aggregation->aggregations->addAggregation($aggregation2);
+        $aggregation->aggregations->add($aggregation2);
 
         $result = [
             'agg_test_agg' => [

Aggregation/RangeAggregationTest.php

@@ -88,7 +88,7 @@ class RangeAggregationTest extends \PHPUnit_Framework_TestCase
         $aggregation2 = new RangeAggregation('test_agg_2');
         $aggregation2->addRange('20', '20');
 
-        $aggregation->aggregations->addAggregation($aggregation2);
+        $aggregation->aggregations->add($aggregation2);
 
         $result = [
             'agg_test_agg' => [

FriendlyBuilderBagTest.php

+<?php
+
+/*
+ * This file is part of the ONGR package.
+ *
+ * (c) NFQ Technologies UAB <info@nfq.com>
+ *
+ * For the full copyright and license information, please view the LICENSE
+ * file that was distributed with this source code.
+ */
+
+namespace ONGR\ElasticsearchBundle\Tests\Unit\DSL;
+
+
+use ONGR\ElasticsearchBundle\DSL\FriendlyBuilderBag;
+use ONGR\ElasticsearchBundle\DSL\FriendlyBuilderInterface;
+
+class FriendlyBuilderBagTest extends \PHPUnit_Framework_TestCase
+{
+    /**
+     * Tests if bag knows if he has a builder.
+     */
+    public function testHas()
+    {
+        $bag = new FriendlyBuilderBag(
+            [
+                $this->getBuilder('foo'),
+            ]
+        );
+        $this->assertTrue($bag->has('foo'));
+    }
+
+    /**
+     * Tests if bag can remove a builder.
+     */
+    public function testRemove()
+    {
+        $bag = new FriendlyBuilderBag(
+            [
+                $this->getBuilder('foo'),
+                $this->getBuilder('baz'),
+            ]
+        );
+
+        $bag->remove('foo');
+
+        $this->assertFalse($bag->has('foo'), 'Foo builder should not exist anymore.');
+        $this->assertTrue($bag->has('baz'), 'Baz builder should exist.');
+    }
+
+    /**
+     * Tests if bag can clear it's builders.
+     */
+    public function testClear()
+    {
+        $bag = new FriendlyBuilderBag(
+            [
+                $this->getBuilder('foo'),
+                $this->getBuilder('baz'),
+            ]
+        );
+
+        $bag->clear();
+
+        $this->assertEmpty($bag->all());
+    }
+
+    /**
+     * Tests if bag can get a builder.
+     */
+    public function testGet()
+    {
+        $bag = new FriendlyBuilderBag(
+            [
+                $this->getBuilder('baz'),
+            ]
+        );
+
+        $this->assertNotEmpty($bag->get('baz'));
+    }
+
+    /**
+     * Returns builder.
+     *
+     * @param string $name
+     *
+     * @return \PHPUnit_Framework_MockObject_MockObject|FriendlyBuilderInterface
+     */
+    private function getBuilder($name)
+    {
+        $friendlyBuilderMock = $this->getMock('ONGR\ElasticsearchBundle\DSL\FriendlyBuilderInterface');
+
+        $friendlyBuilderMock
+            ->expects($this->once())
+            ->method('getName')
+            ->will($this->returnValue($name));
+
+        $friendlyBuilderMock
+            ->expects($this->any())
+            ->method('toArray')
+            ->will($this->returnValue([]));
+
+        return $friendlyBuilderMock;
+    }
+}