Skip to content

Commit 3314372

Browse files
committed
use testutils.GenerateTestNodeName everywhere possible
Signed-off-by: Davanum Srinivas <[email protected]>
1 parent 36edd76 commit 3314372

File tree

3 files changed

+10
-8
lines changed

3 files changed

+10
-8
lines changed

fault-quarantine/pkg/reconciler/reconciler_e2e_test.go

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3672,7 +3672,7 @@ func TestE2E_ManualUncordonWithCancellation(t *testing.T) {
36723672
ctx, cancel := context.WithTimeout(e2eTestContext, 30*time.Second)
36733673
defer cancel()
36743674

3675-
nodeName := "e2e-manual-uncordon-" + generateShortTestID()
3675+
nodeName := testutils.GenerateTestNodeName("e2e-manual-uncordon")
36763676
createE2ETestNode(ctx, t, nodeName, nil, nil, nil, false)
36773677
defer func() {
36783678
_ = e2eTestClient.CoreV1().Nodes().Delete(ctx, nodeName, metav1.DeleteOptions{})
@@ -3757,7 +3757,7 @@ func TestE2E_ManualUncordonMultipleEvents(t *testing.T) {
37573757
ctx, cancel := context.WithTimeout(e2eTestContext, 30*time.Second)
37583758
defer cancel()
37593759

3760-
nodeName := "e2e-manual-multi-" + generateShortTestID()
3760+
nodeName := testutils.GenerateTestNodeName("e2e-manual-multi")
37613761
createE2ETestNode(ctx, t, nodeName, nil, nil, nil, false)
37623762
defer func() {
37633763
_ = e2eTestClient.CoreV1().Nodes().Delete(ctx, nodeName, metav1.DeleteOptions{})

fault-remediation/pkg/reconciler/reconciler_e2e_test.go

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -30,6 +30,7 @@ import (
3030
"github.com/nvidia/nvsentinel/fault-remediation/pkg/config"
3131
"github.com/nvidia/nvsentinel/fault-remediation/pkg/crstatus"
3232
"github.com/nvidia/nvsentinel/store-client/pkg/datastore"
33+
"github.com/nvidia/nvsentinel/store-client/pkg/testutils"
3334

3435
"github.com/nvidia/nvsentinel/commons/pkg/statemanager"
3536
"github.com/prometheus/client_golang/prometheus"
@@ -934,7 +935,7 @@ func TestReconciler_CancelledEventCleansAnnotation(t *testing.T) {
934935
ctx, cancel := context.WithTimeout(testContext, 30*time.Second)
935936
defer cancel()
936937

937-
nodeName := "test-cancelled-clean-" + "test-node-123"
938+
nodeName := testutils.GenerateTestNodeName("test-cancelled-clean")
938939
createTestNode(ctx, nodeName, nil, map[string]string{"test": "label"})
939940
defer func() {
940941
_ = testClient.CoreV1().Nodes().Delete(ctx, nodeName, metav1.DeleteOptions{})
@@ -1039,7 +1040,7 @@ func TestReconciler_CancelledEventClearsAllGroups(t *testing.T) {
10391040
ctx, cancel := context.WithTimeout(testContext, 30*time.Second)
10401041
defer cancel()
10411042

1042-
nodeName := "test-cancelled-all-" + "test-node-123"
1043+
nodeName := testutils.GenerateTestNodeName("test-cancelled-all")
10431044
createTestNode(ctx, nodeName, nil, map[string]string{"test": "label"})
10441045
defer func() {
10451046
_ = testClient.CoreV1().Nodes().Delete(ctx, nodeName, metav1.DeleteOptions{})
@@ -1148,7 +1149,7 @@ func TestReconciler_CancelledAndUnQuarantinedClearAllState(t *testing.T) {
11481149
ctx, cancel := context.WithTimeout(testContext, 30*time.Second)
11491150
defer cancel()
11501151

1151-
nodeName := "test-cancelled-unquarantine-" + "test-node-123"
1152+
nodeName := testutils.GenerateTestNodeName("test-cancelled-unquarantine")
11521153
createTestNode(ctx, nodeName, nil, map[string]string{"test": "label"})
11531154
defer func() {
11541155
_ = testClient.CoreV1().Nodes().Delete(ctx, nodeName, metav1.DeleteOptions{})

node-drainer/pkg/reconciler/reconciler_integration_test.go

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -31,6 +31,7 @@ import (
3131
"github.com/nvidia/nvsentinel/node-drainer/pkg/reconciler"
3232
sdkclient "github.com/nvidia/nvsentinel/store-client/pkg/client"
3333
sdkconfig "github.com/nvidia/nvsentinel/store-client/pkg/config"
34+
"github.com/nvidia/nvsentinel/store-client/pkg/testutils"
3435

3536
"github.com/prometheus/client_golang/prometheus"
3637
dto "github.com/prometheus/client_model/go"
@@ -1043,7 +1044,7 @@ func TestReconciler_CancelledEventWithOngoingDrain(t *testing.T) {
10431044
{Name: "timeout-*", Mode: config.ModeDeleteAfterTimeout},
10441045
})
10451046

1046-
nodeName := "cancel-during-drain-node"
1047+
nodeName := testutils.GenerateTestNodeName("cancel-during-drain-node")
10471048
createNode(setup.ctx, t, setup.client, nodeName)
10481049

10491050
createNamespace(setup.ctx, t, setup.client, "timeout-test")
@@ -1091,7 +1092,7 @@ func TestReconciler_UnQuarantinedEventCancelsOngoingDrain(t *testing.T) {
10911092
{Name: "timeout-*", Mode: config.ModeDeleteAfterTimeout},
10921093
})
10931094

1094-
nodeName := "unquarantine-cancel-node"
1095+
nodeName := testutils.GenerateTestNodeName("unquarantine-cancel-node")
10951096
createNode(setup.ctx, t, setup.client, nodeName)
10961097

10971098
createNamespace(setup.ctx, t, setup.client, "timeout-test")
@@ -1140,7 +1141,7 @@ func TestReconciler_MultipleEventsOnNodeCancelledByUnQuarantine(t *testing.T) {
11401141
{Name: "timeout-*", Mode: config.ModeDeleteAfterTimeout},
11411142
})
11421143

1143-
nodeName := "multi-event-cancel-node"
1144+
nodeName := testutils.GenerateTestNodeName("multi-event-cancel-node")
11441145
createNode(setup.ctx, t, setup.client, nodeName)
11451146

11461147
createNamespace(setup.ctx, t, setup.client, "timeout-test")

0 commit comments

Comments
 (0)