package com.acumenvelocity.ath.common;
   
   import net.sf.okapi.common.LocaleId;
   import org.junit.jupiter.api.Test;
   import org.junit.jupiter.params.ParameterizedTest;
   import org.junit.jupiter.params.provider.Arguments;
   import org.junit.jupiter.params.provider.MethodSource;
   
   import java.util.List;
  import java.util.stream.Stream;
  
  import static org.junit.jupiter.api.Assertions.*;
  
  class TestOkapiWordBreaker {
  
    private static final LocaleId EN = LocaleId.fromString("en");
  
    @ParameterizedTest(name = "{0}")
    @MethodSource("wordBreakProvider")
    void wordBreaksCorrectly(String description, String input, LocaleId locale,
        List<Integer> expected) {
      List<Integer> result = OkapiWordBreaker.getWordBreakPositions(input, locale);
      assertEquals(expected, result, "Failed for: " + description);
    }
  
    static Stream<Arguments> wordBreakProvider() {
      return Stream.of(
          // Original requirement
          Arguments.of("Hello, world!", "Hello, world!", EN,
              List.of(0, 5, 6, 7, 12, 13)),
  
          // Multiple spaces (each space gets a boundary)
          Arguments.of("Multiple spaces", "Hi  there  friend", EN,
              List.of(0, 2, 3, 4, 9, 10, 11, 17)),
  
          // Full sentence
          Arguments.of("Full sentence", "Hello, world! This is a test.", EN,
              List.of(0, 5, 6, 7, 12, 13, 14, 18, 19, 21, 22, 23, 24, 28, 29)),
  
          // Just a word
          Arguments.of("Single word", "Hello", EN,
              List.of(0, 5)),
  
          // Multiple punctuation
          Arguments.of("Multiple punctuation", "What?!", EN,
              List.of(0, 4, 5, 6)),
  
          // Punctuation with spaces
          Arguments.of("Punctuation with spaces", "Yes , no .", EN,
              List.of(0, 3, 4, 5, 6, 8, 9, 10))
      );
    }
  
    @Test
    void emptyTextReturnsEmptyList() {
      assertTrue(OkapiWordBreaker.getWordBreakPositions("", EN).isEmpty());
      assertTrue(OkapiWordBreaker.getWordBreakPositions(null, EN).isEmpty());
    }
  
    @Test
    void visualizeBreaks() {
      String text = "Hello, world!";
      List<Integer> breaks = OkapiWordBreaker.getWordBreakPositions(text, EN);
      
      System.out.println("Text: \"" + text + "\"");
      System.out.println("Positions: " + breaks);
      System.out.println("\nVisualization:");
      
      for (int pos : breaks) {
        String before = text.substring(0, pos);
        String after = pos < text.length() ? text.substring(pos) : "";
        System.out.printf("Position %2d: \"%s|%s\"%n", pos, before, after);
      }
    }
  }