import XCTest

@testable import Transui

final class TranscriptCleanerTests: XCTestCase {
  func testRemovesAsTextPrefix() {
    XCTAssertEqual(
      TranscriptCleaner.clean("language English<astext>Hello there"),
      "Hello there"
    )
  }

  func testRemovesAsrTextPrefix() {
    XCTAssertEqual(
      TranscriptCleaner.clean("language zh <asr_text>你好"),
      "你好"
    )
  }

  func testRemovesPrefixAfterWhitespaceAndBom() {
    XCTAssertEqual(
      TranscriptCleaner.clean("\u{FEFF} \n\tlanguage English<astext>Hello"),
      "Hello"
    )
  }

  func testPreservesOrdinaryTranscript() {
    XCTAssertEqual(TranscriptCleaner.clean("language is useful"), "language is useful")
  }
}