test code for migration

Author: davidmalloncares

functions/migration_opts_default.pp

@@ -7,6 +7,6 @@ function peadm::migration_opts_default () {
     'config'       => false,
     'orchestrator' => true,
     'puppetdb'     => true,
-    'rbac'         => true,
+    'rbac'         => false,
   }
 }

plans/backup.pp

@@ -12,7 +12,7 @@
   Peadm::SingleTargetSpec     $targets,
 
   # backup type determines the backup options
-  Enum['recovery', 'custom']  $backup_type = 'recovery',
+  Enum['recovery', 'custom', 'migration']  $backup_type = 'recovery',
 
   # Which data to backup
   Peadm::Recovery_opts        $backup = {},

plans/check_config.pp

@@ -0,0 +1,20 @@
+plan peadm::check_config(
+  TargetSpec $targets
+) {
+  get_targets($targets).each |$target| {
+    $peadm_config = run_task('peadm::get_peadm_config', $target).first.value
+    out::message("${target} - peadm_config:${$peadm_config}.")
+    $postgresql_a_host = $peadm_config['role-letter']['postgresql']['A']
+    $postgresql_b_host = $peadm_config['role-letter']['postgresql']['B']
+    $pe_status = run_task('service', $target, 'action' => 'status', 'name' => 'puppet.service')
+    $pdb_status = run_task('service', $target, 'action' => 'status', 'name' => 'pe-puppetdb.service')
+    $agent_status = run_task('package', $target,
+      action => 'status',
+    name   => 'puppet-agent').first['status']
+    out::message("${target} - postgresql_a_host:${$postgresql_a_host}.")
+    out::message("${target} - postgresql_b_host:${$postgresql_b_host}.")
+    out::message("${target} - pe_status:${$pe_status}.")
+    out::message("${target} - pdb_status:${$pdb_status}.")
+    out::message("${target} - agent_status:${$agent_status}.")
+  }
+}

plans/restore.pp

@@ -12,7 +12,7 @@
   Peadm::SingleTargetSpec $targets,
 
   # restore type determines the restore options
-  Enum['recovery', 'recovery-db', 'custom'] $restore_type = 'recovery',
+  Enum['recovery', 'recovery-db', 'custom', 'migration'] $restore_type = 'recovery',
 
   # Which data to restore
   Peadm::Recovery_opts $restore = {},

plans/test_restore.pp

@@ -0,0 +1,66 @@
+plan peadm::test_restore (
+  # This plan should be run on the primary server
+  Peadm::SingleTargetSpec $target,
+
+  # Path to the recovery tarball
+  Pattern[/.*\.tar\.gz$/] $input_file,
+
+  # boolean flag to indicate clean target before restore
+  Boolean $clean_target = false,
+) {
+  peadm::assert_supported_bolt_version()
+
+  # prior to running this plan the user will have something like this:
+  # perform a backup on the source PE server:
+  # bolt plan run peadm::backup backup_type=migration -t festive-gantlet.delivery.puppetlabs.net --no-host-key-check 
+  #
+  # download the backup file locally:
+  # bolt file download /tmp/pe-backup-2025-01-30T091722Z.tar.gz /tmp --targets festive-gantlet.delivery.puppetlabs.net --no-host-key-check 
+  #
+  # upload the backup file to the target PE server:
+  # bolt file upload /tmp/festive-gantlet.delivery.puppetlabs.net/pe-backup-2025-01-30T091722Z.tar.gz /tmp --targets inspiring-reign.delivery.puppetlabs.net --no-host-key-check 
+  #
+  # install PE on the target server (using the "code_manager_auto_configure": true parameter)
+  # bolt plan run peadm::install primary_host=inspiring-reign.delivery.puppetlabs.net replica_host=inspiring-reign.delivery.puppetlabs.net code_manager_auto_configure=true --no-host-key-check
+  #
+  # run this plan to restore the backup on the target PE server:
+  # bolt plan run peadm::test_restore clean_target=false input_file=/tmp/festive-gantlet.delivery.puppetlabs.net/pe-backup-2025-01-30T135004Z.tar.gz target=inspiring-reign.delivery.puppetlabs.net --no-host-key-check 
+  #
+
+  $params = loadjson('params.json')
+  $primary_host = $params['primary_host']
+  out::message("params: ${params}")
+  out::message("primary_host:${primary_host}.")
+
+  # check param to see if we need to do an uninstall/install on the target
+  if $clean_target {
+    out::message('Cleaning target before restore')
+    run_plan('peadm::uninstall', $target)
+    run_plan('peadm::install', $params)
+    #run_plan('peadm::install', $target, { 'params' => '@params.json' })
+  } else {
+    out::message('Skipping clean target before restore')
+  }
+
+  # delete the file on the target first in case it is there
+  $file_name = basename($input_file)
+  $dir_name = $file_name[0,-8]
+  out::message("Deleting /tmp/${file_name} and /tmp/${dir_name} on ${target}")
+  run_command("rm -rf /tmp/${file_name}", $target)
+  run_command("rm -rf /tmp/${dir_name}", $target)
+
+  # copy the file over from the bolt controller to the target
+  out::message("Copying ${input_file} to /tmp/${file_name} on ${target}")
+  upload_file($input_file, "/tmp/${file_name}", $target)
+
+  # kick off the restore plan on the target
+  out::message("Running peadm::restore on ${target}")
+  run_plan('peadm::restore', $target, { 'input_file' => "/tmp/${file_name}", 'restore_type' => 'migration' })
+
+  # # add replica back
+  # run_plan('peadm::add_replica',
+  #   primary_host            => $primary_host,
+  #   replica_host            => $replica_host,
+  #   replica_postgresql_host => undef,
+  # )
+}