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")) + +}