update

Author: abhijeet-dhumal

support/machine.go

@@ -31,10 +31,10 @@ func MachineSetId(machineSet machinev1beta1.MachineSet) string {
 	return machineSet.Name
 }
 
-func MachineSet(test Test, namespace string, machineSetName string) func(g gomega.Gomega) *machinev1beta1.MachineSet {
+func MachineSet(t 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())
+		machineset, err := t.Client().Machine().MachineV1beta1().MachineSets(namespace).Get(t.Ctx(), machineSetName, metav1.GetOptions{})
+		g.Expect(err).NotTo(gomega.HaveOccurred())
 		return machineset
 	}
 }

support/machine_test.go

@@ -28,6 +28,20 @@ func TestGetMachineSets(t *testing.T) {
 	test.Expect(machines[0].Name).To(gomega.Equal("test-machineset-1"))
 	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"))