mirror of
https://github.com/Dvorinka/beszel.git
synced 2026-06-03 21:02:56 +00:00
Tomas Dvorak
95 lines
3.2 KiB
Go
95 lines
3.2 KiB
Go
//go:build testing
|
|
|
|
package systems
|
|
|
|
import (
|
|
"errors"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/henrygd/beszel/internal/hub/expirymap"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestRecordSmartFetchResult(t *testing.T) {
|
|
sm := &SystemManager{smartFetchMap: expirymap.New[smartFetchState](time.Hour)}
|
|
t.Cleanup(sm.smartFetchMap.StopCleaner)
|
|
|
|
sys := &System{
|
|
Id: "system-1",
|
|
manager: sm,
|
|
smartInterval: time.Hour,
|
|
}
|
|
|
|
// Successful fetch with devices
|
|
sys.recordSmartFetchResult(nil, 5)
|
|
state, ok := sm.smartFetchMap.GetOk(sys.Id)
|
|
assert.True(t, ok, "expected smart fetch result to be stored")
|
|
assert.True(t, state.Successful, "expected successful fetch state to be recorded")
|
|
|
|
// Failed fetch
|
|
sys.recordSmartFetchResult(errors.New("failed"), 0)
|
|
state, ok = sm.smartFetchMap.GetOk(sys.Id)
|
|
assert.True(t, ok, "expected failed smart fetch state to be stored")
|
|
assert.False(t, state.Successful, "expected failed smart fetch state to be marked unsuccessful")
|
|
|
|
// Successful fetch but no devices
|
|
sys.recordSmartFetchResult(nil, 0)
|
|
state, ok = sm.smartFetchMap.GetOk(sys.Id)
|
|
assert.True(t, ok, "expected fetch with zero devices to be stored")
|
|
assert.False(t, state.Successful, "expected fetch with zero devices to be marked unsuccessful")
|
|
}
|
|
|
|
func TestShouldFetchSmart(t *testing.T) {
|
|
sm := &SystemManager{smartFetchMap: expirymap.New[smartFetchState](time.Hour)}
|
|
t.Cleanup(sm.smartFetchMap.StopCleaner)
|
|
|
|
sys := &System{
|
|
Id: "system-1",
|
|
manager: sm,
|
|
smartInterval: time.Hour,
|
|
}
|
|
|
|
assert.True(t, sys.shouldFetchSmart(), "expected initial smart fetch to be allowed")
|
|
|
|
sys.recordSmartFetchResult(errors.New("failed"), 0)
|
|
assert.False(t, sys.shouldFetchSmart(), "expected smart fetch to be blocked while interval entry exists")
|
|
|
|
sm.smartFetchMap.Remove(sys.Id)
|
|
assert.True(t, sys.shouldFetchSmart(), "expected smart fetch to be allowed after interval entry is cleared")
|
|
}
|
|
|
|
func TestShouldFetchSmart_IgnoresExtendedTTLWhenFetchIsDue(t *testing.T) {
|
|
sm := &SystemManager{smartFetchMap: expirymap.New[smartFetchState](time.Hour)}
|
|
t.Cleanup(sm.smartFetchMap.StopCleaner)
|
|
|
|
sys := &System{
|
|
Id: "system-1",
|
|
manager: sm,
|
|
smartInterval: time.Hour,
|
|
}
|
|
|
|
sm.smartFetchMap.Set(sys.Id, smartFetchState{
|
|
LastAttempt: time.Now().Add(-2 * time.Hour).UnixMilli(),
|
|
Successful: true,
|
|
}, 10*time.Minute)
|
|
sm.smartFetchMap.UpdateExpiration(sys.Id, 3*time.Hour)
|
|
|
|
assert.True(t, sys.shouldFetchSmart(), "expected fetch time to take precedence over updated TTL")
|
|
}
|
|
|
|
func TestResetFailedSmartFetchState(t *testing.T) {
|
|
sm := &SystemManager{smartFetchMap: expirymap.New[smartFetchState](time.Hour)}
|
|
t.Cleanup(sm.smartFetchMap.StopCleaner)
|
|
|
|
sm.smartFetchMap.Set("system-1", smartFetchState{LastAttempt: time.Now().UnixMilli(), Successful: false}, time.Hour)
|
|
sm.resetFailedSmartFetchState("system-1")
|
|
_, ok := sm.smartFetchMap.GetOk("system-1")
|
|
assert.False(t, ok, "expected failed smart fetch state to be cleared on reconnect")
|
|
|
|
sm.smartFetchMap.Set("system-1", smartFetchState{LastAttempt: time.Now().UnixMilli(), Successful: true}, time.Hour)
|
|
sm.resetFailedSmartFetchState("system-1")
|
|
_, ok = sm.smartFetchMap.GetOk("system-1")
|
|
assert.True(t, ok, "expected successful smart fetch state to be preserved")
|
|
}
|