Allow bool in Query::setSource function #818

changes.txt

@@ -1,5 +1,8 @@
 CHANGES
 
+2015-04-23
+- Allow bool in Query::setSource function #818 http://www.elastic.co/guide/en/elasticsearch/reference/current/search-request-source-filtering.html
+
 2015-04-15
 - deleteByQuery() implemented in Elastica\Index
 

lib/Elastica/Query.php

@@ -412,13 +412,13 @@ public function setRescore($rescore)
     /**
      * Sets the _source field to be returned with every hit
      *
-     * @param  array $fields Fields to be returned
+     * @param  array|bool $params Fields to be returned or false to disable source
      * @return $this
      * @link   http://www.elastic.co/guide/en/elasticsearch/reference/current/search-request-source-filtering.html
      */
-    public function setSource(array $fields)
+    public function setSource($params)
     {
-        return $this->setParam('_source', $fields);
+        return $this->setParam('_source', $params);
     }
 
     /**

test/lib/Elastica/Test/QueryTest.php

@@ -13,6 +13,7 @@
 use Elastica\ScriptFields;
 use Elastica\Suggest;
 use Elastica\Test\Base as BaseTest;
+use Elastica\Type;
 
 class QueryTest extends BaseTest
 {
@@ -356,4 +357,41 @@ public function testSetPostFilterToArrayCast()
 
         $this->assertNotEquals($query->toArray(), $anotherQuery->toArray());
     }
+
+    public function testNoSource()
+    {
+        $index = $this->_createIndex();
+
+        $type = new Type($index, 'user');
+
+        // Adds 1 document to the index
+        $doc1 = new Document(1,
+            array('username' => 'ruflin', 'test' => array('2', '3', '5'))
+        );
+        $type->addDocument($doc1);
+
+        // To update index
+        $index->refresh();
+
+        $query = Query::create('ruflin');
+        $resultSet = $type->search($query);
+
+        // Disable source
+        $query->setSource(false);
+
+        $resultSetNoSource = $type->search($query);
+
+        $this->assertEquals(1, $resultSet->count());
+        $this->assertEquals(1, $resultSetNoSource->count());
+
+        // Tests if no source is in response except id
+        $result = $resultSetNoSource->current();
+        $this->assertEquals(1, $result->getId());
+        $this->assertEmpty($result->getData());
+
+        // Tests if source is in response except id
+        $result = $resultSet->current();
+        $this->assertEquals(1, $result->getId());
+        $this->assertNotEmpty($result->getData());
+    }
 }