Replacing _ for -

kontent-ai · Sep 28, 2017 · bad0166 · bad0166
1 parent 1c5cad8
commit bad0166
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 3 deletions.
diff --git a/src/main/java/com/kenticocloud/delivery/DeliveryParameterBuilder.java b/src/main/java/com/kenticocloud/delivery/DeliveryParameterBuilder.java
@@ -182,7 +182,7 @@ public DeliveryParameterBuilder language(String language) {
 
     public DeliveryParameterBuilder language(Locale language) {
         if (language != null) {
-            nameValuePairs.add(new BasicNameValuePair(LANGUAGE, language.toString()));
+            nameValuePairs.add(new BasicNameValuePair(LANGUAGE, language.toString().replace('_', '-')));
         }
         return this;
     }

diff --git a/src/test/java/com/kenticocloud/delivery/DeliveryParameterBuilderTest.java b/src/test/java/com/kenticocloud/delivery/DeliveryParameterBuilderTest.java
@@ -273,10 +273,18 @@ public void testLanguage() {
 
     @Test
     public void testLocaleLanguage() {
-        List<NameValuePair> params = DeliveryParameterBuilder.params().language(new Locale("pt-br")).build();
+        List<NameValuePair> params = DeliveryParameterBuilder.params().language(new Locale("pt", "br")).build();
         Assert.assertEquals(1, params.size());
         Assert.assertEquals("language", params.get(0).getName());
-        Assert.assertEquals("pt-br", params.get(0).getValue());
+        Assert.assertEquals("pt-BR", params.get(0).getValue());
+    }
+
+    @Test
+    public void testLocaleWithNoVariantLanguage() {
+        List<NameValuePair> params = DeliveryParameterBuilder.params().language(new Locale("en")).build();
+        Assert.assertEquals(1, params.size());
+        Assert.assertEquals("language", params.get(0).getName());
+        Assert.assertEquals("en", params.get(0).getValue());
     }
 
     @Test