Verify that AI translation variant is always created with 'includedIn…

…LocalizedFile' set to false
pinterest · Nov 6, 2024 · c37c521 · c37c521
1 parent bf6ee32
commit c37c521
Showing 1 changed file with 18 additions and 0 deletions.
diff --git a/webapp/src/test/java/com/box/l10n/mojito/service/ai/translation/AITranslateCronJobTest.java b/webapp/src/test/java/com/box/l10n/mojito/service/ai/translation/AITranslateCronJobTest.java
@@ -215,6 +215,9 @@ public void testTranslateSuccess() throws Exception {
     List<AITranslation> aiTranslations = aiTranslationCaptor.getValue();
     assertEquals(3, aiTranslations.size());
     assertThat(aiTranslations).extracting("localeId").containsExactlyInAnyOrder(2L, 3L, 4L);
+    assertThat(aiTranslations)
+        .extracting("includedInLocalizedFile")
+        .containsExactlyInAnyOrder(false, false, false);
   }
 
   @Test
@@ -247,6 +250,9 @@ public void testTranslateReuseSource() throws Exception {
     assertThat(aiTranslations)
         .extracting("translation")
         .containsExactlyInAnyOrder("content", "translated", "translated");
+    assertThat(aiTranslations)
+        .extracting("includedInLocalizedFile")
+        .containsExactlyInAnyOrder(false, false, false);
   }
 
   @Test
@@ -265,6 +271,9 @@ public void testNoTranslationIfLeveragedVariantExistsForLocale() {
     List<AITranslation> aiTranslations = aiTranslationCaptor.getValue();
     assertEquals(2, aiTranslations.size());
     assertThat(aiTranslations).extracting("localeId").containsExactlyInAnyOrder(2L, 4L);
+    assertThat(aiTranslations)
+        .extracting("includedInLocalizedFile")
+        .containsExactlyInAnyOrder(false, false);
   }
 
   @Test
@@ -368,6 +377,9 @@ public void testBatchLogic() throws JobExecutionException {
       aiTranslations = aiTranslationCaptor.getAllValues().get(i);
       assertEquals(3, aiTranslations.size());
       assertThat(aiTranslations).extracting("localeId").containsExactlyInAnyOrder(2L, 3L, 4L);
+      assertThat(aiTranslations)
+          .extracting("includedInLocalizedFile")
+          .containsExactlyInAnyOrder(false, false, false);
     }
     verify(aiTranslationService, times(3)).deleteBatch(isA(Queue.class));
   }
@@ -403,6 +415,9 @@ public void testBatchRequestFailLogic() throws JobExecutionException {
       aiTranslations = aiTranslationCaptor.getAllValues().get(i);
       assertEquals(3, aiTranslations.size());
       assertThat(aiTranslations).extracting("localeId").containsExactlyInAnyOrder(2L, 3L, 4L);
+      assertThat(aiTranslations)
+          .extracting("includedInLocalizedFile")
+          .containsExactlyInAnyOrder(false, false, false);
     }
     verify(aiTranslationService, times(3)).deleteBatch(isA(Queue.class));
   }
@@ -437,6 +452,9 @@ public void testBatchLogicFailureToRetrieveTextUnit() throws JobExecutionExcepti
       aiTranslations = aiTranslationCaptor.getAllValues().get(i);
       assertEquals(3, aiTranslations.size());
       assertThat(aiTranslations).extracting("localeId").containsExactlyInAnyOrder(2L, 3L, 4L);
+      assertThat(aiTranslations)
+          .extracting("includedInLocalizedFile")
+          .containsExactlyInAnyOrder(false, false, false);
     }
     verify(aiTranslationService, times(2)).deleteBatch(isA(Queue.class));
   }