]>
Commit | Line | Data |
---|---|---|
2bf945fc | 1 | package PVE::QemuServer; ## no critic |
a157d0fd FG |
2 | |
3 | use strict; | |
4 | use warnings; | |
5 | ||
6 | use lib qw(..); | |
7 | ||
8 | use PVE::Storage; | |
9 | use PVE::Storage::Plugin; | |
10 | use PVE::QemuServer; | |
ffda963f | 11 | use PVE::QemuConfig; |
a157d0fd | 12 | use PVE::Tools; |
d79a3e54 | 13 | use PVE::ReplicationConfig; |
a157d0fd FG |
14 | |
15 | use Test::MockModule; | |
16 | use Test::More; | |
17 | ||
18 | my $nodename; | |
19 | my $snapshot_possible; | |
20 | my $vol_snapshot_possible = {}; | |
21 | my $vol_snapshot_delete_possible = {}; | |
22 | my $vol_snapshot_rollback_possible = {}; | |
23 | my $vol_snapshot_rollback_enabled = {}; | |
24 | my $vol_snapshot = {}; | |
25 | my $vol_snapshot_delete = {}; | |
26 | my $vol_snapshot_rollback = {}; | |
27 | my $running; | |
28 | my $freeze_possible; | |
29 | my $stop_possible; | |
30 | my $save_vmstate_works; | |
31 | my $vm_mon = {}; | |
32 | ||
33 | # Mocked methods | |
34 | ||
35 | sub mocked_volume_snapshot { | |
36 | my ($storecfg, $volid, $snapname) = @_; | |
37 | die "Storage config not mocked! aborting\n" | |
38 | if defined($storecfg); | |
39 | die "volid undefined\n" | |
40 | if !defined($volid); | |
41 | die "snapname undefined\n" | |
42 | if !defined($snapname); | |
43 | if ($vol_snapshot_possible->{$volid}) { | |
44 | if (defined($vol_snapshot->{$volid})) { | |
45 | $vol_snapshot->{$volid} .= ",$snapname"; | |
46 | } else { | |
47 | $vol_snapshot->{$volid} = $snapname; | |
48 | } | |
49 | return 1; | |
50 | } else { | |
51 | die "volume snapshot disabled\n"; | |
52 | } | |
53 | } | |
54 | ||
55 | sub mocked_volume_snapshot_delete { | |
56 | my ($storecfg, $volid, $snapname) = @_; | |
57 | die "Storage config not mocked! aborting\n" | |
58 | if defined($storecfg); | |
59 | die "volid undefined\n" | |
60 | if !defined($volid); | |
61 | die "snapname undefined\n" | |
62 | if !defined($snapname); | |
63 | if ($vol_snapshot_delete_possible->{$volid}) { | |
64 | if (defined($vol_snapshot_delete->{$volid})) { | |
65 | $vol_snapshot_delete->{$volid} .= ",$snapname"; | |
66 | } else { | |
67 | $vol_snapshot_delete->{$volid} = $snapname; | |
68 | } | |
69 | return 1; | |
70 | } else { | |
71 | die "volume snapshot delete disabled\n"; | |
72 | } | |
73 | } | |
74 | ||
75 | sub mocked_volume_snapshot_rollback { | |
76 | my ($storecfg, $volid, $snapname) = @_; | |
77 | die "Storage config not mocked! aborting\n" | |
78 | if defined($storecfg); | |
79 | die "volid undefined\n" | |
80 | if !defined($volid); | |
81 | die "snapname undefined\n" | |
82 | if !defined($snapname); | |
83 | if ($vol_snapshot_rollback_enabled->{$volid}) { | |
84 | if (defined($vol_snapshot_rollback->{$volid})) { | |
85 | $vol_snapshot_rollback->{$volid} .= ",$snapname"; | |
86 | } else { | |
87 | $vol_snapshot_rollback->{$volid} = $snapname; | |
88 | } | |
89 | return 1; | |
90 | } else { | |
91 | die "volume snapshot rollback disabled\n"; | |
92 | } | |
93 | } | |
94 | ||
95 | sub mocked_volume_rollback_is_possible { | |
96 | my ($storecfg, $volid, $snapname) = @_; | |
97 | die "Storage config not mocked! aborting\n" | |
98 | if defined($storecfg); | |
99 | die "volid undefined\n" | |
100 | if !defined($volid); | |
101 | die "snapname undefined\n" | |
102 | if !defined($snapname); | |
103 | return $vol_snapshot_rollback_possible->{$volid} | |
104 | if ($vol_snapshot_rollback_possible->{$volid}); | |
105 | die "volume_rollback_is_possible failed\n"; | |
106 | } | |
107 | ||
101a1b3f FG |
108 | sub mocked_activate_volumes { |
109 | my ($storecfg, $volumes) = @_; | |
110 | die "Storage config not mocked! aborting\n" | |
111 | if defined($storecfg); | |
112 | die "wrong volume - fake vmstate expected!\n" | |
113 | if ((scalar @$volumes != 1) || @$volumes[0] ne "somestorage:state-volume"); | |
114 | return; | |
115 | } | |
116 | ||
117 | sub mocked_deactivate_volumes { | |
118 | my ($storecfg, $volumes) = @_; | |
119 | die "Storage config not mocked! aborting\n" | |
120 | if defined($storecfg); | |
121 | die "wrong volume - fake vmstate expected!\n" | |
122 | if ((scalar @$volumes != 1) || @$volumes[0] ne "somestorage:state-volume"); | |
123 | return; | |
124 | } | |
125 | ||
a157d0fd FG |
126 | sub mocked_vdisk_free { |
127 | my ($storecfg, $vmstate) = @_; | |
128 | die "Storage config not mocked! aborting\n" | |
129 | if defined($storecfg); | |
130 | die "wrong vdisk - fake vmstate expected!\n" | |
131 | if ($vmstate ne "somestorage:state-volume"); | |
132 | return; | |
133 | } | |
134 | ||
135 | sub mocked_run_command { | |
136 | my ($cmd, %param) = @_; | |
137 | my $cmdstring; | |
138 | if (my $ref = ref($cmd)) { | |
139 | $cmdstring = PVE::Tools::cmd2string($cmd); | |
140 | if ($cmdstring =~ m/.*\/qemu-(un)?freeze.*/) { | |
141 | return 1 if $freeze_possible; | |
142 | die "qemu-[un]freeze disabled\n"; | |
143 | } | |
144 | if ($cmdstring =~ m/.*\/qemu-stop.*--kill.*/) { | |
145 | if ($stop_possible) { | |
146 | $running = 0; | |
147 | return 1; | |
148 | } else { | |
149 | return 0; | |
150 | } | |
151 | } | |
152 | } | |
153 | die "unexpected run_command call: '$cmdstring', aborting\n"; | |
154 | } | |
155 | ||
156 | # Testing methods | |
157 | ||
158 | sub test_file { | |
159 | my ($exp_fn, $real_fn) = @_; | |
160 | my $ret; | |
161 | eval { | |
162 | $ret = system("diff -u '$exp_fn' '$real_fn'"); | |
163 | }; | |
164 | die if $@; | |
165 | return !$ret; | |
166 | } | |
167 | ||
168 | sub testcase_prepare { | |
169 | my ($vmid, $snapname, $save_vmstate, $comment, $exp_err) = @_; | |
170 | subtest "Preparing snapshot '$snapname' for vm '$vmid'" => sub { | |
171 | plan tests => 2; | |
172 | $@ = undef; | |
173 | eval { | |
b2c9558d | 174 | PVE::QemuConfig->__snapshot_prepare($vmid, $snapname, $save_vmstate, $comment); |
a157d0fd FG |
175 | }; |
176 | is($@, $exp_err, "\$@ correct"); | |
177 | ok(test_file("snapshot-expected/prepare/qemu-server/$vmid.conf", "snapshot-working/prepare/qemu-server/$vmid.conf"), "config file correct"); | |
178 | }; | |
179 | } | |
180 | ||
181 | sub testcase_commit { | |
182 | my ($vmid, $snapname, $exp_err) = @_; | |
183 | subtest "Committing snapshot '$snapname' for vm '$vmid'" => sub { | |
184 | plan tests => 2; | |
185 | $@ = undef; | |
186 | eval { | |
b2c9558d | 187 | PVE::QemuConfig->__snapshot_commit($vmid, $snapname); |
a157d0fd FG |
188 | }; |
189 | is($@, $exp_err, "\$@ correct"); | |
190 | ok(test_file("snapshot-expected/commit/qemu-server/$vmid.conf", "snapshot-working/commit/qemu-server/$vmid.conf"), "config file correct"); | |
191 | } | |
192 | } | |
193 | ||
194 | sub testcase_create { | |
195 | my ($vmid, $snapname, $save_vmstate, $comment, $exp_err, $exp_vol_snap, $exp_vol_snap_delete) = @_; | |
196 | subtest "Creating snapshot '$snapname' for vm '$vmid'" => sub { | |
197 | plan tests => 4; | |
198 | $vol_snapshot = {}; | |
199 | $vol_snapshot_delete = {}; | |
200 | $exp_vol_snap = {} if !defined($exp_vol_snap); | |
201 | $exp_vol_snap_delete = {} if !defined($exp_vol_snap_delete); | |
202 | $@ = undef; | |
203 | eval { | |
b2c9558d | 204 | PVE::QemuConfig->snapshot_create($vmid, $snapname, $save_vmstate, $comment); |
a157d0fd FG |
205 | }; |
206 | is($@, $exp_err, "\$@ correct"); | |
207 | is_deeply($vol_snapshot, $exp_vol_snap, "created correct volume snapshots"); | |
208 | is_deeply($vol_snapshot_delete, $exp_vol_snap_delete, "deleted correct volume snapshots"); | |
209 | ok(test_file("snapshot-expected/create/qemu-server/$vmid.conf", "snapshot-working/create/qemu-server/$vmid.conf"), "config file correct"); | |
210 | }; | |
211 | } | |
212 | ||
213 | sub testcase_delete { | |
214 | my ($vmid, $snapname, $force, $exp_err, $exp_vol_snap_delete) = @_; | |
215 | subtest "Deleting snapshot '$snapname' of vm '$vmid'" => sub { | |
216 | plan tests => 3; | |
217 | $vol_snapshot_delete = {}; | |
218 | $exp_vol_snap_delete = {} if !defined($exp_vol_snap_delete); | |
219 | $@ = undef; | |
220 | eval { | |
b2c9558d | 221 | PVE::QemuConfig->snapshot_delete($vmid, $snapname, $force); |
a157d0fd FG |
222 | }; |
223 | is($@, $exp_err, "\$@ correct"); | |
224 | is_deeply($vol_snapshot_delete, $exp_vol_snap_delete, "deleted correct volume snapshots"); | |
225 | ok(test_file("snapshot-expected/delete/qemu-server/$vmid.conf", "snapshot-working/delete/qemu-server/$vmid.conf"), "config file correct"); | |
226 | }; | |
227 | } | |
228 | ||
229 | sub testcase_rollback { | |
230 | my ($vmid, $snapname, $exp_err, $exp_vol_snap_rollback) = @_; | |
231 | subtest "Rolling back to snapshot '$snapname' of vm '$vmid'" => sub { | |
232 | plan tests => 3; | |
233 | $vol_snapshot_rollback = {}; | |
234 | $running = 1; | |
235 | $exp_vol_snap_rollback = {} if !defined($exp_vol_snap_rollback); | |
236 | $@ = undef; | |
237 | eval { | |
b2c9558d | 238 | PVE::QemuConfig->snapshot_rollback($vmid, $snapname); |
a157d0fd FG |
239 | }; |
240 | is($@, $exp_err, "\$@ correct"); | |
241 | is_deeply($vol_snapshot_rollback, $exp_vol_snap_rollback, "rolled back to correct volume snapshots"); | |
242 | ok(test_file("snapshot-expected/rollback/qemu-server/$vmid.conf", "snapshot-working/rollback/qemu-server/$vmid.conf"), "config file correct"); | |
243 | }; | |
244 | } | |
245 | ||
ffda963f | 246 | # BEGIN mocked PVE::QemuConfig methods |
a157d0fd FG |
247 | sub config_file_lock { |
248 | return "snapshot-working/pve-test.lock"; | |
249 | } | |
250 | ||
251 | sub cfs_config_path { | |
ffda963f | 252 | my ($class, $vmid, $node) = @_; |
a157d0fd FG |
253 | |
254 | $node = $nodename if !$node; | |
255 | return "snapshot-working/$node/qemu-server/$vmid.conf"; | |
256 | } | |
257 | ||
258 | sub load_config { | |
ffda963f | 259 | my ($class, $vmid, $node) = @_; |
a157d0fd | 260 | |
ffda963f | 261 | my $filename = $class->cfs_config_path($vmid, $node); |
a157d0fd FG |
262 | |
263 | my $raw = PVE::Tools::file_get_contents($filename); | |
264 | ||
265 | my $conf = PVE::QemuServer::parse_vm_config($filename, $raw); | |
266 | return $conf; | |
267 | } | |
268 | ||
269 | sub write_config { | |
ffda963f | 270 | my ($class, $vmid, $conf) = @_; |
a157d0fd | 271 | |
ffda963f | 272 | my $filename = $class->cfs_config_path($vmid); |
a157d0fd FG |
273 | |
274 | if ($conf->{snapshots}) { | |
275 | foreach my $snapname (keys %{$conf->{snapshots}}) { | |
276 | $conf->{snapshots}->{$snapname}->{snaptime} = "1234567890" | |
277 | if $conf->{snapshots}->{$snapname}->{snaptime}; | |
278 | } | |
279 | } | |
280 | ||
281 | my $raw = PVE::QemuServer::write_vm_config($filename, $conf); | |
282 | ||
283 | PVE::Tools::file_set_contents($filename, $raw); | |
284 | } | |
285 | ||
286 | sub has_feature { | |
b2c9558d | 287 | my ($class, $feature, $conf, $storecfg, $snapname, $running, $backup_only) = @_; |
a157d0fd FG |
288 | return $snapshot_possible; |
289 | } | |
290 | ||
b2c9558d FG |
291 | sub __snapshot_save_vmstate { |
292 | my ($class, $vmid, $conf, $snapname, $storecfg) = @_; | |
a157d0fd FG |
293 | die "save_vmstate failed\n" |
294 | if !$save_vmstate_works; | |
295 | ||
296 | my $snap = $conf->{snapshots}->{$snapname}; | |
297 | $snap->{vmstate} = "somestorage:state-volume"; | |
c6737ef1 | 298 | $snap->{runningmachine} = "somemachine" |
a157d0fd | 299 | } |
babf613a SR |
300 | |
301 | sub assert_config_exists_on_node { | |
302 | my ($vmid, $node) = @_; | |
bf116bdf | 303 | return -f cfs_config_path("PVE::QemuConfig", $vmid, $node); |
babf613a | 304 | } |
b2c9558d FG |
305 | # END mocked PVE::QemuConfig methods |
306 | ||
babf613a | 307 | # BEGIN mocked PVE::QemuServer::Helpers methods |
b2c9558d | 308 | |
babf613a | 309 | sub vm_running_locally { |
b2c9558d FG |
310 | return $running; |
311 | } | |
312 | ||
babf613a SR |
313 | # END mocked PVE::QemuServer::Helpers methods |
314 | ||
0a13e08e | 315 | # BEGIN mocked PVE::QemuServer::Monitor methods |
a157d0fd | 316 | |
0a13e08e | 317 | sub qmp_cmd { |
8b26544e | 318 | my ($vmid, $cmd) = @_; |
a157d0fd FG |
319 | |
320 | my $exec = $cmd->{execute}; | |
321 | if ($exec eq "delete-drive-snapshot") { | |
322 | return; | |
323 | } | |
324 | if ($exec eq "guest-ping") { | |
325 | die "guest-ping disabled\n" | |
326 | if !$vm_mon->{guest_ping}; | |
327 | return; | |
328 | } | |
329 | if ($exec eq "guest-fsfreeze-freeze" || $exec eq "guest-fsfreeze-thaw") { | |
330 | die "freeze disabled\n" | |
331 | if !$freeze_possible; | |
332 | return; | |
333 | } | |
334 | if ($exec eq "savevm-start") { | |
335 | die "savevm-start disabled\n" | |
336 | if !$vm_mon->{savevm_start}; | |
337 | return; | |
338 | } | |
339 | if ($exec eq "savevm-end") { | |
340 | die "savevm-end disabled\n" | |
341 | if !$vm_mon->{savevm_end}; | |
342 | return; | |
343 | } | |
344 | if ($exec eq "query-savevm") { | |
345 | return { "status" => "completed" }; | |
346 | } | |
347 | die "unexpected vm_qmp_command!\n"; | |
348 | } | |
349 | ||
0a13e08e SR |
350 | # END mocked PVE::QemuServer::Monitor methods |
351 | ||
352 | # BEGIN redefine PVE::QemuServer methods | |
353 | ||
354 | sub do_snapshots_with_qemu { | |
355 | return 0; | |
356 | } | |
357 | ||
a157d0fd | 358 | sub vm_start { |
0c498cca | 359 | my ($storecfg, $vmid, $params, $migrate_opts) = @_; |
a157d0fd FG |
360 | |
361 | die "Storage config not mocked! aborting\n" | |
362 | if defined($storecfg); | |
363 | ||
364 | die "statefile and forcemachine must be both defined or undefined! aborting\n" | |
0c498cca | 365 | if defined($params->{statefile}) xor defined($params->{forcemachine}); |
a157d0fd FG |
366 | |
367 | return; | |
368 | } | |
369 | ||
370 | sub vm_stop { | |
371 | my ($storecfg, $vmid, $skiplock, $nocheck, $timeout, $shutdown, $force, $keepActive, $migratedfrom) = @_; | |
372 | ||
373 | $running = 0 | |
374 | if $stop_possible; | |
375 | ||
376 | return; | |
377 | } | |
378 | ||
379 | # END redefine PVE::QemuServer methods | |
380 | ||
381 | PVE::Tools::run_command("rm -rf snapshot-working"); | |
382 | PVE::Tools::run_command("cp -a snapshot-input snapshot-working"); | |
383 | ||
f7d1505b | 384 | my $qemu_helpers_module = Test::MockModule->new('PVE::QemuServer::Helpers'); |
babf613a SR |
385 | $qemu_helpers_module->mock('vm_running_locally', \&vm_running_locally); |
386 | ||
f7d1505b | 387 | my $qemu_monitor_module = Test::MockModule->new('PVE::QemuServer::Monitor'); |
0a13e08e SR |
388 | $qemu_monitor_module->mock('qmp_cmd', \&qmp_cmd); |
389 | ||
f7d1505b | 390 | my $qemu_config_module = Test::MockModule->new('PVE::QemuConfig'); |
ffda963f FG |
391 | $qemu_config_module->mock('config_file_lock', \&config_file_lock); |
392 | $qemu_config_module->mock('cfs_config_path', \&cfs_config_path); | |
393 | $qemu_config_module->mock('load_config', \&load_config); | |
394 | $qemu_config_module->mock('write_config', \&write_config); | |
b2c9558d FG |
395 | $qemu_config_module->mock('has_feature', \&has_feature); |
396 | $qemu_config_module->mock('__snapshot_save_vmstate', \&__snapshot_save_vmstate); | |
babf613a | 397 | $qemu_config_module->mock('assert_config_exists_on_node', \&assert_config_exists_on_node); |
ffda963f | 398 | |
d79a3e54 | 399 | # ignore existing replication config |
f7d1505b | 400 | my $repl_config_module = Test::MockModule->new('PVE::ReplicationConfig'); |
9bfca2ca | 401 | $repl_config_module->mock('new' => sub { return bless {}, "PVE::ReplicationConfig" }); |
d1c1af4b | 402 | $repl_config_module->mock('check_for_existing_jobs' => sub { return }); |
d79a3e54 | 403 | |
f7d1505b | 404 | my $storage_module = Test::MockModule->new('PVE::Storage'); |
d1c1af4b | 405 | $storage_module->mock('config', sub { return; }); |
b4dab550 FG |
406 | $storage_module->mock('path', sub { return "/some/store/statefile/path"; }); |
407 | $storage_module->mock('activate_volumes', \&mocked_activate_volumes); | |
408 | $storage_module->mock('deactivate_volumes', \&mocked_deactivate_volumes); | |
409 | $storage_module->mock('vdisk_free', \&mocked_vdisk_free); | |
410 | $storage_module->mock('volume_snapshot', \&mocked_volume_snapshot); | |
411 | $storage_module->mock('volume_snapshot_delete', \&mocked_volume_snapshot_delete); | |
412 | $storage_module->mock('volume_snapshot_rollback', \&mocked_volume_snapshot_rollback); | |
413 | $storage_module->mock('volume_rollback_is_possible', \&mocked_volume_rollback_is_possible); | |
414 | ||
a157d0fd FG |
415 | $running = 1; |
416 | $freeze_possible = 1; | |
417 | $save_vmstate_works = 1; | |
418 | ||
419 | printf("\n"); | |
420 | printf("Running prepare tests\n"); | |
421 | printf("\n"); | |
422 | $nodename = "prepare"; | |
423 | ||
424 | printf("\n"); | |
425 | printf("Setting has_feature to return true\n"); | |
426 | printf("\n"); | |
427 | $snapshot_possible = 1; | |
428 | ||
429 | printf("Successful snapshot_prepare with no existing snapshots\n"); | |
430 | testcase_prepare("101", "test", 0, "test comment", ''); | |
431 | ||
432 | printf("Successful snapshot_prepare with no existing snapshots, including vmstate\n"); | |
433 | testcase_prepare("102", "test", 1, "test comment", ''); | |
434 | ||
435 | printf("Successful snapshot_prepare with one existing snapshot\n"); | |
436 | testcase_prepare("103", "test2", 0, "test comment", ""); | |
437 | ||
438 | printf("Successful snapshot_prepare with one existing snapshot, including vmstate\n"); | |
439 | testcase_prepare("104", "test2", 1, "test comment", ""); | |
440 | ||
441 | printf("Expected error for snapshot_prepare on locked container\n"); | |
442 | testcase_prepare("200", "test", 0, "test comment", "VM is locked (snapshot)\n"); | |
443 | ||
444 | printf("Expected error for snapshot_prepare with duplicate snapshot name\n"); | |
445 | testcase_prepare("201", "test", 0, "test comment", "snapshot name 'test' already used\n"); | |
446 | ||
447 | $save_vmstate_works = 0; | |
448 | ||
449 | printf("Expected error for snapshot_prepare with failing save_vmstate\n"); | |
450 | testcase_prepare("202", "test", 1, "test comment", "save_vmstate failed\n"); | |
451 | ||
452 | $save_vmstate_works = 1; | |
453 | ||
454 | printf("\n"); | |
455 | printf("Setting has_feature to return false\n"); | |
456 | printf("\n"); | |
457 | $snapshot_possible = 0; | |
458 | ||
459 | printf("Expected error for snapshot_prepare if snapshots not possible\n"); | |
460 | testcase_prepare("300", "test", 0, "test comment", "snapshot feature is not available\n"); | |
461 | ||
462 | printf("\n"); | |
463 | printf("Running commit tests\n"); | |
464 | printf("\n"); | |
465 | $nodename = "commit"; | |
466 | ||
467 | printf("\n"); | |
468 | printf("Setting has_feature to return true\n"); | |
469 | printf("\n"); | |
470 | $snapshot_possible = 1; | |
471 | ||
472 | printf("Successful snapshot_commit with one prepared snapshot\n"); | |
473 | testcase_commit("101", "test", ""); | |
474 | ||
475 | printf("Successful snapshot_commit with one committed and one prepared snapshot\n"); | |
476 | testcase_commit("102", "test2", ""); | |
477 | ||
478 | printf("Expected error for snapshot_commit with no snapshot lock\n"); | |
479 | testcase_commit("201", "test", "missing snapshot lock\n"); | |
480 | ||
481 | printf("Expected error for snapshot_commit with invalid snapshot name\n"); | |
482 | testcase_commit("202", "test", "snapshot 'test' does not exist\n"); | |
483 | ||
484 | printf("Expected error for snapshot_commit with invalid snapshot state\n"); | |
485 | testcase_commit("203", "test", "wrong snapshot state\n"); | |
486 | ||
487 | $vol_snapshot_possible->{"local:snapshotable-disk-1"} = 1; | |
488 | $vol_snapshot_possible->{"local:snapshotable-disk-2"} = 1; | |
489 | $vol_snapshot_possible->{"local:snapshotable-disk-3"} = 1; | |
490 | $vol_snapshot_delete_possible->{"local:snapshotable-disk-1"} = 1; | |
491 | $vol_snapshot_delete_possible->{"local:snapshotable-disk-3"} = 1; | |
492 | $vol_snapshot_rollback_enabled->{"local:snapshotable-disk-1"} = 1; | |
493 | $vol_snapshot_rollback_enabled->{"local:snapshotable-disk-2"} = 1; | |
494 | $vol_snapshot_rollback_enabled->{"local:snapshotable-disk-3"} = 1; | |
495 | $vol_snapshot_rollback_possible->{"local:snapshotable-disk-1"} = 1; | |
496 | $vol_snapshot_rollback_possible->{"local:snapshotable-disk-2"} = 1; | |
497 | $vol_snapshot_rollback_possible->{"local:snapshotable-disk-3"} = 1; | |
498 | $vol_snapshot_rollback_possible->{"local:snapshotable-disk-4"} = 1; | |
499 | $vm_mon->{guest_ping} = 1; | |
500 | $vm_mon->{savevm_start} = 1; | |
501 | $vm_mon->{savevm_end} = 1; | |
502 | ||
503 | # possible, but fails | |
504 | $vol_snapshot_rollback_possible->{"local:snapshotable-disk-4"} = 1; | |
505 | ||
a157d0fd FG |
506 | |
507 | #printf("\n"); | |
508 | #printf("Setting up Mocking for PVE::Tools\n"); | |
f7d1505b | 509 | #my $tools_module = Test::MockModule->new('PVE::Tools'); |
a157d0fd FG |
510 | #$tools_module->mock('run_command' => \&mocked_run_command); |
511 | #printf("\trun_command() mocked\n"); | |
512 | # | |
513 | $nodename = "create"; | |
514 | printf("\n"); | |
515 | printf("Running create tests\n"); | |
516 | printf("\n"); | |
517 | ||
518 | printf("Successful snapshot_create with no existing snapshots\n"); | |
519 | testcase_create("101", "test", 0, "test comment", "", { "local:snapshotable-disk-1" => "test" }); | |
520 | ||
521 | printf("Successful snapshot_create with no existing snapshots, including vmstate\n"); | |
522 | testcase_create("102", "test", 1, "test comment", "", { "local:snapshotable-disk-1" => "test" }); | |
523 | ||
524 | printf("Successful snapshot_create with one existing snapshots\n"); | |
525 | testcase_create("103", "test2", 0, "test comment", "", { "local:snapshotable-disk-1" => "test2" }); | |
526 | ||
527 | printf("Successful snapshot_create with one existing snapshots, including vmstate\n"); | |
528 | testcase_create("104", "test2", 1, "test comment", "", { "local:snapshotable-disk-1" => "test2" }); | |
529 | ||
530 | printf("Successful snapshot_create with multiple mps\n"); | |
531 | testcase_create("105", "test", 0, "test comment", "", { "local:snapshotable-disk-1" => "test", "local:snapshotable-disk-2" => "test", "local:snapshotable-disk-3" => "test" }); | |
532 | ||
533 | $freeze_possible = 0; | |
534 | printf("Successful snapshot_create with no existing snapshots and broken freeze\n"); | |
535 | testcase_create("106", "test", 1, "test comment", "", { "local:snapshotable-disk-1" => "test" }); | |
536 | $freeze_possible = 1; | |
537 | ||
538 | printf("Expected error for snapshot_create when volume snapshot is not possible\n"); | |
b2c9558d | 539 | testcase_create("201", "test", 0, "test comment", "volume snapshot disabled\n\n"); |
a157d0fd FG |
540 | |
541 | printf("Expected error for snapshot_create when volume snapshot is not possible for one drive\n"); | |
b2c9558d | 542 | testcase_create("202", "test", 0, "test comment", "volume snapshot disabled\n\n", { "local:snapshotable-disk-1" => "test" }, { "local:snapshotable-disk-1" => "test" }); |
a157d0fd FG |
543 | |
544 | $vm_mon->{savevm_start} = 0; | |
545 | printf("Expected error for snapshot_create when Qemu mon command 'savevm-start' fails\n"); | |
b2c9558d | 546 | testcase_create("203", "test", 0, "test comment", "savevm-start disabled\n\n"); |
a157d0fd FG |
547 | $vm_mon->{savevm_start} = 1; |
548 | ||
28831e10 DC |
549 | printf("Successful snapshot_create with no existing snapshots but set machine type\n"); |
550 | testcase_create("301", "test", 1, "test comment", "", { "local:snapshotable-disk-1" => "test" }); | |
a157d0fd FG |
551 | |
552 | $nodename = "delete"; | |
553 | printf("\n"); | |
554 | printf("Running delete tests\n"); | |
555 | printf("\n"); | |
556 | ||
557 | printf("Successful snapshot_delete of only existing snapshot\n"); | |
558 | testcase_delete("101", "test", 0, "", { "local:snapshotable-disk-1" => "test" }); | |
559 | ||
560 | printf("Successful snapshot_delete of leaf snapshot\n"); | |
561 | testcase_delete("102", "test2", 0, "", { "local:snapshotable-disk-1" => "test2" }); | |
562 | ||
563 | printf("Successful snapshot_delete of root snapshot\n"); | |
564 | testcase_delete("103", "test", 0, "", { "local:snapshotable-disk-1" => "test" }); | |
565 | ||
566 | printf("Successful snapshot_delete of intermediate snapshot\n"); | |
567 | testcase_delete("104", "test2", 0, "", { "local:snapshotable-disk-1" => "test2" }); | |
568 | ||
569 | printf("Successful snapshot_delete with broken volume_snapshot_delete and force=1\n"); | |
570 | testcase_delete("105", "test", 1, ""); | |
571 | ||
572 | printf("Successful snapshot_delete with mp broken volume_snapshot_delete and force=1\n"); | |
573 | testcase_delete("106", "test", 1, "", { "local:snapshotable-disk-1" => "test", "local:snapshotable-disk-3" => "test" }); | |
574 | ||
575 | printf("Expected error when snapshot_delete fails with broken volume_snapshot_delete and force=0\n"); | |
576 | testcase_delete("201", "test", 0, "volume snapshot delete disabled\n"); | |
577 | ||
578 | printf("Expected error when snapshot_delete fails with broken mp volume_snapshot_delete and force=0\n"); | |
579 | testcase_delete("202", "test", 0, "volume snapshot delete disabled\n", { "local:snapshotable-disk-1" => "test" }); | |
580 | ||
581 | printf("Expected error for snapshot_delete with locked config\n"); | |
582 | testcase_delete("203", "test", 0, "VM is locked (backup)\n"); | |
583 | ||
584 | $nodename = "rollback"; | |
585 | printf("\n"); | |
586 | printf("Running rollback tests\n"); | |
587 | printf("\n"); | |
588 | ||
589 | $stop_possible = 1; | |
590 | ||
591 | printf("Successful snapshot_rollback to only existing snapshot\n"); | |
592 | testcase_rollback("101", "test", "", { "local:snapshotable-disk-1" => "test" }); | |
593 | ||
594 | printf("Successful snapshot_rollback to leaf snapshot\n"); | |
595 | testcase_rollback("102", "test2", "", { "local:snapshotable-disk-1" => "test2" }); | |
596 | ||
597 | printf("Successful snapshot_rollback to root snapshot\n"); | |
598 | testcase_rollback("103", "test", "", { "local:snapshotable-disk-1" => "test" }); | |
599 | ||
600 | printf("Successful snapshot_rollback to intermediate snapshot\n"); | |
601 | testcase_rollback("104", "test2", "", { "local:snapshotable-disk-1" => "test2" }); | |
602 | ||
603 | printf("Successful snapshot_rollback with multiple mp\n"); | |
604 | testcase_rollback("105", "test", "", { "local:snapshotable-disk-1" => "test", "local:snapshotable-disk-2" => "test", "local:snapshotable-disk-3" => "test" }); | |
605 | ||
606 | printf("Successful snapshot_rollback to only existing snapshot, with saved vmstate and machine config\n"); | |
607 | testcase_rollback("106", "test", "", { "local:snapshotable-disk-1" => "test" }); | |
608 | ||
609 | printf("Expected error for snapshot_rollback with non-existing snapshot\n"); | |
610 | testcase_rollback("201", "test2", "snapshot 'test2' does not exist\n"); | |
611 | ||
612 | printf("Expected error for snapshot_rollback if volume rollback not possible\n"); | |
613 | testcase_rollback("202", "test", "volume_rollback_is_possible failed\n"); | |
614 | ||
615 | printf("Expected error for snapshot_rollback with incomplete snapshot\n"); | |
616 | testcase_rollback("203", "test", "unable to rollback to incomplete snapshot (snapstate = delete)\n"); | |
617 | ||
618 | printf("Expected error for snapshot_rollback with lock\n"); | |
619 | testcase_rollback("204", "test", "VM is locked (backup)\n"); | |
620 | ||
621 | $stop_possible = 0; | |
622 | ||
623 | printf("Expected error for snapshot_rollback with unkillable container\n"); | |
624 | testcase_rollback("205", "test", "unable to rollback vm 205: vm is running\n"); | |
625 | ||
626 | $stop_possible = 1; | |
627 | ||
628 | printf("Expected error for snapshot_rollback with mp rollback_is_possible failure\n"); | |
629 | testcase_rollback("206", "test", "volume_rollback_is_possible failed\n"); | |
630 | ||
631 | printf("Expected error for snapshot_rollback with mp rollback failure (results in inconsistent state)\n"); | |
632 | testcase_rollback("207", "test", "volume snapshot rollback disabled\n", { "local:snapshotable-disk-1" => "test", "local:snapshotable-disk-2" => "test" }); | |
633 | ||
28831e10 DC |
634 | printf("Successful snapshot_rollback with saved vmstate and machine config only in snapshot\n"); |
635 | testcase_rollback("301", "test", "", { "local:snapshotable-disk-1" => "test" }); | |
636 | ||
637 | printf("Successful snapshot_rollback with saved vmstate and machine config and runningmachine \n"); | |
638 | testcase_rollback("302", "test", "", { "local:snapshotable-disk-1" => "test" }); | |
639 | ||
a157d0fd | 640 | done_testing(); |
2bf945fc TL |
641 | |
642 | 1; |