From e8d1045ff03f7e5f82fce44f62589c9196cb4287 Mon Sep 17 00:00:00 2001
From: Lukas Ruzicka <lruzicka@redhat.com>
Date: Tue, 10 Jul 2018 16:46:06 +0200
Subject: [PATCH] Handle restart.

---
 tests/desktop_upgrade_graphical.pm | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/tests/desktop_upgrade_graphical.pm b/tests/desktop_upgrade_graphical.pm
index 23b74f6f..1cb2678f 100644
--- a/tests/desktop_upgrade_graphical.pm
+++ b/tests/desktop_upgrade_graphical.pm
@@ -65,8 +65,9 @@ sub run {
         if match_has_tag('desktop_package_tool_password'){
             type_very_safely(get_var('USER_PASSWORD','weakpassword'));
             assert_and_click('desktop_package_tool_authenticate','',30);
-        } 
+        }
 
+        assert_and_click('desktop_package_tool_restart','',30); 
 
         boot_to_login_screen;
     }