Update tests. Use lang code for testing

freQuensy23-coder · Mar 26, 2023 · 7804560 · 7804560
1 parent 997c7be
commit 7804560
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/lib/src/test/kotlin/reverso/ReversoTranslatorAPITest.kt b/lib/src/test/kotlin/reverso/ReversoTranslatorAPITest.kt
@@ -1,3 +1,6 @@
+package reverso
+
+import ReversoTranslatorAPI
 import okhttp3.RequestBody
 import okio.Buffer
 import org.junit.jupiter.api.Assertions.*
@@ -79,7 +82,7 @@ internal class ReversoTranslatorAPITest{
 
     @Test
     fun testTranslation(){
-        val translationResponse = reversoTranslatorAPI.translate(translatingText, "en", "ru")
+        val translationResponse = reversoTranslatorAPI.translate(translatingText, LanguageCode("en"), LanguageCode("ru"))
         assertEquals(expectedTranslation, translationResponse.dictionary_entry_list[0].term)
     }
 
@@ -89,15 +92,15 @@ internal class ReversoTranslatorAPITest{
         val textsToTranslate = listOf("hello", "be", "government", "goverment", "go")
         val rightTranslation = listOf("привет", "быть", "правительство", "правительство", "пойти")
         textsToTranslate.forEachIndexed { index, text ->
-            val translationResponse = reversoTranslatorAPI.translate(text, "en", "ru")
+            val translationResponse = reversoTranslatorAPI.translate(text, LanguageCode("en"), LanguageCode("ru"))
             assertEquals(rightTranslation[index], translationResponse.dictionary_entry_list[0].term)
             println("$text -> ${translationResponse.dictionary_entry_list[0].term} ok")
         }
     }
 
     @Test
     fun testSentenceTranslation(){
-        val translationResponse = reversoTranslatorAPI.translate("I am a student", "en", "ru")
+        val translationResponse = reversoTranslatorAPI.translate("I am a student", LanguageCode("en"), LanguageCode("ru"))
         assertEquals("я студент", translationResponse.dictionary_entry_list[0].term)
     }