//go:build darwin

package engine

import (
	"path/filepath"
	"testing"

	"voicesnap/internal/model"
)

func TestOfflineConfigForParakeetUsesNemoTransducer(t *testing.T) {
	root := t.TempDir()
	files := map[string]string{
		"encoder": filepath.Join(root, "encoder.int8.onnx"),
		"decoder": filepath.Join(root, "decoder.int8.onnx"),
		"joiner":  filepath.Join(root, "joiner.int8.onnx"),
		"tokens":  filepath.Join(root, "tokens.txt"),
	}
	resolved := model.ResolvedModel{
		ModelID:     model.ParakeetModelID,
		BackendKind: model.BackendNemoTransducer,
		Profile: model.ModelProfile{
			DisplayName:   "Parakeet English",
			BackendKind:   model.BackendNemoTransducer,
			ProviderOrder: []string{"cpu"},
			NumThreads:    4,
		},
		Files: files,
	}

	config, err := offlineConfigForResolvedModel(resolved, "cpu")
	if err != nil {
		t.Fatal(err)
	}
	if config.ModelConfig.Transducer.Encoder != files["encoder"] {
		t.Fatalf("encoder = %q, want %q", config.ModelConfig.Transducer.Encoder, files["encoder"])
	}
	if config.ModelConfig.Transducer.Decoder != files["decoder"] {
		t.Fatalf("decoder = %q, want %q", config.ModelConfig.Transducer.Decoder, files["decoder"])
	}
	if config.ModelConfig.Transducer.Joiner != files["joiner"] {
		t.Fatalf("joiner = %q, want %q", config.ModelConfig.Transducer.Joiner, files["joiner"])
	}
	if config.ModelConfig.Tokens != files["tokens"] {
		t.Fatalf("tokens = %q, want %q", config.ModelConfig.Tokens, files["tokens"])
	}
	if config.ModelConfig.ModelType != model.BackendNemoTransducer {
		t.Fatalf("model type = %q, want %q", config.ModelConfig.ModelType, model.BackendNemoTransducer)
	}
}