.

Author: NihaNallappagari

cns/restserver/internalapi_test.go

@@ -289,7 +289,7 @@ func TestSyncHostNCVersion(t *testing.T) {
                 t.Errorf("Unexpected imdsContainerStatus.HostVersion %s, expected host version should be 1 in string", imdsContainerStatus.HostVersion)
             }
             if imdsContainerStatus.CreateNetworkContainerRequest.Version != "1" {
-                t.Errorf("Unexpected imdsContainerStatus.CreateNetworkContainerRequest.Version %s, expected VM version should remain 1 in string", imdsContainerStatus.CreateNetworkContainerRequest.Version)
+                t.Errorf("Unexpected version %s, expected VM version should remain 1 in string", imdsContainerStatus.CreateNetworkContainerRequest.Version)
 			}
         })
     }
@@ -1780,7 +1780,6 @@ func TestMustEnsureNoStaleNCs_PanicsWhenIPsFromStaleNCAreAssigned(t *testing.T)
 	})
 }
 
-// mockIMDSAdapter adapts the anonymous struct to implement the imdsClient interface
 type mockIMDSAdapter struct {
 	mock *struct {
 		networkInterfaces func(_ context.Context) ([]imds.NetworkInterface, error)
@@ -1806,7 +1805,6 @@ func (m *mockIMDSAdapter) GetIMDSVersions(ctx context.Context) (*imds.APIVersion
     }, nil
 }
 
-// Add this helper function before TestSyncHostNCVersion
 func setupIMDSMockAPIsWithCustomIDs(svc *HTTPRestService, interfaceIDs []string) func() {
     mockIMDS := &struct {
         networkInterfaces func(_ context.Context) ([]imds.NetworkInterface, error)