Add nodeStatus check state helpers

Author: boris-ilijic

pbm/restore/physical.go

@@ -854,6 +854,20 @@ const (
 )
 
 func (n nodeStatus) is(s nodeStatus) bool { return n&s != 0 }
+// isForCleanup returns true when internal node state if so
+// any sort of clean-up: full cleanup or fallbacksync.
+// Status indicates that content of db path contains state
+// which is not correct, and therefore it needs to be
+// discarded.
+func (n nodeStatus) isForCleanup() bool {
+	return n&restoreStared != 0 && n&restoreDone == 0
+}
+
+// isFailed returns true when internal node state didn't reach
+// done state, no matter whether it was started or not.
+func (n nodeStatus) isFailed() bool {
+	return n&restoreDone == 0
+}
 
 // log buffer that will dump content to the storage on restore
 // finish (whether it's successful or not). It also dumps content

pbm/restore/physical_test.go

@@ -13,6 +13,44 @@ import (
 	"github.com/percona/percona-backup-mongodb/pbm/log"
 )
 
+func TestNodeStatus(t *testing.T) {
+	t.Run("isForCleanup", func(t *testing.T) {
+		var progress nodeStatus
+
+		if progress.isForCleanup() {
+			t.Errorf("in initial progress phase, node is not for the cleanup")
+		}
+
+		progress |= restoreStared
+		if !progress.isForCleanup() {
+			t.Errorf("in point of no return phase cleanup should be done")
+		}
+
+		progress |= restoreDone
+		if progress.isForCleanup() {
+			t.Errorf("in done phase, node is not for the cleanup")
+		}
+	})
+
+	t.Run("isFailed", func(t *testing.T) {
+		var progress nodeStatus
+
+		if !progress.isFailed() {
+			t.Errorf("node is int initial progress phase, so it should be marked as failed")
+		}
+
+		progress |= restoreStared
+		if !progress.isFailed() {
+			t.Errorf("node is in started phase, so it should be marked as failed")
+		}
+
+		progress |= restoreDone
+		if progress.isFailed() {
+			t.Errorf("in done phase, node shouldn't be marked as failed")
+		}
+	})
+}
+
 func TestMoveAll(t *testing.T) {
 	t.Run("move all files and dir", func(t *testing.T) {
 		tempSrc, _ := os.MkdirTemp("", "src")