(PE-40386) assign 'A' to avail_group_letter if this is a migration

Author: davidmalloncares

plans/add_database.pp

@@ -61,12 +61,17 @@
   out::message("Operating in ${operating_mode} mode")
 
   if $operating_mode == 'init' {
-    # If no other PSQL node then match primary group letter
-    $avail_group_letter = peadm::flatten_compact($roles['server'].map |$k,$v| {
-        if $v == $primary_host {
-          $k
-        }
-    })[0]
+    # if we are doing a migration then we need to assign the new node to group A
+    if $is_migration {
+      $avail_group_letter = 'A'
+    } else {
+      # If no other PSQL node then match primary group letter
+      $avail_group_letter = peadm::flatten_compact($roles['server'].map |$k,$v| {
+          if $v == $primary_host {
+            $k
+          }
+      })[0]
+    }
     # Assume PuppetDB backend hosted on Primary if in init mode
     $source_db_host = $primary_host
   } else {