fix: respect streak

soerenschneider · Jul 19, 2023 · 1b97738 · 1b97738
1 parent 69169a5
commit 1b97738
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/internal/agent/state/ok.go b/internal/agent/state/ok.go
@@ -14,7 +14,11 @@ func (s *NoRebootNeeded) Success() {
 }
 
 func (s *NoRebootNeeded) Failure() {
-	s.stateful.SetState(&RebootNeeded{stateful: s.stateful})
+	if s.stateful.StreakUntilRebootState() > 1 {
+		s.stateful.SetState(NewUncertainState(s.stateful))
+	} else {
+		s.stateful.SetState(&RebootNeeded{stateful: s.stateful})
+	}
 }
 
 func (s *NoRebootNeeded) Error(err error) {