Skip to content

Commit 1edd775

Browse files
add getMachineSet function and machineSet test in accordance with instascale issue #193_Add_functional_tests
1 parent 4f0a7af commit 1edd775

File tree

2 files changed

+29
-0
lines changed

2 files changed

+29
-0
lines changed

support/machine.go

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -30,3 +30,15 @@ func GetMachines(t Test, machineSetName string) []machinev1beta1.Machine {
3030
func MachineSetId(machineSet machinev1beta1.MachineSet) string {
3131
return machineSet.Name
3232
}
33+
34+
func MachineSet(t Test, namespace string, machineSetName string) func(g gomega.Gomega) *machinev1beta1.MachineSet {
35+
return func(g gomega.Gomega) *machinev1beta1.MachineSet {
36+
machineset, err := t.Client().Machine().MachineV1beta1().MachineSets(namespace).Get(t.Ctx(), machineSetName, metav1.GetOptions{})
37+
g.Expect(err).NotTo(gomega.HaveOccurred())
38+
return machineset
39+
}
40+
}
41+
42+
func MachineSetReplicas(machineSet *machinev1beta1.MachineSet) *int32 {
43+
return machineSet.Spec.Replicas
44+
}

support/machine_test.go

Lines changed: 17 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -29,3 +29,20 @@ func TestGetMachineSets(t *testing.T) {
2929
test.Expect(machines[0].Namespace).To(gomega.Equal("openshift-machine-api"))
3030

3131
}
32+
33+
func TestMachineSet(t *testing.T) {
34+
test := NewTest(t)
35+
36+
machine := &machinev1beta1.MachineSet{
37+
ObjectMeta: metav1.ObjectMeta{
38+
Name: "test-machineset-1",
39+
Namespace: "openshift-machine-api",
40+
},
41+
}
42+
43+
test.client.Machine().MachineV1beta1().MachineSets("openshift-machine-api").Create(test.ctx, machine, metav1.CreateOptions{})
44+
45+
machineSet := MachineSet(test, "openshift-machine-api", "test-machineset-1")
46+
test.Expect(machineSet(test).Name).To(gomega.Equal("test-machineset-1"))
47+
48+
}

0 commit comments

Comments
 (0)