Add common functions for SDK test

Author: sutaakar

support/core.go

@@ -164,3 +164,23 @@ func CreatePersistentVolumeClaim(t Test, namespace string, storageSize string, a
 
 	return pvc
 }
+
+func GetNodes(t Test) []corev1.Node {
+	t.T().Helper()
+	nodes, err := t.Client().Core().CoreV1().Nodes().List(t.Ctx(), metav1.ListOptions{})
+	t.Expect(err).NotTo(gomega.HaveOccurred())
+	return nodes.Items
+}
+
+func GetNodeInternalIP(t Test, node corev1.Node) (IP string) {
+	t.T().Helper()
+
+	for _, address := range node.Status.Addresses {
+		if address.Type == "InternalIP" {
+			IP = address.Address
+		}
+	}
+	t.Expect(IP).Should(gomega.Not(gomega.BeEmpty()), "Node internal IP address not found")
+
+	return
+}

support/defaults.go

@@ -5,7 +5,7 @@ package support
 // ***********************
 
 const (
-	CodeFlareSDKVersion = "0.9.0"
+	CodeFlareSDKVersion = "0.12.0"
 	RayVersion          = "2.5.0"
 	RayImage            = "rayproject/ray:2.5.0"
 )

support/environment.go

@@ -41,6 +41,9 @@ const (
 
 	// Type of cluster test is run on
 	ClusterTypeEnvVar = "CLUSTER_TYPE"
+
+	// Hostname of the Kubernetes cluster
+	ClusterHostname = "CLUSTER_HOSTNAME"
 )
 
 type ClusterType string
@@ -49,6 +52,7 @@ const (
 	OsdCluster        ClusterType = "OSD"
 	OcpCluster        ClusterType = "OCP"
 	HypershiftCluster ClusterType = "HYPERSHIFT"
+	KindCluster       ClusterType = "KIND"
 	UndefinedCluster  ClusterType = "UNDEFINED"
 )
 
@@ -90,12 +94,22 @@ func GetClusterType(t Test) ClusterType {
 		return OcpCluster
 	case "HYPERSHIFT":
 		return HypershiftCluster
+	case "KIND":
+		return KindCluster
 	default:
 		t.T().Logf("Expected environment variable %s contains unexpected value: '%s'", ClusterTypeEnvVar, clusterType)
 		return UndefinedCluster
 	}
 }
 
+func GetClusterHostname(t Test) string {
+	hostname, ok := os.LookupEnv(ClusterHostname)
+	if !ok {
+		t.T().Fatalf("Expected environment variable %s not found, please define cluster hostname.", ClusterHostname)
+	}
+	return hostname
+}
+
 func lookupEnvOrDefault(key, value string) string {
 	if v, ok := os.LookupEnv(key); ok {
 		return v