fixing e2e tests

Signed-off-by: alice-px <[email protected]>
libopenstorage · Jan 19, 2024 · 9f145b6 · 9f145b6
1 parent 86f2842
commit 9f145b6
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 4 deletions.
diff --git a/api/server/sdk/node_test.go b/api/server/sdk/node_test.go
@@ -460,7 +460,7 @@ func TestSdkVolumeUsageByNode(t *testing.T) {
 
 	s.MockCluster().EXPECT().Enumerate().Return(cluster, nil).Times(1)
 	s.MockCluster().EXPECT().Inspect(nodeid).Return(node, nil).Times(2)
-	s.MockDriver().EXPECT().VolumeUsageByNode(context.TODO(), nodeid).Return(&volumeUsageInfo, nil).Times(1)
+	s.MockDriver().EXPECT().VolumeUsageByNode(context.Background(), nodeid).Return(&volumeUsageInfo, nil).Times(1)
 
 	// Setup client
 	c := api.NewOpenStorageNodeClient(s.Conn())

diff --git a/pkg/loadbalancer/roundrobin.go b/pkg/loadbalancer/roundrobin.go
@@ -149,9 +149,6 @@ func (rr *roundRobin) getTargetAndIncrement(filteredNodes []*api.Node, selfNodeI
 		return "", false
 	}
 	targetNodeNumber = rr.nextCreateNodeNumber % len(filteredNodes)
-	if rr.nextCreateNodeNumber != 0 {
-		targetNodeNumber = rr.nextCreateNodeNumber
-	}
 	targetNode := filteredNodes[targetNodeNumber]
 	if targetNode.Id != selfNodeID {
 		// NodeID set on the cluster object is this node's ID.