TestGoogleMTv3Connector_IT

package net.sf.okapi.connectors.google.v3;

import static org.junit.jupiter.api.Assertions.assertEquals;
import static org.junit.jupiter.api.Assertions.assertFalse;
import static org.junit.jupiter.api.Assertions.assertNotNull;
import static org.junit.jupiter.api.Assertions.assertTrue;

import java.util.List;

import org.junit.jupiter.api.AfterEach;
import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.DisplayName;
import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.condition.EnabledIf;

import com.acumenvelocity.ath.common.Const;

import net.sf.okapi.common.LocaleId;
import net.sf.okapi.common.query.QueryResult;
import net.sf.okapi.common.resource.TextFragment;
import net.sf.okapi.common.resource.TextFragment.TagType;

/**
 * Integration tests for GoogleMTv3Connector using the real Google Cloud Translation API v3.
 * Requires valid credentials via environment variables:
 * - ATH_GT_API_KEY (API key) OR
 * - ATH_GCS_SECRET_FILE (path to service account JSON)
 * Project ID is fixed to "ath-okapi"
 */
@DisplayName("GoogleMTv3Connector Integration Tests")
@EnabledIf("isCredentialsAvailable")
class TestGoogleMTv3Connector_IT {

  private GoogleMTv3Connector connector;

  /** Custom condition – enables the entire test class only if credentials are configured */
  static boolean isCredentialsAvailable() {
    return System.getenv("ATH_GT_API_KEY") != null || System.getenv("ATH_GCS_SECRET_FILE") != null;
  }

  @BeforeEach
  void setUp() {
    String apiKey = System.getenv("ATH_GT_API_KEY");
    String credentialsPath = System.getenv("ATH_GCS_SECRET_FILE");

    GoogleMTv3Parameters params = new GoogleMTv3Parameters();

    params.setProjectLocation(Const.US_CENTRAL1_PROJECT_LOCATION);
    if (apiKey != null && !apiKey.isBlank()) {
      params.setApiKey(apiKey);
    }
    if (credentialsPath != null && !credentialsPath.isBlank()) {
      params.setCredentialsPath(credentialsPath);
    }
    params.setMimeType("text/plain");

    connector = new GoogleMTv3Connector();
    connector.setParameters(params);
    connector.setLanguages(LocaleId.ENGLISH, LocaleId.SPANISH);
    connector.open();
  }

  @AfterEach
  void tearDown() {
    if (connector != null) {
      connector.close();
    }
  }

  @Test
  @DisplayName("Basic plain text translation EN → ES")
  void testPlainTextTranslation() {
    connector.setLanguages(LocaleId.ENGLISH, LocaleId.SPANISH);

    int count = connector.query("Hello, world!");

    assertEquals(1, count);
    assertTrue(connector.hasNext());

    QueryResult qr = connector.next();
    assertNotNull(qr);
    assertEquals(100, qr.getFuzzyScore());
    assertEquals(net.sf.okapi.common.query.MatchType.MT, qr.matchType);

    String translated = qr.target.getCodedText();
    assertFalse(translated.equals("Hello, world!"), "Translation should be different from source");
    assertTrue(
        translated.toLowerCase().contains("hola") || translated.toLowerCase().contains("mundo"),
        "Expected Spanish translation, got: " + translated);
  }

  @Test
  @DisplayName("Coded text translation with inline tags preserved (EN → FR)")
  void testCodedTextTranslation() {
    connector.setLanguages(LocaleId.ENGLISH, LocaleId.FRENCH);

    TextFragment src = new TextFragment();
    src.append("Click the ");
    src.append(TagType.OPENING, "b", "<b>");
    src.append("Save");
    src.append(TagType.CLOSING, "b", "</b>");
    src.append(" button to continue.");

    int count = connector.query(src);

    assertEquals(1, count);
    assertTrue(connector.hasNext());

    QueryResult qr = connector.next();
    assertNotNull(qr);
    assertEquals(100, qr.getFuzzyScore());

    // Check that codes are preserved
    assertTrue(qr.target.hasCode(), "Target should have codes");

    String targetText = qr.target.toText();
    // Should contain French words
    assertTrue(targetText.toLowerCase().contains("bouton") ||
        targetText.toLowerCase().contains("enregistrer") ||
        targetText.toLowerCase().contains("sauvegarder") ||
        targetText.toLowerCase().contains("cliqu"),
        "Expected French translation: " + targetText);
  }

  @Test
  @DisplayName("Language detection works (source = und → DE)")
  void testLanguageDetection() {
    connector.setLanguages(LocaleId.EMPTY, LocaleId.GERMAN);

    int count = connector.query("This is a test sentence in English.");

    assertEquals(1, count);
    assertTrue(connector.hasNext());

    QueryResult qr = connector.next();
    assertNotNull(qr);

    String translated = qr.target.getCodedText().toLowerCase();
    assertTrue(
        translated.contains("ist") || translated.contains("test") || translated.contains("satz"),
        "Should be German: " + translated);
  }

  @Test
  @DisplayName("getSupportedLanguages returns a large list")
  void testGetSupportedLanguages() {
    List<LocaleId> languages = connector.getSupportedLanguages();

    assertNotNull(languages);
    assertTrue(languages.size() > 130,
        "Google Translate v3 supports >130 languages, got " + languages.size());
    assertTrue(languages.contains(LocaleId.fromBCP47("en")));
    assertTrue(languages.contains(LocaleId.fromBCP47("es")));
    assertTrue(languages.contains(LocaleId.fromBCP47("zh-CN")));
    assertTrue(languages.contains(LocaleId.fromBCP47("ja")));
  }

  @Test
  @DisplayName("Connector metadata is correct")
  void testConnectorMetadata() {
    assertEquals("Google-MTv3", connector.getName());
    String settings = connector.getSettingsDisplay();
    assertTrue(settings.contains(Const.US_CENTRAL1_PROJECT_LOCATION));
  }

  @Test
  @DisplayName("Empty query returns 0 and no results")
  void testEmptyQuery() {
    int count = connector.query("");
    assertEquals(0, count);
    assertFalse(connector.hasNext());
  }

  @Test
  @DisplayName("Very long text is handled correctly (batching)")
  void testLongTextBatching() {
    connector.setLanguages(LocaleId.ENGLISH, LocaleId.JAPANESE);

    // Create a long text > 30k codepoints to force batching
    StringBuilder sb = new StringBuilder();

    for (int i = 0; i < 400; i++) {
      sb.append("This is a test sentence number ").append(i).append(". ");
    }

    String longText = sb.toString();

    int count = connector.query(longText);

    assertTrue(count > 0);
    assertTrue(connector.hasNext());

    QueryResult qr = connector.next();
    assertNotNull(qr);
    String translated = qr.target.getCodedText();
    assertFalse(translated.isBlank());
    assertFalse(translated.equals(longText), "Should be translated to Japanese");
  }

  @Test
  @DisplayName("Batch query with multiple fragments")
  void testBatchQuery() {
    connector.setLanguages(LocaleId.ENGLISH, LocaleId.SPANISH);

    List<String> texts = List.of(
        "Hello",
        "Goodbye",
        "Thank you");

    List<List<QueryResult>> results = connector.batchQueryText(texts);

    assertNotNull(results);
    assertEquals(3, results.size());

    for (List<QueryResult> resultList : results) {
      assertEquals(1, resultList.size());
      QueryResult qr = resultList.get(0);
      assertNotNull(qr.target);
      assertEquals(100, qr.getFuzzyScore());
    }
  }
}