fix(test): affinity phase fail (#1615)

Description
Change type of checks for VM migration. Don't try to wait Migrating phase, just wait migration status. Thats approach more more stable, than trying to catch phase.

---------

Signed-off-by: Daniil Antoshin <[email protected]>

Author: danilrwx

tests/e2e/affinity_toleration_test.go

@@ -169,9 +169,15 @@ var _ = Describe("VirtualMachineAffinityAndToleration", framework.CommonE2ETestD
 						if err != nil {
 							return err
 						}
-						if updatedVMObjC.Status.Phase != v1alpha2.MachineMigrating {
-							return fmt.Errorf("the `VirtualMachine` should be %s", v1alpha2.MachineMigrating)
+
+						if updatedVMObjC.Status.MigrationState == nil {
+							return errors.New("the `VirtualMachine` should be migrated")
+						}
+
+						if updatedVMObjC.Status.MigrationState.Result != v1alpha2.MigrationResultSucceeded {
+							return fmt.Errorf("the `VirtualMachine` should migrate to %q", vmObjA.Status.Node)
 						}
+
 						return nil
 					}).WithTimeout(LongWaitDuration).WithPolling(migratingStatusPollingInterval).Should(Succeed())
 				}()
@@ -334,9 +340,15 @@ var _ = Describe("VirtualMachineAffinityAndToleration", framework.CommonE2ETestD
 						if err != nil {
 							return err
 						}
-						if updatedVMObj.Status.Phase != v1alpha2.MachineMigrating {
-							return fmt.Errorf("the `VirtualMachine` should be %s", v1alpha2.MachineMigrating)
+
+						if updatedVMObj.Status.MigrationState == nil {
+							return errors.New("the `VirtualMachine` should be migrated")
+						}
+
+						if updatedVMObj.Status.MigrationState.Result != v1alpha2.MigrationResultSucceeded {
+							return fmt.Errorf("the `VirtualMachine` should migrate to %q", targetNode)
 						}
+
 						return nil
 					}).WithTimeout(Timeout).WithPolling(migratingStatusPollingInterval).Should(Succeed())
 				}()
@@ -429,9 +441,15 @@ var _ = Describe("VirtualMachineAffinityAndToleration", framework.CommonE2ETestD
 						if err != nil {
 							return err
 						}
-						if updatedVMObj.Status.Phase != v1alpha2.MachineMigrating {
-							return fmt.Errorf("the `VirtualMachine` should be %s", v1alpha2.MachineMigrating)
+
+						if updatedVMObj.Status.MigrationState == nil {
+							return errors.New("the `VirtualMachine` should be migrated")
+						}
+
+						if updatedVMObj.Status.MigrationState.Result != v1alpha2.MigrationResultSucceeded {
+							return fmt.Errorf("the `VirtualMachine` should migrate to %q", targetNode)
 						}
+
 						return nil
 					}).WithTimeout(Timeout).WithPolling(migratingStatusPollingInterval).Should(Succeed())
 				}()