diff --git a/internal/monitoring/monitor.go b/internal/monitoring/monitor.go index 92cc85f28..ebc579e4c 100644 --- a/internal/monitoring/monitor.go +++ b/internal/monitoring/monitor.go @@ -16,6 +16,7 @@ import ( "github.com/rcourtman/pulse-go-rewrite/internal/config" "github.com/rcourtman/pulse-go-rewrite/internal/discovery" "github.com/rcourtman/pulse-go-rewrite/internal/errors" + "github.com/rcourtman/pulse-go-rewrite/internal/mock" "github.com/rcourtman/pulse-go-rewrite/internal/models" "github.com/rcourtman/pulse-go-rewrite/internal/notifications" "github.com/rcourtman/pulse-go-rewrite/internal/websocket" @@ -2118,6 +2119,10 @@ func (m *Monitor) pollPBSInstance(ctx context.Context, instanceName string, clie // GetState returns the current state func (m *Monitor) GetState() models.StateSnapshot { + // Check if mock mode is enabled + if mock.IsMockEnabled() { + return mock.GetMockState() + } return m.state.GetSnapshot() }