Add SOL console logging to gpu stress test

Resolves  openbmc/openbmc-test-automation#1112
Change-Id: Ibd3175083a9c3ee7a37ff9a376b288d50c2ba405
Signed-off-by: Steven Sombar <ssombar@us.ibm.com>
diff --git a/systest/gpu_stress_test.robot b/systest/gpu_stress_test.robot
index 1ca0316..9a40951 100644
--- a/systest/gpu_stress_test.robot
+++ b/systest/gpu_stress_test.robot
@@ -3,8 +3,10 @@
 
 Resource         ../syslib/utils_os.robot
 
-Test Setup      Pre Test Case Execution
-Test Teardown   Post Test Case Execution
+Suite Setup      Run Keyword  Start SOL Console Logging
+Test Setup       Pre Test Case Execution
+Test Teardown    Test Teardown Execution
+Suite Teardown   Suite Teardown Execution
 
 *** Variables ****
 
@@ -60,11 +62,10 @@
     ...  AND  Sleep  ${HTX_INTERVAL}
 
 
-Post Test Case Execution
+Test Teardown Execution
     [Documentation]  Do the post test teardown.
-    # 1. Shut down HTX exerciser if test Failed.
-    # 2. Capture FFDC on test failure.
-    # 3. Close all open SSH connections.
+    #  Shut down HTX exerciser if test Failed.
+    #  Collect NVIDIA log.
 
     # Keep HTX running if user set HTX_KEEP_RUNNING to 1.
     Run Keyword If  '${TEST_STATUS}' == 'FAIL' and ${HTX_KEEP_RUNNING} == ${0}
@@ -73,5 +74,16 @@
     # Collect nvidia-smi output data on exit.
     Collect NVIDIA Log File  end
 
+
+Suite Teardown Execution
+    [Documentation]  Do the final teardown and cleanup.
+    #  Stop SOL Console Logging.
+    #  Collect FFDC if Test Case Fail.
+    #  Close Connections.
+
+    ${keyword_buf}=  Catenate  Stop SOL Console Logging
+    ...  \ targ_file_path=${EXECDIR}${/}logs${/}SOL.log
+    Run Key  ${keyword_buf}
+
     FFDC On Test Case Fail
     Close All Connections