Adding total elapsed time to flow/util/genElapsedTime.py (#1663)

* Adding total elapsed time

Signed-off-by: Augusto Berndt <[email protected]>

* Fix test with total elapsed

---------

Signed-off-by: Augusto Berndt <[email protected]>

Author: gudeh

flow/test/test_genElapsedTime.py

@@ -31,7 +31,7 @@ def test_elapsed_time(self, mock_stdout):
         with patch.object(sys, 'argv', sys.argv):
             module = importlib.import_module(self.module_name)
         # check if output is correct
-        expected_output = self.tmp_dir.name + "\n1_test                          5400\n"
+        expected_output = self.tmp_dir.name + "\n1_test                          5400\nTotal                           5400\n"
         self.assertEqual(mock_stdout.getvalue(), expected_output)
 
     @patch("sys.stdout", new_callable=StringIO)
@@ -61,7 +61,7 @@ def test_elapsed_time_longer_duration(self, mock_stdout):
             module = importlib.import_module(self.module_name)
             importlib.reload(module)
         # check if output is correct
-        expected_output = self.tmp_dir.name + "\n1_test                           744\n"
+        expected_output = self.tmp_dir.name + "\n1_test                           744\nTotal                            744\n"
         self.assertEqual(mock_stdout.getvalue(), expected_output)
 
     def test_missing_arg(self):

flow/util/genElapsedTime.py

@@ -27,6 +27,7 @@
 
 def print_log_dir_times(logdir):
     first = True
+    totalElapsed = 0
     print(logdir)
 
     # Loop on all log files in the directory
@@ -65,6 +66,10 @@ def print_log_dir_times(logdir):
                 print("%-25s %10s" % ("Log", "Elapsed seconds"))
                 first = False
             print('%-25s %10s' % (os.path.splitext(os.path.basename(str(f)))[0], elapsedTime))
+        totalElapsed += elapsedTime
+        
+    if totalElapsed != 0:
+        print("%-25s %10s" % ( "Total", totalElapsed ))
 
 for log_dir in args.logDir:
     print_log_dir_times(log_dir)