tests/e2e, persistentip, primaryUDN: Refresh vmi after migration/restart

In some cases after migration/restart the VMI object is not refreshed,
making it so that the object may be already irrelevant.

In order to prevent that, refresh the vmi instance as done in other
tests.

Signed-off-by: Ram Lavi <[email protected]>

test/e2e/persistentips-primary_test.go

@@ -116,6 +116,8 @@ var _ = Describe("Persistent IPs on Primary UDN interface", func() {
 					WithTimeout(5 * time.Minute).
 					Should(testenv.ContainConditionVMIReady())
 
+				Expect(testenv.Client.Get(context.Background(), client.ObjectKeyFromObject(vmi), vmi)).To(Succeed())
+
 				targetVMIDefaultNetworkStatus, err := getDefaultNetworkStatus(vmi)
 				Expect(err).ToNot(HaveOccurred())
 				Expect(targetVMIDefaultNetworkStatus.Interface).To(Equal(originalVMIDefaultNetworkStatus.Interface))
@@ -198,6 +200,8 @@ var _ = Describe("Persistent IPs on Primary UDN interface", func() {
 					WithTimeout(5 * time.Minute).
 					Should(testenv.ContainConditionVMIReady())
 
+				Expect(testenv.Client.Get(context.Background(), client.ObjectKeyFromObject(vmi), vmi)).To(Succeed())
+
 				defaultNetworkStatusAfterRestart, err := getDefaultNetworkStatus(vmi)
 				Expect(err).ToNot(HaveOccurred())
 				Expect(defaultNetworkStatusAfterRestart.Interface).To(Equal(defaultNetworkStatusBeforeRestart.Interface))
@@ -289,6 +293,8 @@ var _ = Describe("Persistent IPs on Primary UDN interface", func() {
 				testenv.LiveMigrateVirtualMachine(td.Namespace, vmi.Name)
 				testenv.CheckLiveMigrationSucceeded(td.Namespace, vmi.Name)
 
+				Expect(testenv.Client.Get(context.Background(), client.ObjectKeyFromObject(vmi), vmi)).To(Succeed())
+
 				targetVMIDefaultNetworkStatus, err := getDefaultNetworkStatus(vmi)
 				Expect(err).ToNot(HaveOccurred())
 				Expect(targetVMIDefaultNetworkStatus.Interface).To(Equal(originalVMIDefaultNetworkStatus.Interface))