package services

import "testing"

func TestEngineServiceDownloadStatusTracksCancelAndFinish(t *testing.T) {
	service := NewEngineService(nil)
	status := service.GetModelDownloadStatus()
	if status["active"].(bool) {
		t.Fatal("new service should not report an active model download")
	}

	_, finish, err := service.beginModelDownload("zipformer-ko")
	if err != nil {
		t.Fatal(err)
	}
	status = service.GetModelDownloadStatus()
	if !status["active"].(bool) {
		t.Fatal("active download was not reported")
	}
	if got := status["modelID"]; got != "zipformer-ko" {
		t.Fatalf("modelID = %v, want zipformer-ko", got)
	}
	if status["cancelling"].(bool) {
		t.Fatal("new download should not start in cancelling state")
	}

	service.updateModelDownloadProgress("zipformer-ko", 42.5, 425, 1000)
	status = service.GetModelDownloadStatus()
	if got := status["percent"]; got != 42.5 {
		t.Fatalf("percent = %v, want 42.5", got)
	}
	if got := status["downloaded"]; got != int64(425) {
		t.Fatalf("downloaded = %v, want 425", got)
	}
	if got := status["total"]; got != int64(1000) {
		t.Fatalf("total = %v, want 1000", got)
	}

	if !service.CancelModelDownload("zipformer-ko") {
		t.Fatal("expected active download to be cancelled")
	}
	status = service.GetModelDownloadStatus()
	if !status["cancelling"].(bool) {
		t.Fatal("cancelled download should report cancelling state until finish")
	}

	finish()
	status = service.GetModelDownloadStatus()
	if status["active"].(bool) {
		t.Fatal("finished download should clear active state")
	}
	if got := status["modelID"]; got != "" {
		t.Fatalf("modelID after finish = %v, want empty", got)
	}
}