CRAYSAT-1923: Update sat bootsys to support CFS v2 or v3

CHANGELOG.md

@@ -25,6 +25,13 @@ All notable changes to this project will be documented in this file.
 The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/),
 and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
 
+## [3.33.2] - 2024-12-09
+
+### Added
+- Added support for using either the CFS v2 or v3 API in `sat bootsys`,
+  depending on the value of the `--cfs-version` command-line option or the
+  `cfs.api_version` config-file option.
+
 ## [3.33.1] - 2024-12-02
 
 ### Fixed

docs/man/sat-bootsys.8.rst

@@ -159,6 +159,9 @@ These options apply to both the ``shutdown`` and ``boot`` actions.
 **--bos-version BOS_VERSION**
         The version of the BOS API to use when launching BOS sessions.
 
+**--cfs-version CFS_VERSION**
+        The version of the CFS API to use when launching CFS sessions.
+
 **--cle-bos-template** *CLE_BOS_TEMPLATE*
         The name of the BOS session template for shutdown or boot of
         COS (formerly known as CLE) compute nodes. If not specified, no

sat/cli/bootsys/parser.py

@@ -285,6 +285,12 @@ def add_bootsys_subparser(subparsers):
         help='The version of the BOS API to use for BOS operations',
     )
 
+    bootsys_parser.add_argument(
+        '--cfs-version',
+        choices=['v2', 'v3'],
+        help='The version of the CFS API to use for CFS operations',
+    )
+
     actions_subparsers = bootsys_parser.add_subparsers(
         metavar='action', dest='action', help='The action to perform.'
     )

sat/cli/bootsys/service_activity.py

@@ -383,9 +383,9 @@ def get_active_sessions(self):
         Raises:
             ServiceCheckError: if unable to get the active CFS sessions.
         """
-        cfs_client = CFSClientBase.get_cfs_client(SATSession(), 'v2')
+        cfs_client = CFSClientBase.get_cfs_client(SATSession(), get_config_value('cfs.api_version'))
         try:
-            sessions = cfs_client.get('sessions').json()
+            sessions = cfs_client.get_sessions()
         except (APIError, ValueError) as err:
             raise self.get_err(str(err))
 

tests/cli/bootsys/test_service_activity.py

@@ -490,26 +490,15 @@ def setUp(self):
 
         # If not None, causes CFSClient().get() to raise this
         self.cfs_err = None
-        # If not None, causes CFSClient().get().json() to raise this
-        self.json_err = None
-
-        def mock_cfs_get(*args):
-            if self.cfs_err:
-                raise self.cfs_err
 
-            def json():
-                if len(args) != 1 or args[0] != 'sessions':
-                    self.fail('Unexpected get to CFSClient with args: {}'.format(args))
-                elif self.json_err:
-                    raise self.json_err
-                else:
-                    return self.cfs_sessions
-            return Mock(json=json)
+        def mock_get_sessions(*args, **kwargs):
+            for session in self.cfs_sessions:
+                yield session
 
         patch('sat.cli.bootsys.service_activity.SATSession').start()
         self.mock_cfs_client = patch(
             'sat.cli.bootsys.service_activity.CFSClientBase.get_cfs_client').start()
-        self.mock_cfs_client.return_value.get = mock_cfs_get
+        self.mock_cfs_client.return_value.get_sessions = mock_get_sessions
 
     def tearDown(self):
         """Stop mocks at the end of each unit test."""
@@ -548,20 +537,15 @@ def test_get_active_sessions_two_active(self):
 
     def test_get_active_sessions_api_err(self):
         """Test get_active_sessions with an API error."""
+        self.mock_cfs_client.return_value.get_sessions = Mock(
+            side_effect=APIError('service unavailable')
+        )
         api_err_msg = 'service unavailable'
         self.cfs_err = APIError(api_err_msg)
         err_regex = '^Unable to get active CFS sessions: {}'.format(api_err_msg)
         with self.assertRaisesRegex(ServiceCheckError, err_regex):
             self.cfs_checker.get_active_sessions()
 
-    def test_get_active_sessions_value_err(self):
-        """Test get_active_sessions"""
-        val_err_msg = 'invalid json'
-        self.json_err = ValueError(val_err_msg)
-        err_regex = '^Unable to get active CFS sessions: {}'.format(val_err_msg)
-        with self.assertRaisesRegex(ServiceCheckError, err_regex):
-            self.cfs_checker.get_active_sessions()
-
 
 class TestFirmwareActivityChecker(ExtendedTestCase):
     """Test the FirmwareActivityChecker class."""