jim800121chen
c54f16fca0
local-tool/: visionA-local desktop app
- M1: Wails shell + Go server + Next.js UI + Mock mode (macOS dmg ready)
- M2: i18n (zh-TW/en) + Settings 4-tab refactor
- M3: Embedded Python 3.12 runtime (python-build-standalone) + KneronPLUS wheels
- M4: Windows Inno Setup script (build on Windows runner)
- M5: Linux AppImage script + udev rule (build on Linux runner)
- M6: ffmpeg (GPL, pending legal review) + yt-dlp bundled
- Lifecycle: watchServer health check, fatal native dialog,
Wails IPC raise endpoint, stale process cleanup
.autoflow/: full PRD / Design Spec / Architecture / Testing docs
(4 rounds tri-party discussion + cross review)
.github/workflows/: macOS / Windows / Linux build CI
Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
94 lines
2.7 KiB
Go
94 lines
2.7 KiB
Go
package device
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"visiona-local/server/internal/driver"
|
|
)
|
|
|
|
type testDriver struct {
|
|
info driver.DeviceInfo
|
|
connected bool
|
|
}
|
|
|
|
func (d *testDriver) Info() driver.DeviceInfo { return d.info }
|
|
func (d *testDriver) Connect() error { d.connected = true; d.info.Status = driver.StatusConnected; return nil }
|
|
func (d *testDriver) Disconnect() error { d.connected = false; d.info.Status = driver.StatusDisconnected; return nil }
|
|
func (d *testDriver) IsConnected() bool { return d.connected }
|
|
func (d *testDriver) Flash(_ string, _ chan<- driver.FlashProgress) error { return nil }
|
|
func (d *testDriver) StartInference() error { return nil }
|
|
func (d *testDriver) StopInference() error { return nil }
|
|
func (d *testDriver) ReadInference() (*driver.InferenceResult, error) { return nil, nil }
|
|
func (d *testDriver) RunInference(_ []byte) (*driver.InferenceResult, error) { return nil, nil }
|
|
func (d *testDriver) GetModelInfo() (*driver.ModelInfo, error) { return nil, nil }
|
|
|
|
func TestNewManager_MockMode(t *testing.T) {
|
|
registry := NewRegistry()
|
|
mgr := NewManager(registry, true, 2, "")
|
|
|
|
devices := mgr.ListDevices()
|
|
if len(devices) != 2 {
|
|
t.Errorf("NewManager mock mode: got %d devices, want 2", len(devices))
|
|
}
|
|
}
|
|
|
|
func TestManager_ListDevices(t *testing.T) {
|
|
registry := NewRegistry()
|
|
mgr := NewManager(registry, false, 0, "")
|
|
|
|
mgr.sessions["test-1"] = NewSession(&testDriver{
|
|
info: driver.DeviceInfo{ID: "test-1", Name: "Test Device", Type: "KL720", Status: driver.StatusDetected},
|
|
})
|
|
|
|
devices := mgr.ListDevices()
|
|
if len(devices) != 1 {
|
|
t.Errorf("ListDevices() = %d, want 1", len(devices))
|
|
}
|
|
}
|
|
|
|
func TestManager_GetDevice(t *testing.T) {
|
|
registry := NewRegistry()
|
|
mgr := NewManager(registry, false, 0, "")
|
|
mgr.sessions["test-1"] = NewSession(&testDriver{
|
|
info: driver.DeviceInfo{ID: "test-1"},
|
|
})
|
|
|
|
t.Run("existing device", func(t *testing.T) {
|
|
s, err := mgr.GetDevice("test-1")
|
|
if err != nil {
|
|
t.Errorf("GetDevice() error = %v", err)
|
|
}
|
|
if s == nil {
|
|
t.Error("GetDevice() returned nil session")
|
|
}
|
|
})
|
|
|
|
t.Run("non-existing device", func(t *testing.T) {
|
|
_, err := mgr.GetDevice("test-999")
|
|
if err == nil {
|
|
t.Error("GetDevice() expected error for non-existing device")
|
|
}
|
|
})
|
|
}
|
|
|
|
func TestManager_Connect(t *testing.T) {
|
|
registry := NewRegistry()
|
|
mgr := NewManager(registry, false, 0, "")
|
|
td := &testDriver{info: driver.DeviceInfo{ID: "test-1", Status: driver.StatusDetected}}
|
|
mgr.sessions["test-1"] = NewSession(td)
|
|
|
|
// Drain event bus in background
|
|
go func() {
|
|
for range mgr.Events() {
|
|
}
|
|
}()
|
|
|
|
err := mgr.Connect("test-1")
|
|
if err != nil {
|
|
t.Errorf("Connect() error = %v", err)
|
|
}
|
|
if !td.connected {
|
|
t.Error("Connect() did not connect device")
|
|
}
|
|
}
|