Merge pull request #389 from grafuls/development

fix: non blocking or failing on --host-list
redhat-performance · Oct 31, 2023 · 3a7cb0a · 3a7cb0a
2 parents 5fd9977 + d055a83
commit 3a7cb0a
Show file tree

Hide file tree

Showing 5 changed files with 22 additions and 4 deletions.
diff --git a/...b/ISSUE_TEMPLATE/pull_request_template.md → ...REQUEST_TEMPLATE/pull_request_template.md b/...b/ISSUE_TEMPLATE/pull_request_template.md → ...REQUEST_TEMPLATE/pull_request_template.md
diff --git a/.github/workflows/codecov.yml b/.github/workflows/codecov.yml
@@ -24,6 +24,7 @@ jobs:
     - name: Upload coverage to Codecov
       uses: codecov/codecov-action@v2
       with:
+        token: ${{ secrets.CODECOV_TOKEN }}
         files: ./coverage.xml
         env_vars: PYTHON
         flags: unittests

diff --git a/src/badfish/main.py b/src/badfish/main.py
@@ -649,7 +649,11 @@ async def get_power_consumed_watts(self):
             data = json.loads(raw.strip())
         except ValueError:
             raise BadfishException("Power value outside operating range.")
-        self.logger.info(f"Current watts consumed: {data['PowerControl'][0]['PowerConsumedWatts']}")
+        try:
+            cwc = data['PowerControl'][0]['PowerConsumedWatts']
+        except IndexError:
+            cwc = "N/A. Try to `--racreset`."
+        self.logger.info(f"Current watts consumed: {cwc}")
         return
 
     async def change_boot(self, host_type, interfaces_path, pxe=False):

diff --git a/tests/config.py b/tests/config.py
@@ -170,7 +170,9 @@ def render_device_dict(index, device):
 )
 # test_power_consumed_watts
 POWER_CONSUMED_RESP = '{"PowerControl":[{"PowerConsumedWatts":"69"}]}'
+NO_POWER = '{"PowerControl":[]}'
 RESPONSE_POWER_CONSUMED_OK = '- INFO     - Current watts consumed: 69\n'
+RESPONSE_NO_POWER_CONSUMED = '- INFO     - Current watts consumed: N/A. Try to `--racreset`.\n'
 RESPONSE_POWER_CONSUMED_404 = '- ERROR    - Operation not supported by vendor.\n'
 RESPONSE_POWER_CONSUMED_VAL_ERR = '- ERROR    - Power value outside operating range.\n'
 # test_reset_%s

diff --git a/tests/test_power_consumed_watts.py b/tests/test_power_consumed_watts.py
@@ -1,6 +1,6 @@
 from asynctest import patch
 from tests.config import (
-    INIT_RESP, POWER_CONSUMED_RESP, RESPONSE_POWER_CONSUMED_OK, RESPONSE_POWER_CONSUMED_404,
+    INIT_RESP, NO_POWER, POWER_CONSUMED_RESP, RESPONSE_NO_POWER_CONSUMED, RESPONSE_POWER_CONSUMED_OK, RESPONSE_POWER_CONSUMED_404,
      RESPONSE_POWER_CONSUMED_VAL_ERR,
 
 )
@@ -25,19 +25,30 @@ def test_power_consumed(self, mock_get, mock_post, mock_delete):
     @patch("aiohttp.ClientSession.get")
     def test_power_consumed_404(self, mock_get, mock_post, mock_delete):
         responses = INIT_RESP + [POWER_CONSUMED_RESP]
-        self.set_mock_response(mock_get,[200,200,200,200,200,404], responses)
+        self.set_mock_response(mock_get, [200,200,200,200,200,404], responses)
         self.set_mock_response(mock_post, 200, "OK", True)
         self.set_mock_response(mock_delete, 200, "OK")
         _, err = self.badfish_call()
         assert err == RESPONSE_POWER_CONSUMED_404
 
+    @patch("aiohttp.ClientSession.delete")
+    @patch("aiohttp.ClientSession.post")
+    @patch("aiohttp.ClientSession.get")
+    def test_no_power(self, mock_get, mock_post, mock_delete):
+        responses = INIT_RESP + [NO_POWER]
+        self.set_mock_response(mock_get, 200, responses)
+        self.set_mock_response(mock_post, 200, "OK", True)
+        self.set_mock_response(mock_delete, 200, "OK")
+        _, err = self.badfish_call()
+        assert err == RESPONSE_NO_POWER_CONSUMED
+
     @patch("aiohttp.ClientSession.delete")
     @patch("aiohttp.ClientSession.post")
     @patch("aiohttp.ClientSession.get")
     def test_power_consumed_value_error(self, mock_get, mock_post, mock_delete):
         responses_add = [""]
         responses = INIT_RESP + responses_add
-        self.set_mock_response(mock_get,200, responses)
+        self.set_mock_response(mock_get, 200, responses)
         self.set_mock_response(mock_post, 200, "OK", True)
         self.set_mock_response(mock_delete, 200, "OK")
         _, err = self.badfish_call()