42 lines
1.2 KiB
Go
42 lines
1.2 KiB
Go
package httpapi
|
|
|
|
import (
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
"testing"
|
|
|
|
"rk3588sys/agent/internal/modelstore"
|
|
)
|
|
|
|
func TestHandleModelsStatusReturnsInstalledModels(t *testing.T) {
|
|
dir := t.TempDir()
|
|
store := modelstore.New(dir, 8)
|
|
itemDir := store.FilesDir()
|
|
if err := os.MkdirAll(itemDir, 0o755); err != nil {
|
|
t.Fatalf("MkdirAll: %v", err)
|
|
}
|
|
modelPath := filepath.Join(itemDir, "face_det_scrfd_500m_640_rk3588__abc.rknn")
|
|
if err := os.WriteFile(modelPath, []byte("abc"), 0o644); err != nil {
|
|
t.Fatalf("WriteFile: %v", err)
|
|
}
|
|
if err := os.WriteFile(store.ManifestPath(), []byte(`{"items":[{"name":"face_det_scrfd_500m_640_rk3588","sha256":"abc","path":"`+filepath.ToSlash(modelPath)+`","size":3,"mtime_ms":1}]}`), 0o644); err != nil {
|
|
t.Fatalf("WriteFile manifest: %v", err)
|
|
}
|
|
|
|
srv := &Server{store: store}
|
|
req := httptest.NewRequest(http.MethodGet, "/v1/models/status", nil)
|
|
rr := httptest.NewRecorder()
|
|
|
|
srv.handleModelsStatus(rr, req)
|
|
|
|
if rr.Code != http.StatusOK {
|
|
t.Fatalf("expected 200, got %d: %s", rr.Code, rr.Body.String())
|
|
}
|
|
if !strings.Contains(rr.Body.String(), "face_det_scrfd_500m_640_rk3588") {
|
|
t.Fatalf("expected response to include model name, got %s", rr.Body.String())
|
|
}
|
|
}
|