test(migration): cover shared migration store

Signed-off-by: Daniil Antoshin <daniil.antoshin@flant.com>

Author: danilrwx

images/virtualization-artifact/pkg/migration/store/store_test.go

@@ -75,4 +75,51 @@ var _ = Describe("Migration store", func() {
 		Expect(fakeClient.Get(context.Background(), types.NamespacedName{Name: ConfigMapName, Namespace: Namespace}, updatedCM)).To(Succeed())
 		Expect(updatedCM.Data).To(HaveKeyWithValue("test-migration", completedAt))
 	})
+
+	It("should return false when shared ConfigMap exists without migration key", func() {
+		cm := &corev1.ConfigMap{
+			ObjectMeta: metav1.ObjectMeta{Name: ConfigMapName, Namespace: Namespace},
+			Data:       map[string]string{"other-migration": "2026-04-29T10:25:06Z"},
+		}
+		fakeClient, err := testutil.NewFakeClientWithObjects(cm)
+		Expect(err).NotTo(HaveOccurred())
+
+		store := NewConfigMapStore(fakeClient)
+		completed, err := store.IsCompleted(context.Background(), "test-migration")
+		Expect(err).NotTo(HaveOccurred())
+		Expect(completed).To(BeFalse())
+	})
+
+	It("should preserve existing migration records", func() {
+		otherCompletedAt := "2026-04-29T10:25:06Z"
+		cm := &corev1.ConfigMap{
+			ObjectMeta: metav1.ObjectMeta{Name: ConfigMapName, Namespace: Namespace},
+			Data:       map[string]string{"other-migration": otherCompletedAt},
+		}
+		fakeClient, err := testutil.NewFakeClientWithObjects(cm)
+		Expect(err).NotTo(HaveOccurred())
+
+		store := NewConfigMapStore(fakeClient)
+		Expect(store.MarkCompleted(context.Background(), "test-migration")).To(Succeed())
+
+		updatedCM := &corev1.ConfigMap{}
+		Expect(fakeClient.Get(context.Background(), types.NamespacedName{Name: ConfigMapName, Namespace: Namespace}, updatedCM)).To(Succeed())
+		Expect(updatedCM.Data).To(HaveKeyWithValue("other-migration", otherCompletedAt))
+		Expect(updatedCM.Data).To(HaveKey("test-migration"))
+	})
+
+	It("should initialize empty data in existing ConfigMap", func() {
+		cm := &corev1.ConfigMap{
+			ObjectMeta: metav1.ObjectMeta{Name: ConfigMapName, Namespace: Namespace},
+		}
+		fakeClient, err := testutil.NewFakeClientWithObjects(cm)
+		Expect(err).NotTo(HaveOccurred())
+
+		store := NewConfigMapStore(fakeClient)
+		Expect(store.MarkCompleted(context.Background(), "test-migration")).To(Succeed())
+
+		updatedCM := &corev1.ConfigMap{}
+		Expect(fakeClient.Get(context.Background(), types.NamespacedName{Name: ConfigMapName, Namespace: Namespace}, updatedCM)).To(Succeed())
+		Expect(updatedCM.Data).To(HaveKey("test-migration"))
+	})
 })

images/virtualization-artifact/pkg/migration/vm_running_transition_time_test.go

@@ -85,6 +85,30 @@ var _ = Describe("Migration VM Running Transition Time", func() {
 		Expect(cond).NotTo(BeNil())
 		Expect(cond.LastTransitionTime.Equal(&wrongTime)).To(BeTrue())
 	})
+
+	It("should preserve other migration records in shared ConfigMap", func() {
+		creationTime := metav1.NewTime(time.Date(2026, 4, 29, 9, 0, 0, 0, time.UTC))
+		wrongTime := metav1.NewTime(time.Date(2026, 4, 29, 10, 25, 6, 0, time.UTC))
+		marker := &corev1.ConfigMap{
+			ObjectMeta: metav1.ObjectMeta{Name: migrationstore.ConfigMapName, Namespace: migrationstore.Namespace},
+			Data:       map[string]string{"other-migration": "2026-04-29T10:30:00Z"},
+		}
+
+		vm := newRunningVM(namespace, "vm", wrongTime)
+		kvvmi := newKVVMI(namespace, "vm", creationTime)
+
+		fakeClient, err := testutil.NewFakeClientWithObjects(vm, kvvmi, marker)
+		Expect(err).NotTo(HaveOccurred())
+
+		m, err := newVMRunningTransitionTime(fakeClient, testutil.NewNoOpLogger())
+		Expect(err).NotTo(HaveOccurred())
+		Expect(m.Migrate(context.Background())).To(Succeed())
+
+		updatedMarker := &corev1.ConfigMap{}
+		Expect(fakeClient.Get(context.Background(), types.NamespacedName{Name: migrationstore.ConfigMapName, Namespace: migrationstore.Namespace}, updatedMarker)).To(Succeed())
+		Expect(updatedMarker.Data).To(HaveKeyWithValue("other-migration", "2026-04-29T10:30:00Z"))
+		Expect(updatedMarker.Data).To(HaveKey(vmRunningTransitionTimeMigrationName))
+	})
 })
 
 func newRunningVM(namespace, name string, transitionTime metav1.Time) *v1alpha2.VirtualMachine {