add XL with and without DR to test migration and migrate plans

Author: B00885424

.github/workflows/test-migration.yaml

@@ -40,9 +40,9 @@ jobs:
           - standard
           - standard-with-dr
           - large
-            # - extra-large
+          - extra-large
           - large-with-dr
-            # - extra-large-with-dr
+          - extra-large-with-dr
         version: [2021.7.9, 2023.8.2, 2025.2.0]
         image: [almalinux-cloud/almalinux-8]
         include:
@@ -54,6 +54,10 @@ jobs:
             version: 2023.8.0
             image: almalinux-cloud/almalinux-8
             new_pe_version: 2025.2.0
+          - architecture: extra-large
+            version: 2023.8.0
+            image: almalinux-cloud/almalinux-8
+            new_pe_version: 2025.2.0
     steps:
       - name: Checkout Source
         uses: actions/checkout@v4

plans/migrate.pp

@@ -8,12 +8,19 @@
 #   The new server that will become the PE primary server
 # @param upgrade_version
 #   Optional version to upgrade to after migration is complete
-#
+# @param replica_host
+#   Optional new replica server to be added to the cluster
+# @param primary_postgresql_host
+#   Optional new primary PostgreSQL server to be added to the cluster
+# @param replica_postgresql_host
+#   Optional new replica PostgreSQL server to be added to the cluster
 plan peadm::migrate (
   Peadm::SingleTargetSpec $old_primary_host,
   Peadm::SingleTargetSpec $new_primary_host,
   Optional[String] $upgrade_version = undef,
   Optional[Peadm::SingleTargetSpec] $replica_host = undef,
+  Optional[Peadm::SingleTargetSpec] $primary_postgresql_host = undef,
+  Optional[Peadm::SingleTargetSpec] $replica_postgresql_host = undef,
 ) {
   # pre-migration checks
   out::message('This plan is a work in progress and it is not recommended to be used until it is fully implemented and supported')
@@ -25,7 +32,9 @@
 
   $new_hosts = peadm::flatten_compact([
       $new_primary_host,
-      $replica_host ? { undef => [], default => [$replica_host] }
+      $replica_host ? { undef => [], default => [$replica_host] },
+      $primary_postgresql_host ? { undef => [], default => [$primary_postgresql_host] },
+      $replica_postgresql_host ? { undef => [], default => [$replica_postgresql_host] },
   ].flatten)
   $all_hosts = peadm::flatten_compact([
       $old_primary_host,
@@ -118,6 +127,17 @@
     out::message('No nodes to purge from old configuration')
   }
 
+  if $primary_postgresql_host {
+    run_plan('peadm::add_database', targets => $primary_postgresql_host,
+      primary_host => $new_primary_host,
+    )
+    if $replica_postgresql_host {
+      run_plan('peadm::add_database', targets => $replica_postgresql_host,
+        primary_host => $new_primary_host,
+      )
+    }
+  }
+
   if $replica_host {
     run_plan('peadm::add_replica', {
         primary_host => $new_primary_host,

spec/acceptance/peadm_spec/plans/test_migration.pp

@@ -30,6 +30,8 @@
     new_primary_host => $new_primary_target,
     upgrade_version => $upgrade_version,
     replica_host => $new_replica_target,
+    primary_postgresql_host => $new_primary_postgresql_target,
+    replica_postgresql_host => $new_replica_postgresql_target,
   )
 
   # run infra status on the new primary
@@ -66,9 +68,9 @@
   # if new_replica_postgresql_target is supplied then check that is in the expected place in the config
   if $new_replica_postgresql_target {
     if $peadm_config['params']['replica_postgresql_host'] == $new_replica_postgresql_target.peadm::certname() {
-      out::message("New primary postgres host ${new_replica_postgresql_target.peadm::certname()} set up correctly")
+      out::message("New replica postgres host ${new_replica_postgresql_target.peadm::certname()} set up correctly")
     } else {
-      fail_plan("New primary postgres host ${new_replica_postgresql_target.peadm::certname()} was not set up correctly")
+      fail_plan("New replica postgres host ${new_replica_postgresql_target.peadm::certname()} was not set up correctly")
     }
   }