vikingowl
1e83819318
Backend tests: - CPU, memory, disk, network collector tests (existing) - Added temperature, processes, system, AMD GPU collector tests - All tests use mock filesystem data Frontend tests: - Added Vitest with jsdom environment - Tests for formatters (formatBytes, formatUptime, etc.) - Tests for theme store 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
101 lines
2.8 KiB
Go
101 lines
2.8 KiB
Go
package collectors
|
|
|
|
import (
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
)
|
|
|
|
func TestProcessCollector(t *testing.T) {
|
|
tmpDir := t.TempDir()
|
|
procPath := filepath.Join(tmpDir, "proc")
|
|
|
|
// Create mock process directories
|
|
pid1 := filepath.Join(procPath, "1")
|
|
pid2 := filepath.Join(procPath, "100")
|
|
if err := os.MkdirAll(pid1, 0755); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err := os.MkdirAll(pid2, 0755); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
// Mock /proc/1/stat
|
|
// Format: pid (comm) state ppid pgrp session tty_nr tpgid flags minflt cminflt majflt cmajflt utime stime ...
|
|
stat1 := "1 (systemd) S 0 1 1 0 -1 4194560 12345 67890 100 200 1000 500 0 0 20 0 1 0 1 123456789 2000 18446744073709551615 0 0 0 0 0 0 0 0 0 0 0 0 17 0 0 0 0 0 0"
|
|
if err := os.WriteFile(filepath.Join(pid1, "stat"), []byte(stat1), 0644); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
stat2 := "100 (bash) S 1 100 100 0 -1 4194304 5678 1234 50 100 500 250 0 0 20 0 1 0 100 98765432 1500 18446744073709551615 0 0 0 0 0 0 0 0 0 0 0 0 17 1 0 0 0 0 0"
|
|
if err := os.WriteFile(filepath.Join(pid2, "stat"), []byte(stat2), 0644); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
// Mock /proc/1/cmdline
|
|
if err := os.WriteFile(filepath.Join(pid1, "cmdline"), []byte("/usr/lib/systemd/systemd\x00--system\x00"), 0644); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if err := os.WriteFile(filepath.Join(pid2, "cmdline"), []byte("/bin/bash\x00"), 0644); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
// Mock /proc/uptime for CPU calculation
|
|
if err := os.WriteFile(filepath.Join(procPath, "uptime"), []byte("12345.67 98765.43\n"), 0644); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
// Mock /proc/meminfo for memory percentage
|
|
meminfo := `MemTotal: 32000000 kB
|
|
MemFree: 16000000 kB
|
|
MemAvailable: 20000000 kB
|
|
`
|
|
if err := os.WriteFile(filepath.Join(procPath, "meminfo"), []byte(meminfo), 0644); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
collector := NewProcessCollector(procPath)
|
|
stats, err := collector.Collect()
|
|
if err != nil {
|
|
t.Fatalf("Collect failed: %v", err)
|
|
}
|
|
|
|
if stats.Total < 2 {
|
|
t.Errorf("Expected at least 2 processes, got %d", stats.Total)
|
|
}
|
|
|
|
// Check that we have some processes in the lists
|
|
if len(stats.TopByCPU) == 0 && len(stats.TopByMemory) == 0 {
|
|
t.Error("Expected some processes in top lists")
|
|
}
|
|
}
|
|
|
|
func TestProcessCollector_EmptyProc(t *testing.T) {
|
|
tmpDir := t.TempDir()
|
|
procPath := filepath.Join(tmpDir, "proc")
|
|
if err := os.MkdirAll(procPath, 0755); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
// Create meminfo (required)
|
|
meminfo := `MemTotal: 32000000 kB
|
|
`
|
|
if err := os.WriteFile(filepath.Join(procPath, "meminfo"), []byte(meminfo), 0644); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
collector := NewProcessCollector(procPath)
|
|
stats, err := collector.Collect()
|
|
if err != nil {
|
|
t.Fatalf("Collect failed: %v", err)
|
|
}
|
|
|
|
if stats.TopByCPU == nil {
|
|
t.Error("TopByCPU should not be nil")
|
|
}
|
|
if stats.TopByMemory == nil {
|
|
t.Error("TopByMemory should not be nil")
|
|
}
|
|
}
|