From: Arunpandian J <arun.j@samsung.com>
Date: Tue, 25 Apr 2023 04:43:43 +0000 (+0530)
Subject: tests: Add tests for controller reset
X-Git-Tag: v2.5~145
X-Git-Url: https://www.infradead.org/git/?a=commitdiff_plain;h=c37eb6dc8bae237a3abf0ae1b710ebb7a1d1cac7;p=users%2Fsagi%2Fnvme-cli.git

tests: Add tests for controller reset

Signed-off-by: Arunpandian J <arun.j@samsung.com>
---

diff --git a/tests/meson.build b/tests/meson.build
index 630f35db..90f1a688 100644
--- a/tests/meson.build
+++ b/tests/meson.build
@@ -28,6 +28,7 @@ tests = [
   'nvme_verify_test.py',
   'nvme_lba_status_log_test.py',
   'nvme_get_lba_status_test.py',
+  'nvme_ctrl_reset_test.py',
 ]
 
 runtests = find_program('nose2', required : false)
diff --git a/tests/nvme_ctrl_reset_test.py b/tests/nvme_ctrl_reset_test.py
new file mode 100644
index 00000000..b8b3c3b9
--- /dev/null
+++ b/tests/nvme_ctrl_reset_test.py
@@ -0,0 +1,48 @@
+# SPDX-License-Identifier: GPL-2.0-or-later
+#
+# This file is part of nvme-cli
+#
+# Copyright (c) 2023 Samsung Electronics Co., Ltd. All Rights Reserved.
+#
+# Author: Arunpandian J <arun.j@samsung.com>
+
+"""
+NVMe controller reset Testcase:-
+
+    1. Execute nvme controller reset.
+
+"""
+
+from nvme_test import TestNVMe
+
+
+class TestNVMeCtrlReset(TestNVMe):
+
+    """
+    Represents NVMe Controller reset testcase.
+        - Attributes:
+              - test_log_dir :  directory for logs, temp files.
+    """
+
+    def setUp(self):
+        """ Pre Section for TestNVMeCtrlReset """
+        super().setUp()
+        self.setup_log_dir(self.__class__.__name__)
+
+    def tearDown(self):
+        """ Post Section for TestNVMeCtrlReset """
+        super().tearDown()
+
+    def ctrl_reset(self):
+        """ Wrapper for nvme controller reset
+            - Args:
+                - None
+            - Returns:
+                - return code for nvme controller reset.
+        """
+        ctrl_reset_cmd = "nvme reset " + self.ctrl
+        return self.exec_cmd(ctrl_reset_cmd)
+
+    def test_ctrl_reset(self):
+        """ Testcase main """
+        self.assertEqual(self.ctrl_reset(), 0)