add getMachineSet function and machineSet test in accordance with instascale issue #193_Add_functional_tests

Author: abhijeet-dhumal

support/machine.go

@@ -30,3 +30,15 @@ func GetMachines(t Test, machineSetName string) []machinev1beta1.Machine {
 func MachineSetId(machineSet machinev1beta1.MachineSet) string {
 	return machineSet.Name
 }
+
+func MachineSet(test Test, namespace string, machineSetName string) func(g gomega.Gomega) *machinev1beta1.MachineSet {
+	return func(g gomega.Gomega) *machinev1beta1.MachineSet {
+		machineset, err := test.Client().Machine().MachineV1beta1().MachineSets(namespace).Get(test.Ctx(), machineSetName, metav1.GetOptions{})
+		test.Expect(err).ToNot(gomega.HaveOccurred())
+		return machineset
+	}
+}
+
+func MachineSetReplicas(machineSet *machinev1beta1.MachineSet) *int32 {
+	return machineSet.Spec.Replicas
+}

support/machineset_test.go

@@ -0,0 +1,30 @@
+package support
+
+import (
+	"testing"
+	"time"
+
+	"github.com/onsi/gomega"
+
+	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
+
+	machinev1beta1 "github.com/openshift/api/machine/v1beta1"
+)
+
+func TestMachineSet(t *testing.T) {
+	test := NewTest(t)
+
+	machineSetData := &machinev1beta1.MachineSet{
+		ObjectMeta: metav1.ObjectMeta{
+			Name:      "test-machineset-1",
+			Namespace: "openshift-machine-api",
+		},
+	}
+
+	test.client.Machine().MachineV1beta1().MachineSets("openshift-machine-api").Create(test.ctx, machineSetData, metav1.CreateOptions{})
+
+	machineSet := MachineSet(test, "openshift-machine-api", "test-machineset-1")
+
+	test.Eventually(machineSet, 10*time.Second).Should(gomega.WithTransform(MachineSetReplicas, gomega.BeNil()))
+
+}