tian
29d9ef5d0d
仓库结构: edge/ 设备端（原根目录设备端代码整体移入） control/ 管理端（清理后） docs/ 文档（PRD 移入 design/） README.md 根导航（新增） 清理: - control/.brainstorm 临时草稿删除 - control 根级重复文档(API表/PRD_04)并入 docs/design/ - control/plan.md -> docs/implementation/control-plan.md - control/safesightd-linux-arm64 二进制取消版本控制(.gitignore) - edge/transform 模型转换产物归入 models/，onnx/pt 大源文件取消跟踪(.gitignore) - Readme.md(PRD) -> docs/design/PRD_Product_v1.2.md（避开 README 大小写冲突） 更新: - 根 README.md 导航、docs/README.md 文档索引 - deployment.md/检查表路径加 edge/ 前缀 - .gitignore 重写（edge/control 分区规则）
64 lines
1.8 KiB
Go
64 lines
1.8 KiB
Go
package modelstore
|
|
|
|
import (
|
|
"crypto/sha256"
|
|
"encoding/hex"
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
)
|
|
|
|
func TestListInstalledModelsScansModelsDirWithoutManifest(t *testing.T) {
|
|
dir := t.TempDir()
|
|
store := New(dir, 8)
|
|
if err := os.MkdirAll(store.ModelsDir, 0o755); err != nil {
|
|
t.Fatalf("MkdirAll: %v", err)
|
|
}
|
|
content := []byte("abc")
|
|
modelPath := filepath.Join(store.ModelsDir, "face_det_scrfd_500m_640_rk3588.rknn")
|
|
if err := os.WriteFile(modelPath, content, 0o644); err != nil {
|
|
t.Fatalf("WriteFile: %v", err)
|
|
}
|
|
if err := os.WriteFile(filepath.Join(store.ModelsDir, "readme.txt"), []byte("ignore"), 0o644); err != nil {
|
|
t.Fatalf("WriteFile readme: %v", err)
|
|
}
|
|
|
|
items, err := store.ListInstalledModels()
|
|
if err != nil {
|
|
t.Fatalf("ListInstalledModels: %v", err)
|
|
}
|
|
if len(items) != 1 {
|
|
t.Fatalf("unexpected installed models: %#v", items)
|
|
}
|
|
expectedSHA := sha256.Sum256(content)
|
|
if items[0].Name != "face_det_scrfd_500m_640_rk3588" {
|
|
t.Fatalf("unexpected model name: %#v", items[0])
|
|
}
|
|
if items[0].FileName != "face_det_scrfd_500m_640_rk3588.rknn" {
|
|
t.Fatalf("unexpected file name: %#v", items[0])
|
|
}
|
|
if items[0].Sha256 != hex.EncodeToString(expectedSHA[:]) {
|
|
t.Fatalf("unexpected sha256: %#v", items[0])
|
|
}
|
|
}
|
|
|
|
func TestListInstalledModelsScansUploadedFilesDir(t *testing.T) {
|
|
dir := t.TempDir()
|
|
store := New(dir, 8)
|
|
if err := os.MkdirAll(store.FilesDir(), 0o755); err != nil {
|
|
t.Fatalf("MkdirAll: %v", err)
|
|
}
|
|
modelPath := filepath.Join(store.FilesDir(), "face_det_scrfd_500m_640_rk3588__abc.rknn")
|
|
if err := os.WriteFile(modelPath, []byte("abc"), 0o644); err != nil {
|
|
t.Fatalf("WriteFile: %v", err)
|
|
}
|
|
|
|
items, err := store.ListInstalledModels()
|
|
if err != nil {
|
|
t.Fatalf("ListInstalledModels: %v", err)
|
|
}
|
|
if len(items) != 1 || items[0].FileName == "" {
|
|
t.Fatalf("unexpected installed models: %#v", items)
|
|
}
|
|
}
|