mock test

Signed-off-by: Rohit-PX <rkulkarni@purestorage.com>

Author: Rohit-PX

test/integration_test/migration_test.go

@@ -55,6 +55,7 @@ func testMigration(t *testing.T) {
 	err = setSourceKubeConfig()
 	require.NoError(t, err, "failed to set kubeconfig to source cluster: %v", err)
 
+	t.Run("migrationStatsExportTest", migrationStatsExportTest)
 	t.Run("deploymentTest", deploymentMigrationTest)
 	t.Run("deploymentMigrationReverseTest", deploymentMigrationReverseTest)
 	t.Run("statefulsetTest", statefulsetMigrationTest)
@@ -1549,3 +1550,36 @@ func transformResourceTest(t *testing.T) {
 	require.NoError(t, err, "failed to set kubeconfig to source cluster: %v", err)
 	validateAndDestroyMigration(t, ctxs, preMigrationCtx, true, false, true, false, true)
 }
+
+func migrationStatsExportTest(t *testing.T) {
+	var err error
+	// Reset config in case of error
+	defer func() {
+		err = setSourceKubeConfig()
+		require.NoError(t, err, "Error resetting source config")
+	}()
+
+	migrationKey := "mysql-migration"
+	migrationAppKey := "mysql-1-pvc"
+
+	currCtxs, err := schedulerDriver.Schedule(migrationKey,
+		scheduler.ScheduleOptions{AppKeys: []string{migrationAppKey}})
+	require.NoError(t, err, "Error scheduling task")
+	require.Equal(t, 1, len(currCtxs), "Only one task should have started")
+
+	err = schedulerDriver.WaitForRunning(currCtxs[0], defaultWaitTimeout, defaultWaitInterval)
+	require.NoError(t, err, "Error waiting for app to get to running state")
+
+	// create, apply and validate cluster pair specs
+	err = scheduleClusterPair(currCtxs[0], false, true, defaultClusterPairDir, "", false)
+	require.NoError(t, err, "Error scheduling cluster pair")
+	ctxs = append(ctxs, currCtxs...)
+
+	currMigNamespace := migrationAppKey + "-" + migrationKey
+	currMig, err := createMigration(t, migrationKey, currMigNamespace, "remoteclusterpair", currMigNamespace, &includeResourcesFlag, &startApplicationsFlag)
+	require.NoError(t, err, "failed to create migration: %s in namespace %s", migrationKey, currMigNamespace)
+
+	migrationList := []*v1alpha1.Migration{icurrMig}
+	err = WaitForMigration(allMigrations)
+	require.NoError(t, err, "Error in scaled migrations")
+}