Added get_total_items to SoftLayerListResult to make figuring out how many results were returned a bit easier

Author: allmightyspiff

.secrets.baseline

@@ -3,7 +3,7 @@
     "files": "^.secrets.baseline$",
     "lines": null
   },
-  "generated_at": "2024-04-25T01:18:20Z",
+  "generated_at": "2024-10-04T22:18:14Z",
   "plugins_used": [
     {
       "name": "AWSKeyDetector"
@@ -720,7 +720,7 @@
         "hashed_secret": "9878e362285eb314cfdbaa8ee8c300c285856810",
         "is_secret": false,
         "is_verified": false,
-        "line_number": 323,
+        "line_number": 324,
         "type": "Secret Keyword",
         "verified_result": null
       }

SoftLayer/transports/fixture.py

@@ -8,6 +8,8 @@
 
 import importlib
 
+from .transport import SoftLayerListResult
+
 
 class FixtureTransport(object):
     """Implements a transport which returns fixtures."""
@@ -21,7 +23,10 @@ def __call__(self, call):
             message = f'{call.service} fixture is not implemented'
             raise NotImplementedError(message) from ex
         try:
-            return getattr(module, call.method)
+            result = getattr(module, call.method)
+            if isinstance(result, list):
+                return SoftLayerListResult(result, len(result))
+            return result
         except AttributeError as ex:
             message = f'{call.service}::{call.method} fixture is not implemented'
             raise NotImplementedError(message) from ex

SoftLayer/transports/rest.py

@@ -138,8 +138,7 @@ def __call__(self, request):
             request.result = result
 
             if isinstance(result, list):
-                return SoftLayerListResult(
-                    result, int(resp.headers.get('softlayer-total-items', 0)))
+                return SoftLayerListResult(result, int(resp.headers.get('softlayer-total-items', 0)))
             else:
                 return result
         except requests.HTTPError as ex:

SoftLayer/transports/transport.py

@@ -121,6 +121,10 @@ def __init__(self, items=None, total_count=0):
         self.total_count = total_count
         super().__init__(items)
 
+    def get_total_items(self):
+        """A simple getter to totalCount, but its called getTotalItems since that is the header returned"""
+        return self.total_count
+
 
 def _proxies_dict(proxy):
     """Makes a proxy dict appropriate to pass to requests."""

SoftLayer/transports/xmlrpc.py

@@ -100,8 +100,7 @@ def __call__(self, request):
             resp.raise_for_status()
             result = xmlrpc.client.loads(resp.content)[0][0]
             if isinstance(result, list):
-                return SoftLayerListResult(
-                    result, int(resp.headers.get('softlayer-total-items', 0)))
+                return SoftLayerListResult(result, int(resp.headers.get('softlayer-total-items', 0)))
             else:
                 return result
         except xmlrpc.client.Fault as ex:

tests/transports/rest_tests.py

@@ -37,6 +37,7 @@ def test_basic(self, request):
         self.assertEqual(resp, [])
         self.assertIsInstance(resp, transports.SoftLayerListResult)
         self.assertEqual(resp.total_count, 10)
+        self.assertEqual(resp.get_total_items(), 10)
         request.assert_called_with(
             'GET', 'http://something9999999999999999999999.com/SoftLayer_Service/Resource.json',
             headers=mock.ANY,

tests/transports/transport_tests.py

@@ -20,6 +20,13 @@ def test_basic(self):
         resp = self.transport(req)
         self.assertEqual(resp['accountId'], 1234)
 
+    def test_total_items(self):
+        req = transports.Request()
+        req.service = 'SoftLayer_Account'
+        req.method = 'getHardware'
+        resp = self.transport(req)
+        self.assertEqual(resp.get_total_items(), 4)
+
     def test_no_module(self):
         req = transports.Request()
         req.service = 'Doesnt_Exist'

tests/transports/xmlrpc_tests.py

@@ -409,6 +409,7 @@ def test_nonascii_characters(self, request):
         self.assertEqual(resp, [])
         self.assertIsInstance(resp, transports.SoftLayerListResult)
         self.assertEqual(resp.total_count, 10)
+        self.assertEqual(resp.get_total_items(), 10)
 
 
 @mock.patch('SoftLayer.transports.xmlrpc.requests.Session.request')