Merge pull request #35 from juriansluiman/feature/query-assembling

Add assembling method for query strategy
basz · Sep 18, 2013 · b9d5cfc · b9d5cfc
2 parents a9f3fad + e1a299a
commit b9d5cfc
Show file tree

Hide file tree

Showing 2 changed files with 62 additions and 0 deletions.
diff --git a/src/SlmLocale/Strategy/QueryStrategy.php b/src/SlmLocale/Strategy/QueryStrategy.php
@@ -100,4 +100,20 @@ public function detect(LocaleEvent $event)
         return $locale;
     }
 
+    /**
+     * {@inheritdoc}
+     */
+    public function assemble(LocaleEvent $event)
+    {
+        $uri     = $event->getUri();
+        $locale  = $event->getLocale();
+
+        $query = $uri->getQueryAsArray();
+        $key   = $this->getQueryKey();
+
+        $query[$key] = $locale;
+
+        $uri->setQuery($query);
+        return $uri;
+    }
 }
diff --git a/tests/SlmLocaleTest/Strategy/QueryStrategyTest.php b/tests/SlmLocaleTest/Strategy/QueryStrategyTest.php
@@ -46,6 +46,7 @@
 
 use Zend\Http\Request  as HttpRequest;
 use Zend\Http\Response as HttpResponse;
+use Zend\Uri\Uri;
 
 class QueryStrategyTest extends TestCase
 {
@@ -125,4 +126,49 @@ public function testQueryKeyCanBeModifiedAndHaveLocaleReturned()
         $this->assertEquals('locale', $locale);
     }
 
+    public function testAssemblingAddsQueryParameter()
+    {
+        $strategy = $this->strategy;
+        $event    = $this->event;
+        $uri      = new Uri('/');
+
+        $event->setLocale('en-US');
+        $event->setUri($uri);
+        $strategy->assemble($event);
+
+        $query    = $event->getUri()->getQuery();
+        $expected = 'lang=en-US';
+        $this->assertEquals($expected, $query);
+    }
+
+    public function testAssemblingReplacesExistingQueryParameter()
+    {
+        $strategy = $this->strategy;
+        $event    = $this->event;
+        $uri      = new Uri('/?lang=nl-NL');
+
+        $event->setLocale('en-US');
+        $event->setUri($uri);
+        $strategy->assemble($event);
+
+        $query    = $event->getUri()->getQuery();
+        $expected = 'lang=en-US';
+        $this->assertEquals($expected, $query);
+    }
+
+    public function testAsssemblingUsesQueryKeyParamter()
+    {
+        $strategy = $this->strategy;
+        $event    = $this->event;
+        $uri      = new Uri('/');
+
+        $event->setLocale('en-US');
+        $event->setUri($uri);
+        $strategy->setOptions(array('query_key' => 'language'));
+        $strategy->assemble($event);
+
+        $query    = $event->getUri()->getQuery();
+        $expected = 'language=en-US';
+        $this->assertEquals($expected, $query);
+    }
 }