Add unit tests for check and assert shutdown

cpanfile

@@ -59,6 +59,7 @@ on 'test' => sub {
   requires 'Test::Pod';
   requires 'Test::Simple';
   requires 'Test::Warnings';
+  requires 'Test::Exception';
   requires 'Socket::MsgHdr';
 };
 

t/03-testapi.t

@@ -9,6 +9,7 @@ use Test::Fatal;
 use Test::Mock::Time;
 use Test::Warnings;
 use File::Temp;
+use Test::Exception;
 
 BEGIN {
     unshift @INC, '..';
@@ -291,6 +292,30 @@ subtest 'set console tty and other args' => sub {
     is($console->{args}->{foo}, 'bar');
 };
 
+subtest 'check_assert_shutdown' => sub {
+    # Test cases, when shutdown is finished before timeout is hit
+    $mod->mock(
+        read_json => sub {
+            return {ret => 1};
+        });
+    ok(check_shutdown, 'check_shutdown should return "true" if shutdown finished before timeout is hit');
+    is(assert_shutdown, undef, 'assert_shutdown should return "undef" if shutdown finished before timeout is hit');
+    $mod->mock(
+        read_json => sub {
+            return {ret => -1};
+        });
+    ok(check_shutdown, 'check_shutdown should return "true" if backend does not implement is_shutdown');
+    is(assert_shutdown, undef, 'assert_shutdown should return "undef" if backend does not implement is_shutdown');
+    # Test cases, when shutdown is not finished if timeout is hit
+    $mod->mock(
+        read_json => sub {
+            return {ret => 0};
+        });
+    is(check_shutdown, 0, 'check_shutdown should return "false" if timeout is hit');
+    throws_ok { assert_shutdown } qr/Machine didn't shut down!/, 'assert_shutdown should throw exception if timeout is hit';
+
+};
+
 done_testing;
 
 # vim: set sw=4 et:

t/data/tests/tests/shutdown.pm

@@ -21,7 +21,12 @@ sub run {
     wait_idle 1;
     type_string "sudo su\n";
     type_string "poweroff\n";
-    assert_shutdown;
+    if (get_var('INTEGRATION_TESTS')) {
+        assert_shutdown(90);
+    }
+    else {
+        assert_shutdown;
+    }
 }
 
 sub test_flags {