From 7f16a6b7f015c7b1a8d8ff2a2a04d3ae69f7faeb Mon Sep 17 00:00:00 2001
From: abdhumal <abdhumal@redhat.com>
Date: Thu, 25 Jan 2024 15:25:23 +0530
Subject: [PATCH] add getMachineSet function and machineSet test in accordance
 with instascale issue #193_Add_functional_tests

---
 support/machine.go      | 12 ++++++++++++
 support/machine_test.go | 17 +++++++++++++++++
 2 files changed, 29 insertions(+)

diff --git a/support/machine.go b/support/machine.go
index 26acb4b..db0af88 100644
--- a/support/machine.go
+++ b/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(t Test, namespace string, machineSetName string) func(g gomega.Gomega) *machinev1beta1.MachineSet {
+	return func(g gomega.Gomega) *machinev1beta1.MachineSet {
+		machineset, err := t.Client().Machine().MachineV1beta1().MachineSets(namespace).Get(t.Ctx(), machineSetName, metav1.GetOptions{})
+		g.Expect(err).NotTo(gomega.HaveOccurred())
+		return machineset
+	}
+}
+
+func MachineSetReplicas(machineSet *machinev1beta1.MachineSet) *int32 {
+	return machineSet.Spec.Replicas
+}
diff --git a/support/machine_test.go b/support/machine_test.go
index f043020..82ca38f 100644
--- a/support/machine_test.go
+++ b/support/machine_test.go
@@ -29,3 +29,20 @@ func TestGetMachineSets(t *testing.T) {
 	test.Expect(machines[0].Namespace).To(gomega.Equal("openshift-machine-api"))
 
 }
+
+func TestMachineSet(t *testing.T) {
+	test := NewTest(t)
+
+	machine := &machinev1beta1.MachineSet{
+		ObjectMeta: metav1.ObjectMeta{
+			Name:      "test-machineset-1",
+			Namespace: "openshift-machine-api",
+		},
+	}
+
+	test.client.Machine().MachineV1beta1().MachineSets("openshift-machine-api").Create(test.ctx, machine, metav1.CreateOptions{})
+
+	machineSet := MachineSet(test, "openshift-machine-api", "test-machineset-1")
+	test.Expect(machineSet(test).Name).To(gomega.Equal("test-machineset-1"))
+
+}