mirror of
https://github.com/henrygd/beszel.git
synced 2026-09-21 08:57:48 +02:00
feat: add ZFS monitoring (#2209)
- track pool capacity, health, I/O, scrub status, and vdev errors - report dataset usage and correct ZFS filesystem metrics - add pool charts, detail views, refresh controls, and health alerts - persist pool details and include ZFS usage in disk alerts - support configurable detail intervals and legacy agent compatibility --------- Co-authored-by: hank <hank@henrygd.me>
This commit is contained in:
153
internal/hub/systems/system_zfs_test.go
Normal file
153
internal/hub/systems/system_zfs_test.go
Normal file
@@ -0,0 +1,153 @@
|
||||
//go:build testing
|
||||
|
||||
package systems
|
||||
|
||||
import (
|
||||
"errors"
|
||||
"testing"
|
||||
"time"
|
||||
|
||||
"github.com/blang/semver"
|
||||
"github.com/henrygd/beszel/internal/entities/system"
|
||||
"github.com/henrygd/beszel/internal/entities/zfs"
|
||||
"github.com/henrygd/beszel/internal/hub/expirymap"
|
||||
"github.com/stretchr/testify/assert"
|
||||
"github.com/stretchr/testify/require"
|
||||
)
|
||||
|
||||
func TestSupportsZfsData(t *testing.T) {
|
||||
sys := &System{agentVersion: semver.MustParse("0.18.8")}
|
||||
assert.False(t, sys.supportsZfsData())
|
||||
|
||||
sys.agentVersion = semver.MustParse("0.18.9")
|
||||
assert.True(t, sys.supportsZfsData())
|
||||
}
|
||||
|
||||
func TestRecordZfsFetchResult(t *testing.T) {
|
||||
sm := &SystemManager{zfsFetchMap: expirymap.New[zfsFetchState](time.Hour)}
|
||||
t.Cleanup(sm.zfsFetchMap.StopCleaner)
|
||||
|
||||
sys := &System{
|
||||
Id: "system-1",
|
||||
manager: sm,
|
||||
zfsInterval: time.Hour,
|
||||
}
|
||||
|
||||
// Successful fetch with pools
|
||||
sys.recordZfsFetchResult(nil, 2)
|
||||
state, ok := sm.zfsFetchMap.GetOk(sys.Id)
|
||||
assert.True(t, ok, "expected zfs fetch result to be stored")
|
||||
assert.True(t, state.Successful, "expected successful fetch state to be recorded")
|
||||
|
||||
// Failed fetch
|
||||
sys.recordZfsFetchResult(errors.New("failed"), 0)
|
||||
state, ok = sm.zfsFetchMap.GetOk(sys.Id)
|
||||
assert.True(t, ok, "expected failed zfs fetch state to be stored")
|
||||
assert.False(t, state.Successful, "expected failed zfs fetch state to be marked unsuccessful")
|
||||
|
||||
// Successful fetch but no pools
|
||||
sys.recordZfsFetchResult(nil, 0)
|
||||
state, ok = sm.zfsFetchMap.GetOk(sys.Id)
|
||||
assert.True(t, ok, "expected fetch with zero pools to be stored")
|
||||
assert.False(t, state.Successful, "expected fetch with zero pools to be marked unsuccessful")
|
||||
}
|
||||
|
||||
func TestShouldFetchZfs(t *testing.T) {
|
||||
sm := &SystemManager{zfsFetchMap: expirymap.New[zfsFetchState](time.Hour)}
|
||||
t.Cleanup(sm.zfsFetchMap.StopCleaner)
|
||||
|
||||
sys := &System{
|
||||
Id: "system-1",
|
||||
manager: sm,
|
||||
zfsInterval: time.Hour,
|
||||
}
|
||||
|
||||
assert.True(t, sys.shouldFetchZfs(), "expected initial zfs fetch to be allowed")
|
||||
|
||||
sys.recordZfsFetchResult(errors.New("failed"), 0)
|
||||
assert.False(t, sys.shouldFetchZfs(), "expected zfs fetch to be blocked while interval entry exists")
|
||||
|
||||
sm.zfsFetchMap.Remove(sys.Id)
|
||||
assert.True(t, sys.shouldFetchZfs(), "expected zfs fetch to be allowed after interval entry is cleared")
|
||||
}
|
||||
|
||||
func TestZfsFetchIntervalDefault(t *testing.T) {
|
||||
sys := &System{}
|
||||
assert.Equal(t, time.Hour, sys.zfsFetchInterval())
|
||||
|
||||
sys.zfsInterval = 5 * time.Minute
|
||||
assert.Equal(t, 5*time.Minute, sys.zfsFetchInterval())
|
||||
}
|
||||
|
||||
func TestResetFailedZfsFetchState(t *testing.T) {
|
||||
sm := &SystemManager{zfsFetchMap: expirymap.New[zfsFetchState](time.Hour)}
|
||||
t.Cleanup(sm.zfsFetchMap.StopCleaner)
|
||||
|
||||
sm.zfsFetchMap.Set("system-1", zfsFetchState{LastAttempt: time.Now().UnixMilli(), Successful: false}, time.Hour)
|
||||
sm.resetFailedZfsFetchState("system-1")
|
||||
_, ok := sm.zfsFetchMap.GetOk("system-1")
|
||||
assert.False(t, ok, "expected failed zfs fetch state to be cleared on reconnect")
|
||||
|
||||
sm.zfsFetchMap.Set("system-1", zfsFetchState{LastAttempt: time.Now().UnixMilli(), Successful: true}, time.Hour)
|
||||
sm.resetFailedZfsFetchState("system-1")
|
||||
_, ok = sm.zfsFetchMap.GetOk("system-1")
|
||||
assert.True(t, ok, "expected successful zfs fetch state to be preserved")
|
||||
}
|
||||
|
||||
func TestSaveZfsPoolsCompleteEmptyPrunesFinalPool(t *testing.T) {
|
||||
sys, app := newTestSystemWithHub(t)
|
||||
require.NoError(t, sys.saveZfsPools(&zfs.ZfsData{
|
||||
Complete: true,
|
||||
Pools: []*zfs.PoolDetail{{Name: "tank", Health: "ONLINE"}},
|
||||
}))
|
||||
records, err := app.FindRecordsByFilter("zfs_pools", "system={:system}", "", 0, 0, map[string]any{"system": sys.Id})
|
||||
require.NoError(t, err)
|
||||
require.Len(t, records, 1)
|
||||
assert.False(t, records[0].GetDateTime("details_updated").Time().IsZero())
|
||||
|
||||
require.NoError(t, sys.saveZfsPools(&zfs.ZfsData{Complete: true}))
|
||||
records, err = app.FindRecordsByFilter("zfs_pools", "system={:system}", "", 0, 0, map[string]any{"system": sys.Id})
|
||||
require.NoError(t, err)
|
||||
assert.Empty(t, records)
|
||||
}
|
||||
|
||||
func TestSaveZfsPoolsIncompletePreservesRecords(t *testing.T) {
|
||||
sys, app := newTestSystemWithHub(t)
|
||||
require.NoError(t, sys.saveZfsPools(&zfs.ZfsData{
|
||||
Complete: true,
|
||||
Pools: []*zfs.PoolDetail{{Name: "tank", Health: "ONLINE"}},
|
||||
}))
|
||||
assert.ErrorIs(t, sys.saveZfsPools(&zfs.ZfsData{}), errIncompleteZfsData)
|
||||
records, err := app.FindRecordsByFilter("zfs_pools", "system={:system}", "", 0, 0, map[string]any{"system": sys.Id})
|
||||
require.NoError(t, err)
|
||||
assert.Len(t, records, 1)
|
||||
}
|
||||
|
||||
func TestSyncZfsPoolHealthWritesOnlyTransitions(t *testing.T) {
|
||||
sys, app := newTestSystemWithHub(t)
|
||||
collection, err := app.FindCachedCollectionByNameOrId("zfs_pools")
|
||||
require.NoError(t, err)
|
||||
|
||||
require.NoError(t, sys.syncZfsPoolHealth(app, map[string]*system.ZfsPool{
|
||||
"tank": {Total: 100, Used: 25, Health: "ONLINE"},
|
||||
}))
|
||||
record, err := app.FindRecordById(collection, makeStableHashId(sys.Id, "tank"))
|
||||
require.NoError(t, err)
|
||||
firstUpdated := record.GetDateTime("updated")
|
||||
assert.Equal(t, "ONLINE", record.GetString("health"))
|
||||
assert.EqualValues(t, 100*1024*1024*1024, record.GetInt("size"))
|
||||
|
||||
require.NoError(t, sys.syncZfsPoolHealth(app, map[string]*system.ZfsPool{
|
||||
"tank": {Total: 100, Used: 30, Health: "ONLINE"},
|
||||
}))
|
||||
record, err = app.FindRecordById(collection, record.Id)
|
||||
require.NoError(t, err)
|
||||
assert.Equal(t, firstUpdated, record.GetDateTime("updated"))
|
||||
|
||||
require.NoError(t, sys.syncZfsPoolHealth(app, map[string]*system.ZfsPool{
|
||||
"tank": {Total: 100, Used: 30, Health: "DEGRADED"},
|
||||
}))
|
||||
record, err = app.FindRecordById(collection, record.Id)
|
||||
require.NoError(t, err)
|
||||
assert.Equal(t, "DEGRADED", record.GetString("health"))
|
||||
}
|
||||
Reference in New Issue
Block a user