-package PVE::QemuServer;
+package PVE::QemuServer; ## no critic
use strict;
use warnings;
use PVE::Storage;
use PVE::Storage::Plugin;
use PVE::QemuServer;
+use PVE::QemuConfig;
use PVE::Tools;
+use PVE::ReplicationConfig;
use Test::MockModule;
use Test::More;
die "volume_rollback_is_possible failed\n";
}
+sub mocked_activate_volumes {
+ my ($storecfg, $volumes) = @_;
+ die "Storage config not mocked! aborting\n"
+ if defined($storecfg);
+ die "wrong volume - fake vmstate expected!\n"
+ if ((scalar @$volumes != 1) || @$volumes[0] ne "somestorage:state-volume");
+ return;
+}
+
+sub mocked_deactivate_volumes {
+ my ($storecfg, $volumes) = @_;
+ die "Storage config not mocked! aborting\n"
+ if defined($storecfg);
+ die "wrong volume - fake vmstate expected!\n"
+ if ((scalar @$volumes != 1) || @$volumes[0] ne "somestorage:state-volume");
+ return;
+}
+
sub mocked_vdisk_free {
my ($storecfg, $vmstate) = @_;
die "Storage config not mocked! aborting\n"
plan tests => 2;
$@ = undef;
eval {
- PVE::QemuServer::snapshot_prepare($vmid, $snapname, $save_vmstate, $comment);
+ PVE::QemuConfig->__snapshot_prepare($vmid, $snapname, $save_vmstate, $comment);
};
is($@, $exp_err, "\$@ correct");
ok(test_file("snapshot-expected/prepare/qemu-server/$vmid.conf", "snapshot-working/prepare/qemu-server/$vmid.conf"), "config file correct");
plan tests => 2;
$@ = undef;
eval {
- PVE::QemuServer::snapshot_commit($vmid, $snapname);
+ PVE::QemuConfig->__snapshot_commit($vmid, $snapname);
};
is($@, $exp_err, "\$@ correct");
ok(test_file("snapshot-expected/commit/qemu-server/$vmid.conf", "snapshot-working/commit/qemu-server/$vmid.conf"), "config file correct");
$exp_vol_snap_delete = {} if !defined($exp_vol_snap_delete);
$@ = undef;
eval {
- PVE::QemuServer::snapshot_create($vmid, $snapname, $save_vmstate, $comment);
+ PVE::QemuConfig->snapshot_create($vmid, $snapname, $save_vmstate, $comment);
};
is($@, $exp_err, "\$@ correct");
is_deeply($vol_snapshot, $exp_vol_snap, "created correct volume snapshots");
$exp_vol_snap_delete = {} if !defined($exp_vol_snap_delete);
$@ = undef;
eval {
- PVE::QemuServer::snapshot_delete($vmid, $snapname, $force);
+ PVE::QemuConfig->snapshot_delete($vmid, $snapname, $force);
};
is($@, $exp_err, "\$@ correct");
is_deeply($vol_snapshot_delete, $exp_vol_snap_delete, "deleted correct volume snapshots");
$exp_vol_snap_rollback = {} if !defined($exp_vol_snap_rollback);
$@ = undef;
eval {
- PVE::QemuServer::snapshot_rollback($vmid, $snapname);
+ PVE::QemuConfig->snapshot_rollback($vmid, $snapname);
};
is($@, $exp_err, "\$@ correct");
is_deeply($vol_snapshot_rollback, $exp_vol_snap_rollback, "rolled back to correct volume snapshots");
};
}
-# BEGIN redefine PVE::QemuServer methods
+# BEGIN mocked PVE::QemuConfig methods
sub config_file_lock {
return "snapshot-working/pve-test.lock";
}
sub cfs_config_path {
- my ($vmid, $node) = @_;
+ my ($class, $vmid, $node) = @_;
$node = $nodename if !$node;
return "snapshot-working/$node/qemu-server/$vmid.conf";
}
sub load_config {
- my ($vmid, $node) = @_;
+ my ($class, $vmid, $node) = @_;
- my $filename = cfs_config_path($vmid, $node);
+ my $filename = $class->cfs_config_path($vmid, $node);
my $raw = PVE::Tools::file_get_contents($filename);
}
sub write_config {
- my ($vmid, $conf) = @_;
+ my ($class, $vmid, $conf) = @_;
- my $filename = cfs_config_path($vmid);
+ my $filename = $class->cfs_config_path($vmid);
if ($conf->{snapshots}) {
foreach my $snapname (keys %{$conf->{snapshots}}) {
}
sub has_feature {
- my ($feature, $conf, $storecfg, $snapname, $running, $backup_only) = @_;
+ my ($class, $feature, $conf, $storecfg, $snapname, $running, $backup_only) = @_;
return $snapshot_possible;
}
-sub check_running {
- return $running;
-}
-
-sub snapshot_save_vmstate {
- my ($vmid, $conf, $snapname, $storecfg) = @_;
+sub __snapshot_save_vmstate {
+ my ($class, $vmid, $conf, $snapname, $storecfg) = @_;
die "save_vmstate failed\n"
if !$save_vmstate_works;
my $snap = $conf->{snapshots}->{$snapname};
$snap->{vmstate} = "somestorage:state-volume";
- $snap->{machine} = "somemachine";
+ $snap->{runningmachine} = "somemachine"
}
-sub do_snapshots_with_qemu {
- return 0;
+sub assert_config_exists_on_node {
+ my ($vmid, $node) = @_;
+ return -f cfs_config_path("PVE::QemuConfig", $vmid, $node);
}
+# END mocked PVE::QemuConfig methods
-sub vm_qmp_command {
- my ($vmid, $cmd, $nocheck) = @_;
+# BEGIN mocked PVE::QemuServer::Helpers methods
+
+sub vm_running_locally {
+ return $running;
+}
+
+# END mocked PVE::QemuServer::Helpers methods
+
+# BEGIN mocked PVE::QemuServer::Monitor methods
+
+sub qmp_cmd {
+ my ($vmid, $cmd) = @_;
my $exec = $cmd->{execute};
if ($exec eq "delete-drive-snapshot") {
return;
}
if ($exec eq "query-savevm") {
- return { "status" => "completed" };
+ return {
+ "status" => "completed",
+ "bytes" => 1024*1024*1024,
+ "total-time" => 5000,
+ };
}
die "unexpected vm_qmp_command!\n";
}
+# END mocked PVE::QemuServer::Monitor methods
+
+# BEGIN redefine PVE::QemuServer methods
+
+sub do_snapshots_with_qemu {
+ return 0;
+}
+
sub vm_start {
- my ($storecfg, $vmid, $statefile, $skiplock, $migratedfrom, $paused, $forcemachine) = @_;
+ my ($storecfg, $vmid, $params, $migrate_opts) = @_;
die "Storage config not mocked! aborting\n"
if defined($storecfg);
die "statefile and forcemachine must be both defined or undefined! aborting\n"
- if defined($statefile) xor defined($forcemachine);
+ if defined($params->{statefile}) xor defined($params->{forcemachine});
return;
}
PVE::Tools::run_command("rm -rf snapshot-working");
PVE::Tools::run_command("cp -a snapshot-input snapshot-working");
+my $qemu_helpers_module = Test::MockModule->new('PVE::QemuServer::Helpers');
+$qemu_helpers_module->mock('vm_running_locally', \&vm_running_locally);
+
+my $qemu_monitor_module = Test::MockModule->new('PVE::QemuServer::Monitor');
+$qemu_monitor_module->mock('qmp_cmd', \&qmp_cmd);
+
+my $qemu_config_module = Test::MockModule->new('PVE::QemuConfig');
+$qemu_config_module->mock('config_file_lock', \&config_file_lock);
+$qemu_config_module->mock('cfs_config_path', \&cfs_config_path);
+$qemu_config_module->mock('load_config', \&load_config);
+$qemu_config_module->mock('write_config', \&write_config);
+$qemu_config_module->mock('has_feature', \&has_feature);
+$qemu_config_module->mock('__snapshot_save_vmstate', \&__snapshot_save_vmstate);
+$qemu_config_module->mock('assert_config_exists_on_node', \&assert_config_exists_on_node);
+
+# ignore existing replication config
+my $repl_config_module = Test::MockModule->new('PVE::ReplicationConfig');
+$repl_config_module->mock('new' => sub { return bless {}, "PVE::ReplicationConfig" });
+$repl_config_module->mock('check_for_existing_jobs' => sub { return });
+
+my $storage_module = Test::MockModule->new('PVE::Storage');
+$storage_module->mock('config', sub { return; });
+$storage_module->mock('path', sub { return "/some/store/statefile/path"; });
+$storage_module->mock('activate_volumes', \&mocked_activate_volumes);
+$storage_module->mock('deactivate_volumes', \&mocked_deactivate_volumes);
+$storage_module->mock('vdisk_free', \&mocked_vdisk_free);
+$storage_module->mock('volume_snapshot', \&mocked_volume_snapshot);
+$storage_module->mock('volume_snapshot_delete', \&mocked_volume_snapshot_delete);
+$storage_module->mock('volume_snapshot_rollback', \&mocked_volume_snapshot_rollback);
+$storage_module->mock('volume_rollback_is_possible', \&mocked_volume_rollback_is_possible);
+
$running = 1;
$freeze_possible = 1;
$save_vmstate_works = 1;
# possible, but fails
$vol_snapshot_rollback_possible->{"local:snapshotable-disk-4"} = 1;
-printf("\n");
-printf("Setting up Mocking for PVE::Storage\n");
-my $storage_module = new Test::MockModule('PVE::Storage');
-$storage_module->mock('config', sub { return undef; });
-$storage_module->mock('path', sub { return "/some/store/statefile/path"; });
-$storage_module->mock('vdisk_free', \&mocked_vdisk_free);
-$storage_module->mock('volume_snapshot', \&mocked_volume_snapshot);
-$storage_module->mock('volume_snapshot_delete', \&mocked_volume_snapshot_delete);
-$storage_module->mock('volume_snapshot_rollback', \&mocked_volume_snapshot_rollback);
-$storage_module->mock('volume_rollback_is_possible', \&mocked_volume_rollback_is_possible);
-printf("\tconfig(), volume_snapshot(), volume_snapshot_delete(), volume_snapshot_rollback() and volume_rollback_is_possible() mocked\n");
#printf("\n");
#printf("Setting up Mocking for PVE::Tools\n");
-#my $tools_module = new Test::MockModule('PVE::Tools');
+#my $tools_module = Test::MockModule->new('PVE::Tools');
#$tools_module->mock('run_command' => \&mocked_run_command);
#printf("\trun_command() mocked\n");
#
$freeze_possible = 1;
printf("Expected error for snapshot_create when volume snapshot is not possible\n");
-testcase_create("201", "test", 0, "test comment", "volume snapshot disabled\n");
+testcase_create("201", "test", 0, "test comment", "volume snapshot disabled\n\n");
printf("Expected error for snapshot_create when volume snapshot is not possible for one drive\n");
-testcase_create("202", "test", 0, "test comment", "volume snapshot disabled\n", { "local:snapshotable-disk-1" => "test" }, { "local:snapshotable-disk-1" => "test" });
+testcase_create("202", "test", 0, "test comment", "volume snapshot disabled\n\n", { "local:snapshotable-disk-1" => "test" }, { "local:snapshotable-disk-1" => "test" });
$vm_mon->{savevm_start} = 0;
printf("Expected error for snapshot_create when Qemu mon command 'savevm-start' fails\n");
-testcase_create("203", "test", 0, "test comment", "savevm-start disabled\n");
+testcase_create("203", "test", 0, "test comment", "savevm-start disabled\n\n");
$vm_mon->{savevm_start} = 1;
+printf("Successful snapshot_create with no existing snapshots but set machine type\n");
+testcase_create("301", "test", 1, "test comment", "", { "local:snapshotable-disk-1" => "test" });
$nodename = "delete";
printf("\n");
printf("Expected error for snapshot_rollback with mp rollback failure (results in inconsistent state)\n");
testcase_rollback("207", "test", "volume snapshot rollback disabled\n", { "local:snapshotable-disk-1" => "test", "local:snapshotable-disk-2" => "test" });
+printf("Successful snapshot_rollback with saved vmstate and machine config only in snapshot\n");
+testcase_rollback("301", "test", "", { "local:snapshotable-disk-1" => "test" });
+
+printf("Successful snapshot_rollback with saved vmstate and machine config and runningmachine \n");
+testcase_rollback("302", "test", "", { "local:snapshotable-disk-1" => "test" });
+
done_testing();
+
+1;