1 package MigrationTest
::Shared
;
8 use Socket
qw(AF_INET);
11 use PVE
::Tools
qw(file_set_contents file_get_contents lock_file_full);
13 my $RUN_DIR_PATH = $ENV{RUN_DIR_PATH
} or die "no RUN_DIR_PATH set\n";
15 my $storage_config = decode_json
(file_get_contents
("${RUN_DIR_PATH}/storage_config"));
16 my $replication_config = decode_json
(file_get_contents
("${RUN_DIR_PATH}/replication_config"));
17 my $fail_config = decode_json
(file_get_contents
("${RUN_DIR_PATH}/fail_config"));
18 my $migrate_params = decode_json
(file_get_contents
("${RUN_DIR_PATH}/migrate_params"));
19 my $test_vmid = $migrate_params->{vmid
};
23 sub add_target_volid
{
26 PVE
::Storage
::parse_volume_id
($volid);
28 lock_file_full
("${RUN_DIR_PATH}/target_volids.lock", undef, 0, sub {
29 my $target_volids = decode_json
(file_get_contents
("${RUN_DIR_PATH}/target_volids"));
30 die "target volid already present " if defined($target_volids->{$volid});
31 $target_volids->{$volid} = 1;
32 file_set_contents
("${RUN_DIR_PATH}/target_volids", to_json
($target_volids));
37 sub remove_target_volid
{
40 PVE
::Storage
::parse_volume_id
($volid);
42 lock_file_full
("${RUN_DIR_PATH}/target_volids.lock", undef, 0, sub {
43 my $target_volids = decode_json
(file_get_contents
("${RUN_DIR_PATH}/target_volids"));
44 die "target volid does not exist " if !defined($target_volids->{$volid});
45 delete $target_volids->{$volid};
46 file_set_contents
("${RUN_DIR_PATH}/target_volids", to_json
($target_volids));
51 my $mocked_cfs_read_file = sub {
54 if ($file eq 'datacenter.cfg') {
56 } elsif ($file eq 'replication.cfg') {
57 return $replication_config;
59 die "cfs_read_file (mocked) - implement me: $file\n";
64 our $cluster_module = Test
::MockModule-
>new("PVE::Cluster");
65 $cluster_module->mock(
66 cfs_read_file
=> $mocked_cfs_read_file,
67 check_cfs_quorum
=> sub {
72 our $ha_config_module = Test
::MockModule-
>new("PVE::HA::Config");
73 $ha_config_module->mock(
74 vm_is_ha_managed
=> sub {
79 our $qemu_config_module = Test
::MockModule-
>new("PVE::QemuConfig");
80 $qemu_config_module->mock(
81 assert_config_exists_on_node
=> sub {
85 my ($class, $vmid, $node) = @_;
86 die "trying to load wrong config: '$vmid'\n" if $vmid ne $test_vmid;
87 return decode_json
(file_get_contents
("${RUN_DIR_PATH}/vm_config"));
89 lock_config
=> sub { # no use locking here because lock is local to node
90 my ($self, $vmid, $code, @param) = @_;
91 return $code->(@param);
94 my ($class, $vmid, $conf) = @_;
95 die "trying to write wrong config: '$vmid'\n" if $vmid ne $test_vmid;
96 file_set_contents
("${RUN_DIR_PATH}/vm_config", to_json
($conf));
100 our $qemu_server_cloudinit_module = Test
::MockModule-
>new("PVE::QemuServer::Cloudinit");
101 $qemu_server_cloudinit_module->mock(
102 generate_cloudinitconfig
=> sub {
107 our $qemu_server_module = Test
::MockModule-
>new("PVE::QemuServer");
108 $qemu_server_module->mock(
109 clear_reboot_request
=> sub {
112 get_efivars_size
=> sub {
117 our $replication_module = Test
::MockModule-
>new("PVE::Replication");
118 $replication_module->mock(
119 run_replication
=> sub {
120 die "run_replication error" if $fail_config->{run_replication
};
122 my $vm_config = PVE
::QemuConfig-
>load_config($test_vmid);
123 return PVE
::QemuConfig-
>get_replicatable_volumes(
131 our $replication_config_module = Test
::MockModule-
>new("PVE::ReplicationConfig");
132 $replication_config_module->mock(
133 cfs_read_file
=> $mocked_cfs_read_file,
136 our $storage_module = Test
::MockModule-
>new("PVE::Storage");
137 $storage_module->mock(
138 activate_volumes
=> sub {
141 deactivate_volumes
=> sub {
145 return $storage_config;
147 get_bandwidth_limit
=> sub {
150 cfs_read_file
=> $mocked_cfs_read_file,
153 our $storage_plugin_module = Test
::MockModule-
>new("PVE::Storage::Plugin");
154 $storage_plugin_module->mock(
155 cluster_lock_storage
=> sub {
156 my ($class, $storeid, $shared, $timeout, $func, @param) = @_;
158 mkdir "${RUN_DIR_PATH}/lock";
160 my $path = "${RUN_DIR_PATH}/lock/pve-storage-${storeid}";
161 return PVE
::Tools
::lock_file
($path, $timeout, $func, @param);
165 our $systemd_module = Test
::MockModule-
>new("PVE::Systemd");
166 $systemd_module->mock(
167 wait_for_unit_removed
=> sub {
170 enter_systemd_scope
=> sub {
175 my $migrate_port_counter = 60000;
177 our $tools_module = Test
::MockModule-
>new("PVE::Tools");
179 get_host_address_family
=> sub {
182 next_migrate_port
=> sub {
183 return $migrate_port_counter++;