150 lines
3.5 KiB
Perl
Executable File
150 lines
3.5 KiB
Perl
Executable File
#!/usr/bin/perl
|
|
|
|
# Note:
|
|
# 1.) Start replication job with single disk
|
|
# 2.) add non-existent disk (replication fails)
|
|
# 3.) create disk (replication continues).
|
|
# 4.) remove job
|
|
|
|
use strict;
|
|
use warnings;
|
|
use JSON;
|
|
|
|
use lib ('.', '../..');
|
|
|
|
use Test::MockModule;
|
|
use ReplicationTestEnv;
|
|
|
|
use PVE::Tools;
|
|
|
|
$ReplicationTestEnv::mocked_nodename = 'node1';
|
|
|
|
use PVE::INotify;
|
|
use PVE::Cluster;
|
|
use PVE::QemuConfig;
|
|
use PVE::QemuServer;
|
|
use PVE::LXC::Config;
|
|
use PVE::LXC;
|
|
use PVE::Storage;
|
|
|
|
my $replicated_volume_status = {};
|
|
|
|
my $mocked_remote_prepare_local_job = sub {
|
|
my ($ssh_info, $jobid, $vmid, $volumes, $storeid_list, $last_sync, $parent_snapname, $force) = @_;
|
|
|
|
my $target = $ssh_info->{node};
|
|
|
|
my $last_snapshots = {};
|
|
|
|
return $last_snapshots if !defined($replicated_volume_status->{$target});
|
|
|
|
my $last_sync_snapname = PVE::ReplicationState::replication_snapshot_name($jobid, $last_sync);
|
|
|
|
foreach my $volid (keys %{$replicated_volume_status->{$target}}) {
|
|
if (!grep { $_ eq $volid } @$volumes) {
|
|
delete $replicated_volume_status->{$target}->{$volid};
|
|
next;
|
|
}
|
|
my $snapname = $replicated_volume_status->{$target}->{$volid};
|
|
|
|
$last_snapshots->{$volid}->{$snapname} = 1 if $last_sync_snapname eq $snapname;
|
|
}
|
|
|
|
return $last_snapshots;
|
|
};
|
|
|
|
my $mocked_remote_finalize_local_job = sub {
|
|
my ($ssh_info, $jobid, $vmid, $volumes, $last_sync) = @_;
|
|
|
|
# do nothing
|
|
};
|
|
|
|
my $mocked_replicate_volume = sub {
|
|
my ($ssh_info, $storecfg, $volid, $base_snapshot, $sync_snapname) = @_;
|
|
|
|
my $target = $ssh_info->{node};
|
|
|
|
$replicated_volume_status->{$target}->{$volid} = $sync_snapname;
|
|
};
|
|
|
|
my $mocked_delete_job = sub {
|
|
my ($jobid) = @_;
|
|
|
|
delete $ReplicationTestEnv::mocked_replication_jobs->{$jobid};
|
|
};
|
|
|
|
my $pve_replication_config_module = Test::MockModule->new('PVE::ReplicationConfig');
|
|
$pve_replication_config_module->mock(delete_job => $mocked_delete_job);
|
|
|
|
my $pve_replication_module = Test::MockModule->new('PVE::Replication');
|
|
$pve_replication_module->mock(
|
|
remote_prepare_local_job => $mocked_remote_prepare_local_job,
|
|
remote_finalize_local_job => $mocked_remote_finalize_local_job,
|
|
replicate_volume => $mocked_replicate_volume);
|
|
|
|
my $testjob = {
|
|
'type' => 'local',
|
|
'target' => 'node1',
|
|
'guest' => 900,
|
|
};
|
|
|
|
$ReplicationTestEnv::mocked_replication_jobs = {
|
|
job_900_to_node2 => {
|
|
'type' => 'local',
|
|
'target' => 'node2',
|
|
'guest' => 900,
|
|
},
|
|
};
|
|
|
|
$ReplicationTestEnv::mocked_vm_configs = {
|
|
900 => {
|
|
node => 'node1',
|
|
snapshots => {},
|
|
ide0 => 'local-zfs:vm-900-disk-1,size=4G',
|
|
memory => 512,
|
|
ide2 => 'none,media=cdrom',
|
|
},
|
|
};
|
|
|
|
ReplicationTestEnv::setup();
|
|
|
|
ReplicationTestEnv::register_mocked_volid('local-zfs:vm-900-disk-1');
|
|
|
|
my $ctime = 1000;
|
|
|
|
my $status;
|
|
|
|
ReplicationTestEnv::openlog();
|
|
|
|
for (my $i = 0; $i < 15; $i++) {
|
|
ReplicationTestEnv::track_jobs($ctime);
|
|
$ctime += 60;
|
|
}
|
|
|
|
# add a new, disk (but disk does not exist, so replication fails)
|
|
$ReplicationTestEnv::mocked_vm_configs->{900}->{ide1} = 'local-zfs:vm-900-disk-2,size=4G';
|
|
for (my $i = 0; $i < 15; $i++) {
|
|
ReplicationTestEnv::track_jobs($ctime);
|
|
$ctime += 60;
|
|
}
|
|
|
|
# register disk, so replication should succeed
|
|
ReplicationTestEnv::register_mocked_volid('local-zfs:vm-900-disk-2');
|
|
for (my $i = 0; $i < 15; $i++) {
|
|
ReplicationTestEnv::track_jobs($ctime);
|
|
$ctime += 60;
|
|
}
|
|
|
|
# mark job for removal
|
|
$ReplicationTestEnv::mocked_replication_jobs->{job_900_to_node2}->{remove_job} = 'full';
|
|
for (my $i = 0; $i < 15; $i++) {
|
|
ReplicationTestEnv::track_jobs($ctime);
|
|
$ctime += 60;
|
|
}
|
|
|
|
|
|
|
|
ReplicationTestEnv::commit_log();
|
|
|
|
exit(0);
|