MnemonicInputNormalizerTest.java

package com.example.bip39.normalize;

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

import com.example.bip39.api.DefaultBip39Service;
import com.example.bip39.error.Bip39ErrorCode;
import com.example.bip39.error.Bip39Exception;
import com.example.bip39.model.ValidationResult;
import org.junit.jupiter.api.Test;

class MnemonicInputNormalizerTest {

  private static final String ZERO_ENTROPY_MNEMONIC =
      "abandon abandon abandon abandon abandon abandon abandon abandon abandon abandon abandon"
          + " about";

  @Test
  void normalizesWhitespaceAndCaseForEnglishProfile() {
    String rawMnemonic =
        "  ABANDON\tabandon\nABANDON\rabandon  abandon\tabandon\nabandon\rabandon"
            + "  abandon\tabandon\nABANDON\rABOUT  ";

    assertEquals(
        ZERO_ENTROPY_MNEMONIC, MnemonicInputNormalizer.normalizeMnemonicInput(rawMnemonic));
  }

  @Test
  void appliesNfkdBeforeLowercasing() {
    String rawMnemonic = "\tＡＢＡＮＤＯＮ\nＡＢＯＵＴ\r";

    assertEquals("abandon about", MnemonicInputNormalizer.normalizeMnemonicInput(rawMnemonic));
  }

  @Test
  void letsUiLayerExplicitlyNormalizeBeforeCallingStrictCoreApis() {
    DefaultBip39Service service = new DefaultBip39Service();
    String rawMnemonic =
        "  ABANDON\tabandon\nABANDON\rabandon  abandon\tabandon\nabandon\rabandon"
            + "  abandon\tabandon\nABANDON\rABOUT  ";

    Bip39Exception exception =
        assertThrows(Bip39Exception.class, () -> service.mnemonicToEntropy(rawMnemonic));
    assertEquals(Bip39ErrorCode.ERR_INVALID_MNEMONIC_FORMAT, exception.getErrorCode());

    ValidationResult rawValidation = service.validateMnemonic(rawMnemonic);
    assertFalse(rawValidation.ok());
    assertEquals(Bip39ErrorCode.ERR_INVALID_MNEMONIC_FORMAT, rawValidation.errorCode());

    String normalizedMnemonic = MnemonicInputNormalizer.normalizeMnemonicInput(rawMnemonic);
    assertArrayEquals(new byte[16], service.mnemonicToEntropy(normalizedMnemonic));

    ValidationResult normalizedValidation = service.validateMnemonic(normalizedMnemonic);
    assertTrue(normalizedValidation.ok());
    assertEquals(ZERO_ENTROPY_MNEMONIC, normalizedValidation.normalizedMnemonic());
  }
}