Await for kernel restart to complete

Author: krassowski

atest/_resources/Keywords.resource

@@ -500,6 +500,8 @@ Restart Kernel
     Lab Command    Restart Kernel…
     Wait For Dialog
     Accept Default Dialog Option
+    Wait Until Page Contains Element    css:.jp-Notebook-ExecutionIndicator[data-status="restarting"]
+    Wait Until Page Contains Element    css:.jp-Notebook-ExecutionIndicator[data-status="idle"]
 
 Expand Menu Entry
     [Arguments]    ${label}

atest/suites/04_Interface/DiagnosticsPanel.robot

@@ -35,9 +35,7 @@ Diagnostics Panel Works After Rename
 Diagnostics Panel Works After Kernel Restart
     [Documentation]    Test for #475 bug
     Close Diagnostics Panel
-    Lab Command    Restart Kernel…
-    Wait For Dialog
-    Accept Default Dialog Option
+    Restart Kernel
     Wait Until Page Contains Diagnostic    [title*="${DIAGNOSTIC}"]    timeout=20s
     Open Diagnostics Panel
     Wait Until Keyword Succeeds    10 x    1s    Should Have Expected Rows Count    ${EXPECTED_COUNT}