Add show reboot-history

doc/Command-Reference.md

@@ -541,6 +541,26 @@ This command displays the cause of the previous reboot
   User issued reboot command [User: admin, Time: Mon Mar 25 01:02:03 UTC 2019]
   ```
 
+**show reboot-cause history**
+
+This command displays the history of the previous reboots up to 10 entry
+
+- Usage:
+  ```
+  show reboot-cause history
+  ```
+
+- Example:
+  ```
+  admin@sonic:~$ show reboot-cause history
+  Name                 Cause        Time                          User    Comment
+  -------------------  -----------  ----------------------------  ------  ---------
+  2020_10_09_02_33_06  reboot       Fri Oct  9 02:29:44 UTC 2020  admin
+  2020_10_09_01_56_59  reboot       Fri Oct  9 01:53:49 UTC 2020  admin
+  2020_10_09_02_00_53  fast-reboot  Fri Oct  9 01:58:04 UTC 2020  admin
+  2020_10_09_04_53_58  warm-reboot  Fri Oct  9 04:51:47 UTC 2020  admin
+  ```
+
 **show uptime**
 
 This command displays the current system uptime

show/main.py

@@ -1800,23 +1800,56 @@ def mmu():
 
 
 #
-# 'reboot-cause' command ("show reboot-cause")
-#
-@cli.command('reboot-cause')
-def reboot_cause():
-    """Show cause of most recent reboot"""
-    PREVIOUS_REBOOT_CAUSE_FILE = "/host/reboot-cause/previous-reboot-cause.txt"
-
-    # At boot time, PREVIOUS_REBOOT_CAUSE_FILE is generated based on
-    # the contents of the 'reboot cause' file as it was left when the device
-    # went down for reboot. This file should always be created at boot,
-    # but check first just in case it's not present.
-    if not os.path.isfile(PREVIOUS_REBOOT_CAUSE_FILE):
-        click.echo("Unable to determine cause of previous reboot\n")
-    else:
-        cmd = "cat {}".format(PREVIOUS_REBOOT_CAUSE_FILE)
-        proc = subprocess.Popen(cmd, stdout=subprocess.PIPE, shell=True)
-        click.echo(proc.stdout.read())
+# 'reboot-cause' group ("show reboot-cause")
+#
+@cli.group('reboot-cause', invoke_without_command=True)
+@click.pass_context
+def reboot_cause(ctx):
+    if ctx.invoked_subcommand is None:
+        """Show cause of reboot"""
+        REBOOT_CAUSE_DIR = "/host/reboot-cause/"
+        PREVIOUS_REBOOT_CAUSE_FILE = os.path.join(REBOOT_CAUSE_DIR, "previous-reboot-cause.json")
+        USER_ISSUED_REBOOT_CAUSE_REGEX ="User issued \'{}\' command [User: {}, Time: {}]"
+        REBOOT_CAUSE_UNKNOWN = "Unknown"
+
+        last_reboot_cause = REBOOT_CAUSE_UNKNOWN
+        # Read the last previous reboot cause
+        if os.path.exists(PREVIOUS_REBOOT_CAUSE_FILE):
+            with open(PREVIOUS_REBOOT_CAUSE_FILE, "r") as last_cause_file:
+                data = json.load(last_cause_file)
+                if data['user']:
+                    last_reboot_cause = USER_ISSUED_REBOOT_CAUSE_REGEX.format(data['cause'], data['user'], data['time'])
+                else:
+                    last_reboot_cause = "{}".format(data['cause'])
+
+        click.echo(last_reboot_cause)
+
+# 'history' subcommand ("show reboot-cause history")
+@reboot_cause.command()
+def history():
+    """Show history of reboot-cause"""
+    REBOOT_CAUSE_TABLE_NAME = "REBOOT_CAUSE"
+    TABLE_NAME_SEPARATOR = '|'
+    db = SonicV2Connector(host='127.0.0.1')
+    db.connect(db.STATE_DB, False)   # Make one attempt only
+    prefix = REBOOT_CAUSE_TABLE_NAME + TABLE_NAME_SEPARATOR
+    _hash = '{}{}'.format(prefix, '*')
+    table_keys = db.keys(db.STATE_DB, _hash)
+    table_keys.sort(reverse=True)
+
+    table = []
+    for tk in table_keys:
+        entry = db.get_all(db.STATE_DB, tk)
+        r = []
+        r.append(tk.replace(prefix,""))
+        r.append(entry['cause'] if 'cause' in entry else "")
+        r.append(entry['time'] if 'time' in entry else "")
+        r.append(entry['user'] if 'user' in entry else "")
+        r.append(entry['comment'] if 'comment' in entry else "")
+        table.append(r)
+
+    header = ['Name', 'Cause', 'Time', 'User', 'Comment']
+    click.echo(tabulate(table, header))
 
 
 #

tests/mock_tables/state_db.json

@@ -203,5 +203,17 @@
         "speed_target": "50",
         "led_status": "green",
         "timestamp": "20200813 01:32:30"
+    },
+    "REBOOT_CAUSE|2020_10_09_04_53_58": {
+        "cause": "warm-reboot",
+        "time": "Fri Oct  9 04:51:47 UTC 2020",
+        "user": "admin",
+        "comment": ""
+    },
+    "REBOOT_CAUSE|2020_10_09_02_33_06": {
+        "cause": "reboot",
+        "time": "Fri Oct  9 02:29:44 UTC 2020",
+        "user": "admin",
+        "comment": ""
     }
 }

tests/show_reboot_cause_test.py

@@ -0,0 +1,87 @@
+import os
+import sys
+import textwrap
+
+import mock
+
+from click.testing import CliRunner
+
+test_path = os.path.dirname(os.path.abspath(__file__))
+modules_path = os.path.dirname(test_path)
+sys.path.insert(0, modules_path)
+
+import show.main as show
+
+module_ = "builtins"
+module_ = module_ if module_ in sys.modules else '__builtin__'
+
+reboot_cause_json_file="/host/reboot-cause/previous-reboot-cause.json"
+"""
+    Note: The following 'show reboot-cause' commands simply call other SONiC
+    CLI utilities, so the unit tests for the other utilities are expected
+    to cover testing their functionality:
+
+        show reboot-cause
+        show reboot-cause history
+"""
+
+class TestShowRebootCause(object):
+    @classmethod
+    def setup_class(cls):
+        print("SETUP")
+        os.environ["UTILITIES_UNIT_TESTING"] = "1"
+
+    # Test 'show reboot-cause' without previous-reboot-cause.json 
+    def test_reboot_cause_no_history_file(self):
+        expected_output = "Unknown\n"
+        runner = CliRunner()
+        result = runner.invoke(show.cli.commands["reboot-cause"], [])
+        assert result.output == expected_output
+
+    # Test 'show reboot-cause' with user issued reboot
+    def test_reboot_cause_user(self):
+        expected_output = "User issued reboot command [User: admin, Time: Thu Oct 22 03:11:08 UTC 2020]\n"
+        reboot_cause_user_json = """\
+            {"comment": "", "gen_time": "2020_10_22_03_14_07", "cause": "reboot", "user": "admin", "time": "Thu Oct 22 03:11:08 UTC 2020"}
+            """
+        runner = CliRunner()
+        with mock.patch("os.path.isfile") as mock_isfile:
+            mock_isfile.return_value = True
+            open_mocked = mock.mock_open(read_data=textwrap.dedent(reboot_cause_user_json))
+            with mock.patch("{}.open".format(module_), open_mocked):
+                result = runner.invoke(show.cli.commands["reboot-cause"], [])
+                assert result == expected_output
+                open_mocked.assert_called_once_with(reboot_cause_json_file)
+
+    # Test 'show reboot-cause' with non-user issue reboot (hardware reboot-cause or unknown reboot-cause)
+    def test_reboot_cause_non_user(self):
+        expected_output = "Watchdog\n"
+        reboot_cause_watchdog_json = """\
+            {"comment": "", "gen_time": "2020_10_22_03_15_08", "cause": "Watchdog", "user": "", "time": ""}
+            """
+        runner = CliRunner()
+        with mock.patch("os.path.isfile") as mock_isfile:
+            mock_isfile.return_value = True
+            open_mocked = mock.mock_open(read_data=reboot_cause_watchdog_json)
+            with mock.patch("{}.open".format(module_), open_mocked):
+                result = runner.invoke(show.cli.commands["reboot-cause"], [])
+                assert result == expected_output
+                open_mocked.assert_called_once_with(reboot_cause_json_file)
+
+    # Test 'show reboot-cause history'
+    def test_reboot_cause_history(self):
+        expected_output = """\
+Name                 Cause        Time                          User    Comment
+-------------------  -----------  ----------------------------  ------  ---------
+2020_10_09_04_53_58  warm-reboot  Fri Oct  9 04:51:47 UTC 2020  admin
+2020_10_09_02_33_06  reboot       Fri Oct  9 02:29:44 UTC 2020  admin
+"""
+        runner = CliRunner()
+        result = runner.invoke(show.cli.commands["reboot-cause"].commands["history"], [])
+        assert result.output == expected_output
+
+    @classmethod
+    def teardown_class(cls):
+        print("TEARDOWN")
+        os.environ["PATH"] = os.pathsep.join(os.environ["PATH"].split(os.pathsep)[:-1])
+        os.environ["UTILITIES_UNIT_TESTING"] = "0"