Fetch nodes with nic count.

Author: sivakami

test/integration/swiftv2/longRunningCluster/datapath.go

@@ -4,6 +4,7 @@ import (
 	"bytes"
 	"fmt"
 	"os/exec"
+	"strings"
 	"text/template"
 
 	"github.com/Azure/azure-container-networking/test/integration/swiftv2/helpers"
@@ -130,29 +131,46 @@ type NodePoolInfo struct {
 	HighNicNodes []string
 }
 
-// GetNodesByNicCount categorizes nodes by NIC count (1 NIC vs 7+ NICs)
+// GetNodesByNicCount categorizes nodes by NIC count based on node pool labels
 func GetNodesByNicCount(kubeconfig string) (NodePoolInfo, error) {
-	nodes, err := helpers.GetClusterNodes(kubeconfig)
-	if err != nil {
-		return NodePoolInfo{}, err
-	}
-
 	nodeInfo := NodePoolInfo{
 		LowNicNodes:  []string{},
 		HighNicNodes: []string{},
 	}
 
-	// In a real implementation, you'd query node labels or annotations
-	// For now, we'll assume first 2 nodes are low-NIC, last 2 are high-NIC
-	// You can enhance this by checking actual node labels like "agentpool"
-	if len(nodes) >= 4 {
-		nodeInfo.LowNicNodes = nodes[:2]
-		nodeInfo.HighNicNodes = nodes[2:]
-	} else if len(nodes) >= 2 {
-		nodeInfo.LowNicNodes = nodes[:1]
-		nodeInfo.HighNicNodes = nodes[1:]
+	// Get nodes from default node pool (low-NIC nodes)
+	cmd := exec.Command("kubectl", "--kubeconfig", kubeconfig, "get", "nodes", 
+		"-l", "agentpool!=nplinux", "-o", "name")
+	out, err := cmd.CombinedOutput()
+	if err != nil {
+		return NodePoolInfo{}, fmt.Errorf("failed to get default pool nodes: %w\nOutput: %s", err, string(out))
+	}
+
+	lines := strings.Split(strings.TrimSpace(string(out)), "\n")
+	for _, line := range lines {
+		if strings.HasPrefix(line, "node/") {
+			nodeInfo.LowNicNodes = append(nodeInfo.LowNicNodes, strings.TrimPrefix(line, "node/"))
+		}
 	}
 
+	// Get nodes from nplinux node pool (high-NIC nodes)
+	cmd = exec.Command("kubectl", "--kubeconfig", kubeconfig, "get", "nodes", 
+		"-l", "agentpool=nplinux", "-o", "name")
+	out, err = cmd.CombinedOutput()
+	if err != nil {
+		return NodePoolInfo{}, fmt.Errorf("failed to get nplinux pool nodes: %w\nOutput: %s", err, string(out))
+	}
+
+	lines = strings.Split(strings.TrimSpace(string(out)), "\n")
+	for _, line := range lines {
+		if line != "" && strings.HasPrefix(line, "node/") {
+			nodeInfo.HighNicNodes = append(nodeInfo.HighNicNodes, strings.TrimPrefix(line, "node/"))
+		}
+	}
+
+	fmt.Printf("Found %d low-NIC nodes (default pool) and %d high-NIC nodes (nplinux pool)\n", 
+		len(nodeInfo.LowNicNodes), len(nodeInfo.HighNicNodes))
+
 	return nodeInfo, nil
 }