Merge pull request #9278 from Barakmor1/backport_fix_flacks

[release-0.58]Should remove and restore labeller only from workers

tests/hyperv_test.go

@@ -1,7 +1,6 @@
 package tests_test
 
 import (
-	"context"
 	"fmt"
 	"strings"
 	"time"
@@ -125,21 +124,15 @@ var _ = Describe("[Serial][sig-compute] Hyper-V enlightenments", func() {
 
 			BeforeEach(func() {
 				if isTSCFrequencyExposed(virtClient) {
-					nodeList, err := virtClient.CoreV1().Nodes().List(context.Background(), metav1.ListOptions{})
-					Expect(err).ToNot(HaveOccurred())
-
-					for _, node := range nodeList.Items {
+					for _, node := range libnode.GetAllSchedulableNodes(virtClient).Items {
 						stopNodeLabeller(node.Name, virtClient)
 						removeTSCFrequencyFromNode(node)
 					}
 				}
 			})
 
 			AfterEach(func() {
-				nodeList, err := virtClient.CoreV1().Nodes().List(context.Background(), metav1.ListOptions{})
-				Expect(err).ToNot(HaveOccurred())
-
-				for _, node := range nodeList.Items {
+				for _, node := range libnode.GetAllSchedulableNodes(virtClient).Items {
 					_ = resumeNodeLabeller(node.Name, virtClient)
 				}
 			})