stats_manager: add title banner pretty-printing

If a title is supplied it now gets printed centralized on top of the
summary when calling SummaryToString.

BRANCH=None
BUG=chromium:760267
TEST=unit tests are passing

Change-Id: I7c59896ebac82d2ee7b632fd18350b9b4fff2c24
Signed-off-by: Ruben Rodriguez Buchillon <coconutruben@chromium.org>
Reviewed-on: https://chromium-review.googlesource.com/1140030
Reviewed-by: Mengqi Guo <mqg@chromium.org>

1 files changed, 9 insertions, 0 deletions

diff --git a/extra/usb_power/stats_manager_unittest.py b/extra/usb_power/stats_manager_unittest.py
index a669ab8e8f..e2dda06f91 100644
--- a/extra/usb_power/stats_manager_unittest.py
+++ b/extra/usb_power/stats_manager_unittest.py
@@ -14,6 +14,7 @@ import unittest
 
 import stats_manager
 
+
 class TestStatsManager(unittest.TestCase):
   """Test to verify StatsManager methods work as expected.
 
@@ -152,6 +153,14 @@ class TestStatsManager(unittest.TestCase):
     for fname in files:
       self.assertTrue(os.path.basename(fname).startswith(identifier))
 
+  def test_SummaryToStringTitle(self):
+    """Title shows up in SummaryToString if title specified."""
+    title = 'titulo'
+    data = stats_manager.StatsManager(title=title)
+    self._populate_dummy_stats()
+    summary_str = data.SummaryToString()
+    self.assertIn(title, summary_str)
+
   def test_SummaryToStringHideDomains(self):
     """Keys indicated in hide_domains are not printed in the summary."""
     data = stats_manager.StatsManager(hide_domains=['A-domain'])