@@ -21,11 +21,11 @@ var _ = Describe("[sig-kubevirt] migration", func() {
2121 mgmtFramework := e2e .NewDefaultFramework ("mgmt-framework" )
2222 mgmtFramework .SkipNamespaceCreation = true
2323
24- f1 := e2e .NewDefaultFramework ("server -framework" )
25- f1 .NamespacePodSecurityEnforceLevel = admissionapi .LevelPrivileged
24+ hostedFramework := e2e .NewDefaultFramework ("hosted -framework" )
25+ hostedFramework .NamespacePodSecurityEnforceLevel = admissionapi .LevelPrivileged
2626 var (
2727 numberOfReadyNodes = func () (int , error ) {
28- nodeList , err := f1 .ClientSet .CoreV1 ().Nodes ().List (context .Background (), metav1.ListOptions {})
28+ nodeList , err := hostedFramework .ClientSet .CoreV1 ().Nodes ().List (context .Background (), metav1.ListOptions {})
2929 if err != nil {
3030 return 0 , err
3131 }
@@ -40,12 +40,24 @@ var _ = Describe("[sig-kubevirt] migration", func() {
4040 return numberOfReadyNodes , nil
4141 }
4242 )
43- AfterLiveMigrateWorkersContext (mgmtFramework , func () {
44- It ("should maintain node readiness" , func () {
45- nodeList , err := f1 .ClientSet .CoreV1 ().Nodes ().List (context .Background (), metav1.ListOptions {})
43+ Context ("and live migrate hosted control plane workers [Early]" , func () {
44+ var (
45+ numberOfNodes = 0
46+ )
47+ BeforeEach (func () {
48+ nodeList , err := hostedFramework .ClientSet .CoreV1 ().Nodes ().List (context .Background (), metav1.ListOptions {})
4649 Expect (err ).ToNot (HaveOccurred ())
47- numberOfNodes : = len (nodeList .Items )
50+ numberOfNodes = len (nodeList .Items )
4851
52+ Eventually (numberOfReadyNodes ).
53+ WithTimeout (2 * time .Minute ).
54+ WithPolling (5 * time .Second ).
55+ Should (Equal (numberOfNodes ), "nodes should have ready state before migration" )
56+
57+ setMgmtFramework (mgmtFramework )
58+ expectNoError (migrateWorkers (mgmtFramework ))
59+ })
60+ It ("should maintain node readiness" , func () {
4961 By ("Check node readiness is as expected" )
5062 isAWS , err := mgmtClusterIsAWS (mgmtFramework )
5163 Expect (err ).ToNot (HaveOccurred ())
0 commit comments