@@ -30,17 +30,14 @@ import (
30
30
31
31
csi "github.com/container-storage-interface/spec/lib/go/csi"
32
32
"github.com/google/go-cmp/cmp"
33
- "github.com/google/go-cmp/cmp/cmpopts"
34
33
"google.golang.org/grpc/codes"
35
34
"google.golang.org/grpc/status"
36
35
"k8s.io/mount-utils"
37
- "sigs.k8s.io/gcp-compute-persistent-disk-csi-driver/pkg/common"
38
36
"sigs.k8s.io/gcp-compute-persistent-disk-csi-driver/pkg/deviceutils"
39
37
metadataservice "sigs.k8s.io/gcp-compute-persistent-disk-csi-driver/pkg/gce-cloud-provider/metadata"
40
38
mountmanager "sigs.k8s.io/gcp-compute-persistent-disk-csi-driver/pkg/mount-manager"
41
39
42
40
corev1 "k8s.io/api/core/v1"
43
- metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
44
41
"k8s.io/apimachinery/pkg/runtime"
45
42
"k8s.io/client-go/kubernetes/fake"
46
43
)
@@ -358,93 +355,6 @@ func NewFakeKubeClient(nodes []*corev1.Node) kubernetes.Interface {
358
355
return clientset
359
356
}
360
357
361
- func TestNodeGetInfo_Topologies (t * testing.T ) {
362
- nodeA := & corev1.Node {
363
- ObjectMeta : metav1.ObjectMeta {
364
- Name : testNodeA ,
365
- Labels : map [string ]string {
366
- common .TopologyKeyZone : testZoneA ,
367
- common .TopologyLabelKey (testDiskA ): "true" ,
368
- },
369
- },
370
- }
371
- nodeB := & corev1.Node {
372
- ObjectMeta : metav1.ObjectMeta {
373
- Name : testNodeB ,
374
- Labels : map [string ]string {
375
- common .TopologyKeyZone : testZoneB ,
376
- common .TopologyLabelKey (testDiskB ): "true" ,
377
- },
378
- },
379
- }
380
- gceDriver := getTestGCEDriver (t )
381
- ns := gceDriver .ns
382
-
383
- volumeLimit , err := ns .GetVolumeLimits ()
384
- if err != nil {
385
- t .Fatalf ("Failed to get volume limits: %v" , err )
386
- }
387
-
388
- testCases := []struct {
389
- name string
390
- node * corev1.Node
391
- enableDiskTopology bool
392
- want * csi.NodeGetInfoResponse
393
- }{
394
- {
395
- name : "success default: zone only" ,
396
- node : nodeB ,
397
- want : & csi.NodeGetInfoResponse {
398
- NodeId : common .CreateNodeID (ns .MetadataService .GetProject (), testZoneB , testNodeB ),
399
- MaxVolumesPerNode : volumeLimit ,
400
- AccessibleTopology : & csi.Topology {
401
- Segments : map [string ]string {
402
- common .TopologyKeyZone : testZoneB ,
403
- // Note the absence of the Disk Support Label
404
- },
405
- },
406
- },
407
- },
408
- {
409
- name : "success: disk topology enabled" ,
410
- node : nodeA ,
411
- enableDiskTopology : true ,
412
- want : & csi.NodeGetInfoResponse {
413
- NodeId : common .CreateNodeID (ns .MetadataService .GetProject (), testZoneA , testNodeA ),
414
- MaxVolumesPerNode : volumeLimit ,
415
- AccessibleTopology : & csi.Topology {
416
- Segments : map [string ]string {
417
- common .TopologyKeyZone : testZoneA ,
418
- common .TopologyLabelKey (testDiskA ): "true" ,
419
- },
420
- },
421
- },
422
- },
423
- }
424
-
425
- for _ , tc := range testCases {
426
- t .Logf ("Test case: %s" , tc .name )
427
-
428
- ns .EnableDiskTopology = tc .enableDiskTopology
429
- metadataservice .SetZone (tc .node .Labels [common .TopologyKeyZone ])
430
- metadataservice .SetName (tc .node .Name )
431
-
432
- res , err := ns .NodeGetInfo (context .Background (), & csi.NodeGetInfoRequest {})
433
- if err != nil {
434
- t .Fatalf ("Unexpected error: %v" , err )
435
- }
436
- if res == nil {
437
- t .Fatalf ("Expected non-nil response, got nil" )
438
- }
439
-
440
- if diff := cmp .Diff (tc .want , res , cmpopts .IgnoreUnexported (csi.NodeGetInfoResponse {}, csi.Topology {})); diff != "" {
441
- t .Errorf ("Unexpected NodeGetInfoResponse (-want +got):\n %s" , diff )
442
- }
443
-
444
- t .Logf ("Get node info: %v" , res )
445
- }
446
- }
447
-
448
358
func TestNodePublishVolume (t * testing.T ) {
449
359
gceDriver := getTestGCEDriver (t )
450
360
ns := gceDriver .ns
0 commit comments