Added Elastica\Query\QueryString::setTimezone

Author: im-denisenko

changes.txt

@@ -2,6 +2,7 @@ CHANGES
 
 2015-04-07
 - [BC break] Elastica\Query\QueryString::setLowercaseExpandedTerms removed
+- Added Elastica\Query\QueryString::setTimezone
 
 2015-04-06
 - Update Elasticsearch version to 1.5

lib/Elastica/Query/QueryString.php

@@ -242,6 +242,17 @@ public function setRewrite($rewrite = "")
         return $this->setParam('rewrite', $rewrite);
     }
 
+    /**
+     * Set timezone option.
+     *
+     * @param  string $timezone
+     * @return $this
+     */
+    public function setTimezone($timezone)
+    {
+        return $this->setParam('timezone', $timezone);
+    }
+
     /**
      * Converts query to array
      *

test/lib/Elastica/Test/Query/QueryStringTest.php

@@ -123,4 +123,55 @@ public function testSetQueryInvalid()
         $query = new QueryString();
         $query->setQuery(array());
     }
+
+    public function testSetTimezone()
+    {
+        $timezone = 'Europe/Paris';
+        $text = 'date:[2012 TO 2014]';
+
+        $query = new QueryString($text);
+        $query->setTimezone($timezone);
+
+        $expected = array(
+            'query_string' => array(
+                'text' => $text,
+                'timezone' => $timezone,
+            ),
+        );
+
+        $this->assertEquals($expected, $query->toArray());
+        $this->assertInstanceOf('Elastica\Query\QueryString', $query->setTimezone($timezone));
+    }
+
+    public function testSetTimezoneWorks()
+    {
+        $index = $this->_createIndex();
+        $type = $index->getType('timezone_test');
+
+        $type->setMapping(array(
+            'tweet_date' => array('type' => 'datetime', 'format' => 'yyyy-MM-dd HH:mm Z')
+        ));
+
+        $documents = array(
+            new Document(1, array(
+                'tweet_date' => '2015-04-07 15:00 Europe/Moscow',
+            )),
+            new Document(2, array(
+                'tweet_date' => '2015-04-08 01:00 Europe/Moscow',
+            )),
+        );
+
+        $type->addDocuments($documents);
+        $index->refresh();
+
+        $query = new QueryString('tweet_date:[* TO 2015-04-07]');
+
+        $query->setTimezone('Europe/Moscow');
+        $results = $type->search($query);
+        $this->assertEquals(1, $resultSet->count());
+
+        $query->setTimezone('Europe/Paris');
+        $results = $type->search($query);
+        $this->assertEquals(2, $resultSet->count());
+    }
 }