openclaw/apps/macos/Tests/OpenClawIPCTests/VoicePushToTalkTests.swift

import AppKit
import AVFoundation
import Testing
@testable import OpenClaw

struct VoicePushToTalkTests {
    @Test func `speech normalizer passes through mono buffers`() throws {
        let format = try #require(AVAudioFormat(
            commonFormat: .pcmFormatFloat32,
            sampleRate: 16_000,
            channels: 1,
            interleaved: false))
        let buffer = try #require(AVAudioPCMBuffer(pcmFormat: format, frameCapacity: 4))
        buffer.frameLength = 4

        let normalized = SpeechAudioBufferNormalizer.speechCompatibleBuffer(from: buffer)

        #expect(normalized === buffer)
    }

    @Test func `speech normalizer downmixes multichannel float buffers to mono`() throws {
        var layout = AudioChannelLayout()
        layout.mChannelLayoutTag = kAudioChannelLayoutTag_Quadraphonic
        let channelLayout = AVAudioChannelLayout(layout: &layout)
        let format = AVAudioFormat(
            commonFormat: .pcmFormatFloat32,
            sampleRate: 16_000,
            interleaved: false,
            channelLayout: channelLayout)
        let buffer = try #require(AVAudioPCMBuffer(pcmFormat: format, frameCapacity: 2))
        buffer.frameLength = 2
        let channels = try #require(buffer.floatChannelData)
        for frame in 0..<2 {
            channels[0][frame] = 1
            channels[1][frame] = 3
            channels[2][frame] = 5
            channels[3][frame] = 7
        }

        let normalized = SpeechAudioBufferNormalizer.speechCompatibleBuffer(from: buffer)

        #expect(normalized.format.channelCount == 1)
        #expect(normalized.frameLength == 2)
        let output = try #require(normalized.floatChannelData?[0])
        #expect(output[0] == 4)
        #expect(output[1] == 4)
    }

    @Test func `delta trims committed prefix`() {
        let delta = VoiceOverlayTextFormatting.delta(after: "hello ", current: "hello world again")
        #expect(delta == "world again")
    }

    @Test func `delta falls back when prefix differs`() {
        let delta = VoiceOverlayTextFormatting.delta(after: "goodbye", current: "hello world")
        #expect(delta == "hello world")
    }

    @Test func `partial transcript distinguishes volatile text`() throws {
        let text = VoiceOverlayTextFormatting.makeAttributed(committed: "a", volatile: "b", isFinal: false)
        let committed = try #require(text.attribute(.foregroundColor, at: 0, effectiveRange: nil) as? NSColor)
        let volatile = try #require(text.attribute(.foregroundColor, at: 1, effectiveRange: nil) as? NSColor)
        #expect(committed != volatile)
    }

    @Test func `final transcript uses one text color`() throws {
        let text = VoiceOverlayTextFormatting.makeAttributed(committed: "a", volatile: "b", isFinal: true)
        let committed = try #require(text.attribute(.foregroundColor, at: 0, effectiveRange: nil) as? NSColor)
        let volatile = try #require(text.attribute(.foregroundColor, at: 1, effectiveRange: nil) as? NSColor)
        #expect(committed == volatile)
    }
}