chore: more fixes

Signed-off-by: Ajay Mishra <[email protected]>

Author: XRFXLP

health-monitors/syslog-health-monitor/Dockerfile

@@ -59,4 +59,4 @@ COPY --from=builder /go/src/nvsentinel/health-monitors/syslog-health-monitor/sys
 RUN apt-get update && \
     apt-get install -y libsystemd0 liblz4-1 libzstd1
 
-ENTRYPOINT ["/app/syslog-health-monitor"] 
+ENTRYPOINT ["/app/syslog-health-monitor"] 

tests/fault_management_test.go

@@ -1,3 +1,17 @@
+// Copyright (c) 2025, NVIDIA CORPORATION.  All rights reserved.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
 package tests
 
 import (

tests/helpers/fault_quarantine.go

@@ -49,6 +49,45 @@ type QuarantineAssertion struct {
 	ExpectAnnotation bool
 }
 
+func ApplyQuarantineConfig(ctx context.Context, t *testing.T, c *envconf.Config, configMapPath string) context.Context {
+	t.Helper()
+	client, err := c.NewClient()
+	require.NoError(t, err)
+
+	t.Log("Backing up current fault-quarantine configmap")
+	backupData, err := BackupConfigMap(ctx, client, "fault-quarantine-config", NVSentinelNamespace)
+	require.NoError(t, err)
+	ctx = context.WithValue(ctx, CELKeyConfigMapBackup, backupData)
+
+	t.Logf("Applying test configmap: %s", configMapPath)
+	err = createConfigMapFromFilePath(ctx, client, configMapPath, "fault-quarantine-config", NVSentinelNamespace)
+	require.NoError(t, err)
+
+	t.Log("Restarting fault-quarantine deployment to load configuration")
+	err = RestartDeployment(ctx, t, client, "nvsentinel-fault-quarantine", NVSentinelNamespace)
+	require.NoError(t, err)
+
+	return ctx
+}
+
+func RestoreQuarantineConfig(ctx context.Context, t *testing.T, c *envconf.Config) {
+	t.Helper()
+	client, err := c.NewClient()
+	require.NoError(t, err)
+
+	backupDataVal := ctx.Value(CELKeyConfigMapBackup)
+	if backupDataVal != nil {
+		backupData := backupDataVal.([]byte)
+		t.Log("Restoring fault-quarantine configmap from memory")
+		err = createConfigMapFromBytes(ctx, client, backupData, "fault-quarantine-config", NVSentinelNamespace)
+		require.NoError(t, err)
+
+		t.Log("Restarting fault-quarantine deployment to load restored configuration")
+		err = RestartDeployment(ctx, t, client, "nvsentinel-fault-quarantine", NVSentinelNamespace)
+		require.NoError(t, err)
+	}
+}
+
 func SetupQuarantineTest(ctx context.Context, t *testing.T, c *envconf.Config, configMapPath string) (context.Context, *QuarantineTestContext) {
 	ctx, testCtx, _ := SetupQuarantineTestWithOptions(ctx, t, c, configMapPath, nil)
 	return ctx, testCtx

tests/helpers/kube.go

@@ -1099,7 +1099,7 @@ func AssertPodsNeverDeleted(ctx context.Context, t *testing.T, client klient.Cli
 			}
 		}
 		return false
-	}, 30*WaitInterval, WaitInterval, "pods should not be deleted")
+	}, 15*time.Second, 5*time.Second, "pods should not be deleted")
 	t.Logf("All %d pods remain running in namespace %s", len(podNames), namespace)
 }
 

tests/helpers/node_drainer.go

@@ -20,6 +20,7 @@ import (
 	"strings"
 	"testing"
 
+	"github.com/nvidia/nvsentinel/commons/pkg/statemanager"
 	"github.com/stretchr/testify/assert"
 	"github.com/stretchr/testify/require"
 	v1 "k8s.io/api/core/v1"
@@ -93,6 +94,25 @@ func SetupNodeDrainerTest(ctx context.Context, t *testing.T, c *envconf.Config,
 	return ctx, testCtx
 }
 
+func ResetNodeAndTriggerDrain(ctx context.Context, t *testing.T, client klient.Client, nodeName, namespace string) ([]string, string) {
+	t.Helper()
+
+	SendHealthyEvent(ctx, t, nodeName)
+	WaitForNodesCordonState(ctx, t, client, []string{nodeName}, false)
+
+	podNames := CreatePodsFromTemplate(ctx, t, client, "data/busybox-pods.yaml", nodeName, namespace)
+	WaitForPodsRunning(ctx, t, client, namespace, podNames)
+
+	event := NewHealthEvent(nodeName).
+		WithErrorCode("79").
+		WithMessage("GPU Fallen off the bus")
+	tempFile := SendHealthEvent(ctx, t, event)
+
+	WaitForNodeLabel(ctx, t, client, nodeName, statemanager.NVSentinelStateLabelKey, DrainingLabelValue)
+
+	return podNames, tempFile
+}
+
 func TeardownNodeDrainer(ctx context.Context, t *testing.T, c *envconf.Config) context.Context {
 	t.Helper()
 	client, err := c.NewClient()