mirror of
https://git.proxmox.com/git/qemu-server
synced 2025-05-03 07:18:36 +00:00
48831384b8
and the associated parts for 'qm start'. Each test will first populate the MigrationTest/run directory with the relevant configuration files and files keeping track of the state of everything necessary. Second, the mock-script for migration is executed, which in turn will execute the 'qm start' mock-script (if it's an online test that gets far enough). The scripts will simulate a migration and update the relevant files in the MigrationTest/run directory. Finally, the main test script will evaluate the state. The main checks are the volume IDs on the source and target and the VM configuration itself. Additional checks are the vm_status and expected_calls, keeping track if certain calls have been made. The rationale behind creating two mock-scripts is two-fold: 1. It removes the need to hard code responses for the tunnel and to recycle logic for determining and allocating migration volumes. Some of that logic already happens in the API part, so it was necessary to mock the whole CLI-Handler. 2. It allows testing the code relevant for migration in 'qm start' as well, and it should even be possible to test different versions of the mock-scripts against each other. With a bit of extra work and things like 'git worktree', it might even be possible to automate this. The helper get_patched config is useful to change pre-defined configuration files on the fly, avoiding the new to explicitly define whole configurations to test for something in many cases. Signed-off-by: Fabian Ebner <f.ebner@proxmox.com>
143 lines
3.3 KiB
Perl
143 lines
3.3 KiB
Perl
package MigrationTest::QmMock;
|
|
|
|
use strict;
|
|
use warnings;
|
|
|
|
use JSON;
|
|
use Test::MockModule;
|
|
|
|
use MigrationTest::Shared;
|
|
|
|
use PVE::API2::Qemu;
|
|
use PVE::Storage;
|
|
use PVE::Tools qw(file_set_contents file_get_contents);
|
|
|
|
use PVE::CLIHandler;
|
|
use base qw(PVE::CLIHandler);
|
|
|
|
my $RUN_DIR_PATH = $ENV{RUN_DIR_PATH} or die "no RUN_DIR_PATH set\n";
|
|
|
|
my $target_volids = decode_json(file_get_contents("${RUN_DIR_PATH}/target_volids"));
|
|
my $fail_config = decode_json(file_get_contents("${RUN_DIR_PATH}/fail_config"));
|
|
my $migrate_params = decode_json(file_get_contents("${RUN_DIR_PATH}/migrate_params"));
|
|
my $nodename = $migrate_params->{target};
|
|
|
|
my $kvm_exectued = 0;
|
|
|
|
sub setup_environment {
|
|
my $rpcenv = PVE::RPCEnvironment::init('MigrationTest::QmMock', 'cli');
|
|
}
|
|
|
|
# mock RPCEnvironment directly
|
|
|
|
sub get_user {
|
|
return 'root@pam';
|
|
}
|
|
|
|
sub fork_worker {
|
|
my ($self, $dtype, $id, $user, $function, $background) = @_;
|
|
$function->(123456);
|
|
return '123456';
|
|
}
|
|
|
|
# mocked modules
|
|
|
|
my $inotify_module = Test::MockModule->new("PVE::INotify");
|
|
$inotify_module->mock(
|
|
nodename => sub {
|
|
return $nodename;
|
|
},
|
|
);
|
|
|
|
$MigrationTest::Shared::qemu_server_module->mock(
|
|
nodename => sub {
|
|
return $nodename;
|
|
},
|
|
config_to_command => sub {
|
|
return [ 'mocked_kvm_command' ];
|
|
},
|
|
);
|
|
|
|
my $qemu_server_helpers_module = Test::MockModule->new("PVE::QemuServer::Helpers");
|
|
$qemu_server_helpers_module->mock(
|
|
vm_running_locally => sub {
|
|
return $kvm_exectued;
|
|
},
|
|
);
|
|
|
|
# to make sure we get valid and predictable names
|
|
my $disk_counter = 10;
|
|
|
|
$MigrationTest::Shared::storage_module->mock(
|
|
vdisk_alloc => sub {
|
|
my ($cfg, $storeid, $vmid, $fmt, $name, $size) = @_;
|
|
|
|
die "vdisk_alloc (mocked) - name is not expected to be set - implement me\n"
|
|
if defined($name);
|
|
|
|
my $name_without_extension = "vm-${vmid}-disk-${disk_counter}";
|
|
$disk_counter++;
|
|
|
|
my $volid;
|
|
my $scfg = PVE::Storage::storage_config($cfg, $storeid);
|
|
if ($scfg->{path}) {
|
|
$volid = "${storeid}:${vmid}/${name_without_extension}.${fmt}";
|
|
} else {
|
|
$volid = "${storeid}:${name_without_extension}";
|
|
}
|
|
|
|
die "vdisk_alloc '$volid' error\n" if $fail_config->{vdisk_alloc}
|
|
&& $fail_config->{vdisk_alloc} eq $volid;
|
|
|
|
MigrationTest::Shared::add_target_volid($volid);
|
|
|
|
return $volid;
|
|
},
|
|
);
|
|
|
|
$MigrationTest::Shared::qemu_server_module->mock(
|
|
mon_cmd => sub {
|
|
my ($vmid, $command, %params) = @_;
|
|
|
|
if ($command eq 'nbd-server-start') {
|
|
return;
|
|
} elsif ($command eq 'nbd-server-add') {
|
|
return;
|
|
} elsif ($command eq 'qom-set') {
|
|
return;
|
|
}
|
|
die "mon_cmd (mocked) - implement me: $command";
|
|
},
|
|
run_command => sub {
|
|
my ($cmd_full, %param) = @_;
|
|
|
|
my $cmd_msg = to_json($cmd_full);
|
|
|
|
my $cmd = shift @{$cmd_full};
|
|
|
|
if ($cmd eq '/bin/systemctl') {
|
|
return;
|
|
} elsif ($cmd eq 'mocked_kvm_command') {
|
|
$kvm_exectued = 1;
|
|
return 0;
|
|
}
|
|
die "run_command (mocked) - implement me: ${cmd_msg}";
|
|
},
|
|
set_migration_caps => sub {
|
|
return;
|
|
},
|
|
vm_migrate_alloc_nbd_disks => sub{
|
|
my $nbd = $MigrationTest::Shared::qemu_server_module->original('vm_migrate_alloc_nbd_disks')->(@_);
|
|
file_set_contents("${RUN_DIR_PATH}/nbd_info", to_json($nbd));
|
|
return $nbd;
|
|
},
|
|
);
|
|
|
|
our $cmddef = {
|
|
start => [ "PVE::API2::Qemu", 'vm_start', ['vmid'], { node => $nodename } ],
|
|
};
|
|
|
|
MigrationTest::QmMock->run_cli_handler();
|
|
|
|
1;
|