fix: test case expectation

Signed-off-by: Hiroshi Miura <[email protected]>
omegat-org · Sep 13, 2023 · 8f42400 · 8f42400
1 parent a8f0a80
commit 8f42400
Showing 1 changed file with 8 additions and 7 deletions.
diff --git a/src/test/java/org/omegat/machinetranslators/libretranslate/LibreTranslateTest.java b/src/test/java/org/omegat/machinetranslators/libretranslate/LibreTranslateTest.java
@@ -1,5 +1,7 @@
 package org.omegat.machinetranslators.libretranslate;
 
+import com.fasterxml.jackson.databind.JsonNode;
+import com.fasterxml.jackson.databind.ObjectMapper;
 import org.omegat.core.TestCore;
 import org.omegat.util.Language;
 import org.omegat.util.Preferences;
@@ -42,29 +44,28 @@ public void testGetJsonResultsWithWrongJson() {
 
     @Test
     public void testResponse() throws Exception {
+        ObjectMapper mapper = new ObjectMapper();
         Preferences.setPreference(LibreTranslate.ALLOW_LIBRE_TRANSLATE, true);
         String sourceText = "Hello, how are you today?";
         String encoded = URLEncoder.encode(sourceText, StandardCharsets.UTF_8);
+        JsonNode json = mapper.readTree("{\n  \"translatedText\": \n      \"Hola, ¿cómo estás hoy?\"\n}");
+
         WireMock.stubFor(WireMock.post(WireMock.anyUrl())
                 .withHeader("Content-Type", WireMock.equalTo("application/x-www-form-urlencoded"))
                 .withRequestBody(WireMock.and(
                         WireMock.containing("q=" + encoded),
                         WireMock.containing("source=en"),
                         WireMock.containing("target=es")
                 ))
-                // .withHeader("Accept", WireMock.equalTo("application/json"))
                 .willReturn(WireMock.aResponse()
                         .withStatus(200)
-                        .withHeader("Content-Type", "application/json")
-                        .withBody("{\n"
-                                + "  \"translatedText\": \n"
-                                + "      \"Hola, \u00BFc\u00F3mo est\u00E1s hoy?\"\n"
-                                + "}")));
+                        .withHeader("Content-Type", "application/json;charset=UTF-8")
+                        .withJsonBody(json)));
 
         int port = wireMockRule.port();
         String url = String.format("http://localhost:%d/translate", port);
         LibreTranslate translator = new LibreTranslate(url);
         String result = translator.translate(new Language("en"), new Language("es"), sourceText);
-        assertEquals("Hola, \u00BFc\u00F3mo est\u00E1s hoy?", result);
+        assertEquals("Hola, ¿cómo estás hoy?", result);
     }
 }