sfq 0.0.33__tar.gz → 0.0.35__tar.gz

{sfq-0.0.33 → sfq-0.0.35}/.github/workflows/publish.yml

@@ -82,6 +82,9 @@ jobs:
           echo "Updating src/sfq/__init__.py __version__ to $VERSION"
           sed -i -E "s/(self.__version__ = \")[0-9]+\.[0-9]+\.[0-9]+(\")/\1$VERSION\2/" src/sfq/__init__.py
           sed -i -E "s/(__version__ = \")[0-9]+\.[0-9]+\.[0-9]+(\")/\1$VERSION\2/" src/sfq/__init__.py
+          
+          echo "Updating src/sfq/http_client.py user_agent to $VERSION"
+          sed -i -E "s/(user_agent: str = \"sfq\/)[0-9]+\.[0-9]+\.[0-9]+(\")/\1$VERSION\2/" src/sfq/http_client.py
 
       - name: Run tests
         run: pytest --verbose --strict-config

{sfq-0.0.33 → sfq-0.0.35}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sfq
-Version: 0.0.33
+Version: 0.0.35
 Summary: Python wrapper for the Salesforce's Query API.
 Author-email: David Moruzzi <sfq.pypi@dmoruzi.com>
 Keywords: salesforce,salesforce query

{sfq-0.0.33 → sfq-0.0.35}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "sfq"
-version = "0.0.33"
+version = "0.0.35"
 description = "Python wrapper for the Salesforce's Query API."
 readme = "README.md"
 authors = [{ name = "David Moruzzi", email = "sfq.pypi@dmoruzi.com" }]

{sfq-0.0.33 → sfq-0.0.35}/src/sfq/__init__.py

@@ -25,6 +25,7 @@ from .http_client import HTTPClient
 from .query import QueryClient
 from .soap import SOAPClient
 from .utils import get_logger
+from .debug_cleanup import DebugCleanup
 
 # Define public API for documentation tools
 __all__ = [
@@ -42,7 +43,7 @@ __all__ = [
     "__version__",
 ]
 
-__version__ = "0.0.33"
+__version__ = "0.0.35"
 """
 ### `__version__`
 
@@ -66,7 +67,7 @@ class SFAuth:
         access_token: Optional[str] = None,
         token_expiration_time: Optional[float] = None,
         token_lifetime: int = 15 * 60,
-        user_agent: str = "sfq/0.0.33",
+        user_agent: str = "sfq/0.0.35",
         sforce_client: str = "_auto",
         proxy: str = "_auto",
     ) -> None:
@@ -127,8 +128,11 @@ class SFAuth:
             api_version=api_version,
         )
 
+        # Initialize the DebugCleanup
+        self._debug_cleanup = DebugCleanup(sf_auth=self)
+
         # Store version information
-        self.__version__ = "0.0.33"
+        self.__version__ = "0.0.35"
         """
         ### `__version__`
         
@@ -530,25 +534,20 @@ class SFAuth:
             sobject, insert_list, batch_size, max_workers, api_type
         )
 
-    def _debug_cleanup_apex_logs(self):
-        """
-        This function performs cleanup operations for Apex debug logs.
-        """
-        apex_logs = self.query("SELECT Id FROM ApexLog ORDER BY LogLength DESC")
-        if apex_logs and apex_logs.get("records"):
-            log_ids = [log["Id"] for log in apex_logs["records"]]
-            if log_ids:
-                delete_response = self.cdelete(log_ids)
-                logger.debug("Deleted Apex logs: %s", delete_response)
-        else:
-            logger.debug("No Apex logs found to delete.")
-
-    def debug_cleanup(self, apex_logs: bool = True) -> None:
+    def debug_cleanup(
+        self,
+        apex_logs: bool = True,
+        expired_apex_flags: bool = True,
+        all_apex_flags: bool = False,
+    ) -> None:
         """
         Perform cleanup operations for Apex debug logs.
         """
-        if apex_logs:
-            self._debug_cleanup_apex_logs()
+        self._debug_cleanup.debug_cleanup(
+            apex_logs=apex_logs,
+            expired_apex_flags=expired_apex_flags,
+            all_apex_flags=all_apex_flags,
+        )
 
     def open_frontdoor(self) -> None:
         """

{sfq-0.0.33 → sfq-0.0.35}/src/sfq/crud.py

@@ -37,23 +37,17 @@ class CRUDClient:
         self.soap_client = soap_client
         self.api_version = api_version
 
-    def create(
+    def _soap_batch_operation(
         self,
         sobject: str,
-        insert_list: List[Dict[str, Any]],
+        data_list,
+        method: str,
         batch_size: int = 200,
         max_workers: int = None,
         api_type: Literal["enterprise", "tooling"] = "enterprise",
     ) -> Optional[Dict[str, Any]]:
         """
-        Execute the Insert API to insert multiple records via SOAP calls.
-
-        :param sobject: The name of the sObject to insert into.
-        :param insert_list: A list of dictionaries, each representing a record to insert.
-        :param batch_size: The number of records to insert in each batch (default is 200).
-        :param max_workers: The maximum number of threads to spawn for concurrent execution.
-        :param api_type: API type to use ('enterprise' or 'tooling').
-        :return: JSON response from the insert request or None on failure.
+        Internal helper for batch SOAP operations (create/delete).
         """
         endpoint = "/services/Soap/"
         if api_type == "enterprise":
@@ -67,21 +61,20 @@ class CRUDClient:
             )
             return None
 
-        # Handle API versioning in the endpoint
         endpoint = endpoint.replace("/v", "/")
 
-        if isinstance(insert_list, dict):
-            insert_list = [insert_list]
+        if isinstance(data_list, dict) and method == "create":
+            data_list = [data_list]
+        if isinstance(data_list, str) and method == "delete":
+            data_list = [data_list]
 
         chunks = [
-            insert_list[i : i + batch_size]
-            for i in range(0, len(insert_list), batch_size)
+            data_list[i : i + batch_size]
+            for i in range(0, len(data_list), batch_size)
         ]
 
-        def insert_chunk(chunk: List[Dict[str, Any]]) -> Optional[Dict[str, Any]]:
-            """Insert a chunk of records using SOAP API."""
+        def process_chunk(chunk):
             try:
-                # Get the access token for the SOAP header
                 access_token = self.http_client.auth_manager.access_token
                 if not access_token:
                     logger.error("No access token available for SOAP request")
@@ -89,7 +82,7 @@ class CRUDClient:
 
                 header = self.soap_client.generate_soap_header(access_token)
                 body = self.soap_client.generate_soap_body(
-                    sobject=sobject, method="create", data=chunk
+                    sobject=sobject, method=method, data=chunk
                 )
                 envelope = self.soap_client.generate_soap_envelope(
                     header=header, body=body, api_type=api_type
@@ -99,8 +92,8 @@ class CRUDClient:
                 soap_headers["Content-Type"] = "text/xml; charset=UTF-8"
                 soap_headers["SOAPAction"] = '""'
 
-                logger.trace("SOAP request envelope: %s", envelope)
-                logger.trace("SOAP request headers: %s", soap_headers)
+                logger.trace(f"SOAP {method} request envelope: %s", envelope)
+                logger.trace(f"SOAP {method} request headers: %s", soap_headers)
 
                 status_code, resp_data = self.http_client.send_request(
                     method="POST",
@@ -109,25 +102,27 @@ class CRUDClient:
                     body=envelope,
                 )
 
+                logger.trace(f"SOAP {method} response status: {status_code}")
+                logger.trace(f"SOAP {method} raw response: {resp_data}")
+
                 if status_code == 200:
-                    logger.debug("Insert API request successful.")
-                    logger.trace("Insert API response: %s", resp_data)
+                    logger.debug(f"{method.capitalize()} API request successful.")
+                    logger.trace(f"{method.capitalize()} API response: %s", resp_data)
                     result = self.soap_client.extract_soap_result_fields(resp_data)
                     if result:
                         return result
                     logger.error("Failed to extract fields from SOAP response.")
                 else:
-                    logger.error("Insert API request failed: %s", status_code)
+                    logger.error(f"{method.capitalize()} API request failed: %s", status_code)
                     logger.debug("Response body: %s", resp_data)
                     return None
-
             except Exception as e:
-                logger.exception("Exception during insert chunk: %s", e)
+                logger.exception(f"Exception during {method} chunk: %s", e)
                 return None
 
         results = []
         with ThreadPoolExecutor(max_workers=max_workers) as executor:
-            futures = [executor.submit(insert_chunk, chunk) for chunk in chunks]
+            futures = [executor.submit(process_chunk, chunk) for chunk in chunks]
             for future in as_completed(futures):
                 result = future.result()
                 if result:
@@ -142,6 +137,79 @@ class CRUDClient:
 
         return combined_response or None
 
+    def create(
+        self,
+        sobject: str,
+        insert_list: List[Dict[str, Any]],
+        batch_size: int = 200,
+        max_workers: int = None,
+        api_type: Literal["enterprise", "tooling"] = "enterprise",
+    ) -> Optional[Dict[str, Any]]:
+        """
+        Execute the Insert API to insert multiple records via SOAP calls.
+        """
+        return self._soap_batch_operation(
+            sobject=sobject,
+            data_list=insert_list,
+            method="create",
+            batch_size=batch_size,
+            max_workers=max_workers,
+            api_type=api_type,
+        )
+    
+    def update(
+        self,
+        sobject: str,
+        update_list: List[Dict[str, Any]],
+        batch_size: int = 200,
+        max_workers: int = None,
+        api_type: Literal["enterprise", "tooling"] = "enterprise",
+    ) -> Optional[Dict[str, Any]]:
+        """
+        Execute the Update API to update multiple records via SOAP calls.
+        :param sobject: The name of the sObject to update.
+        :param update_list: A list of dictionaries, each representing a record to update (must include Id).
+        :param batch_size: The number of records to update in each batch (default is 200).
+        :param max_workers: The maximum number of threads to spawn for concurrent execution.
+        :param api_type: API type to use ('enterprise' or 'tooling').
+        :return: JSON response from the update request or None on failure.
+        """
+        return self._soap_batch_operation(
+            sobject=sobject,
+            data_list=update_list,
+            method="update",
+            batch_size=batch_size,
+            max_workers=max_workers,
+            api_type=api_type,
+        )
+    
+    def delete(
+        self,
+        sobject: str,
+        id_list: List[str],
+        batch_size: int = 200,
+        max_workers: int = None,
+        api_type: Literal["enterprise", "tooling"] = "enterprise",
+    ) -> Optional[Dict[str, Any]]:
+        """
+        Execute the Delete API to remove multiple records via SOAP calls.
+        :param sobject: The name of the sObject to delete from.
+        :param id_list: A list of record IDs to delete (strings, not dicts).
+        :param batch_size: The number of records to delete in each batch (default is 200).
+        :param max_workers: The maximum number of threads to spawn for concurrent execution.
+        :param api_type: API type to use ('enterprise' or 'tooling').
+        :return: JSON response from the delete request or None on failure.
+        """
+        # Pass list of IDs directly for SOAP delete
+        return self._soap_batch_operation(
+            sobject=sobject,
+            data_list=id_list,
+            method="delete",
+            batch_size=batch_size,
+            max_workers=max_workers,
+            api_type=api_type,
+        )
+
     def cupdate(
         self, update_dict: Dict[str, Any], batch_size: int = 25, max_workers: int = None
     ) -> Optional[Dict[str, Any]]:

sfq-0.0.35/src/sfq/debug_cleanup.py

@@ -0,0 +1,71 @@
+"""
+Debug cleanup module for Salesforce-related debug artifacts.
+"""
+
+from datetime import datetime, timezone
+import logging
+from typing import Dict, Any, Optional
+
+logger = logging.getLogger("sfq")
+
+
+class DebugCleanup:
+    """
+    Class handling debug artifact cleanup operations in Salesforce.
+    """
+
+    def __init__(self, sf_auth):
+        """
+        Initialize the DebugCleanup with an SFAuth instance.
+
+        :param sf_auth: An authenticated SFAuth instance
+        """
+        self._sf_auth = sf_auth
+
+    def _debug_cleanup_apex_logs(self) -> None:
+        """
+        This function performs cleanup operations for Apex debug logs.
+        """
+        apex_logs = self._sf_auth.query("SELECT Id FROM ApexLog ORDER BY LogLength DESC")
+        if apex_logs and apex_logs.get("records"):
+            log_ids = [log["Id"] for log in apex_logs["records"]]
+            if log_ids:
+                delete_response = self._sf_auth.cdelete(log_ids)
+                logger.debug("Deleted Apex logs: %s", delete_response)
+        else:
+            logger.debug("No Apex logs found to delete.")
+
+    def _debug_cleanup_trace_flags(self, expired_only: bool) -> None:
+        """
+        This function performs cleanup operations for Trace Flags.
+        """
+        if expired_only:
+            now_iso_format = datetime.now(timezone.utc).strftime('%Y-%m-%dT%H:%M:%SZ')
+            query = f"SELECT Id, ExpirationDate FROM TraceFlag WHERE ExpirationDate < {now_iso_format}"
+        else:
+            query = "SELECT Id, ExpirationDate FROM TraceFlag"
+
+        traceflags = self._sf_auth.tooling_query(query)
+        if traceflags['totalSize'] == 0:
+            logger.debug("No expired Trace Flag configurations to delete.")
+            return
+
+        trace_flag_ids = [tf["Id"] for tf in traceflags["records"]]
+        logger.debug("Deleting Trace Flags: %s", trace_flag_ids)
+        results = self._sf_auth._crud_client.delete("TraceFlag", trace_flag_ids, api_type="tooling")
+        # results = self._sf_auth.delete("TraceFlag", trace_flag_ids)
+        logger.debug("Deleted Trace Flags: %s", results)
+
+    def debug_cleanup(self, apex_logs: bool = True, expired_apex_flags: bool = True, all_apex_flags: bool = False) -> None:
+        """
+        Perform cleanup operations for Apex debug logs.
+
+        :param apex_logs: Whether to clean up Apex logs (default: True)
+        """
+        if apex_logs:
+            self._debug_cleanup_apex_logs()
+
+        if all_apex_flags:
+            self._debug_cleanup_trace_flags(expired_only=False)
+        elif expired_apex_flags:
+            self._debug_cleanup_trace_flags(expired_only=True)

{sfq-0.0.33 → sfq-0.0.35}/src/sfq/http_client.py

@@ -28,7 +28,7 @@ class HTTPClient:
     def __init__(
         self,
         auth_manager: AuthManager,
-        user_agent: str = "sfq/0.0.33",
+        user_agent: str = "sfq/0.0.35",
         sforce_client: str = "_auto",
         high_api_usage_threshold: int = 80,
     ) -> None:

{sfq-0.0.33 → sfq-0.0.35}/src/sfq/soap.py

@@ -100,13 +100,18 @@ class SOAPClient:
         else:
             records = data
 
-        sobjects = "".join(
-            f'<sObjects xsi:type="{sobject}">'
-            + "".join(f"<{k}>{v}</{k}>" for k, v in record.items())
-            + "</sObjects>"
-            for record in records
-        )
-        return f"<soapenv:Body><{method}>{sobjects}</{method}></soapenv:Body>"
+        if method == "delete":
+            # For delete, records is a list of IDs (strings)
+            id_tags = "".join(f"<ID>{id}</ID>" for id in records)
+            return f"<soapenv:Body><delete>{id_tags}</delete></soapenv:Body>"
+        else:
+            sobjects = "".join(
+                f'<sObjects xsi:type="{sobject}">'
+                + "".join(f"<{k}>{v}</{k}>" for k, v in record.items())
+                + "</sObjects>"
+                for record in records
+            )
+            return f"<soapenv:Body><{method}>{sobjects}</{method}></soapenv:Body>"
 
     def extract_soap_result_fields(
         self, xml_string: str

{sfq-0.0.33 → sfq-0.0.35}/tests/test_cdelete.py

@@ -21,7 +21,7 @@ def sf_instance():
     sf = SFAuth(
         instance_url=os.getenv("SF_INSTANCE_URL"),
         client_id=os.getenv("SF_CLIENT_ID"),
-        client_secret=os.getenv("SF_CLIENT_SECRET"),
+        client_secret=os.getenv("SF_CLIENT_SECRET").strip(),
         refresh_token=os.getenv("SF_REFRESH_TOKEN"),
     )
     return sf

{sfq-0.0.33 → sfq-0.0.35}/tests/test_compatibility.py

@@ -541,9 +541,6 @@ class TestPrivateMethodCompatibility:
         assert hasattr(mock_sf_auth, "debug_cleanup")
         assert callable(mock_sf_auth.debug_cleanup)
 
-        assert hasattr(mock_sf_auth, "_debug_cleanup_apex_logs")
-        assert callable(mock_sf_auth._debug_cleanup_apex_logs)
-
     def test_frontdoor_method_available(self, mock_sf_auth):
         """Test that open_frontdoor method is available."""
         assert hasattr(mock_sf_auth, "open_frontdoor")

{sfq-0.0.33 → sfq-0.0.35}/tests/test_cquery.py

@@ -21,7 +21,7 @@ def sf_instance():
     sf = SFAuth(
         instance_url=os.getenv("SF_INSTANCE_URL"),
         client_id=os.getenv("SF_CLIENT_ID"),
-        client_secret=os.getenv("SF_CLIENT_SECRET"),
+        client_secret=os.getenv("SF_CLIENT_SECRET").strip(),
         refresh_token=os.getenv("SF_REFRESH_TOKEN"),
     )
     return sf

{sfq-0.0.33 → sfq-0.0.35}/tests/test_create.py

@@ -21,7 +21,7 @@ def sf_instance():
     sf = SFAuth(
         instance_url=os.getenv("SF_INSTANCE_URL"),
         client_id=os.getenv("SF_CLIENT_ID"),
-        client_secret=os.getenv("SF_CLIENT_SECRET"),
+        client_secret=os.getenv("SF_CLIENT_SECRET").strip(),
         refresh_token=os.getenv("SF_REFRESH_TOKEN"),
     )
     return sf

{sfq-0.0.33 → sfq-0.0.35}/tests/test_crud_e2e.py

@@ -34,7 +34,7 @@ def auth_manager():
     return AuthManager(
         instance_url=os.getenv("SF_INSTANCE_URL"),
         client_id=os.getenv("SF_CLIENT_ID"),
-        client_secret=os.getenv("SF_CLIENT_SECRET"),
+        client_secret=os.getenv("SF_CLIENT_SECRET").strip(),
         refresh_token=os.getenv("SF_REFRESH_TOKEN"),
     )
 
@@ -74,7 +74,7 @@ def sf_auth():
     return SFAuth(
         instance_url=os.getenv("SF_INSTANCE_URL"),
         client_id=os.getenv("SF_CLIENT_ID"),
-        client_secret=os.getenv("SF_CLIENT_SECRET"),
+        client_secret=os.getenv("SF_CLIENT_SECRET").strip(),
         refresh_token=os.getenv("SF_REFRESH_TOKEN"),
     )
 

{sfq-0.0.33 → sfq-0.0.35}/tests/test_cupdate.py

@@ -23,7 +23,7 @@ def sf_instance():
     sf = SFAuth(
         instance_url=os.getenv("SF_INSTANCE_URL"),
         client_id=os.getenv("SF_CLIENT_ID"),
-        client_secret=os.getenv("SF_CLIENT_SECRET"),
+        client_secret=os.getenv("SF_CLIENT_SECRET").strip(),
         refresh_token=os.getenv("SF_REFRESH_TOKEN"),
     )
     return sf

sfq-0.0.33/tests/test_debug_cleanup.py → sfq-0.0.35/tests/test_debug_cleanup_e2e.py

@@ -26,12 +26,54 @@ def sf_instance():
     sf = SFAuth(
         instance_url=os.getenv("SF_INSTANCE_URL"),
         client_id=os.getenv("SF_CLIENT_ID"),
-        client_secret=os.getenv("SF_CLIENT_SECRET"),
+        client_secret=os.getenv("SF_CLIENT_SECRET").strip(),
         refresh_token=os.getenv("SF_REFRESH_TOKEN"),
     )
     return sf
 
 
+def test_trace_flag_expired_deletions(sf_instance):
+    """
+    This test case ensures that unused TraceFlag's in the Tooling API are deleted when called.
+    """
+    now_iso_format = datetime.now(timezone.utc).strftime('%Y-%m-%dT%H:%M:%SZ')
+
+    query = f"SELECT Id, ExpirationDate FROM TraceFlag WHERE ExpirationDate < {now_iso_format}"
+
+    traceflags = sf_instance.tooling_query(query)
+    if traceflags['totalSize'] == 0:
+        pytest.skip("No expired Trace Flag configurations to delete.")
+
+    initial_count = len(traceflags["records"])
+    assert initial_count > 0, "Initial count of TraceFlags should be greater than zero."
+
+    results = sf_instance.debug_cleanup(apex_logs=False, expired_apex_flags=True, all_apex_flags=False)
+    assert results is None, "Expected no return value from debug_cleanup."
+
+    traceflags_after = sf_instance.tooling_query(query)
+    assert len(traceflags_after["records"]) == 0, "TraceFlags were not deleted successfully."
+
+
+def test_trace_flag_all_deletions(sf_instance):
+    """
+    This test case ensures that unused TraceFlag's in the Tooling API are deleted when called.
+    """
+    query = f"SELECT Id, ExpirationDate FROM TraceFlag"
+
+    traceflags = sf_instance.tooling_query(query)
+    if traceflags['totalSize'] == 0:
+        pytest.skip("No expired Trace Flag configurations to delete.")
+
+    initial_count = len(traceflags["records"])
+    assert initial_count > 0, "Initial count of TraceFlags should be greater than zero."
+
+    results = sf_instance.debug_cleanup(apex_logs=False, expired_apex_flags=False, all_apex_flags=True)
+    assert results is None, "Expected no return value from debug_cleanup."
+
+
+    traceflags_after = sf_instance.tooling_query(query)
+    assert len(traceflags_after["records"]) == 0, "TraceFlags were not deleted successfully."
+
 def test_debug_cleanup(sf_instance, already_executed: bool = False):
     """
     Test the debug_cleanup method of SFAuth.
@@ -40,7 +82,7 @@ def test_debug_cleanup(sf_instance, already_executed: bool = False):
     # Check if any Apex logs already exist
     apex_logs = sf_instance.query("SELECT Id FROM ApexLog LIMIT 1")
     if apex_logs.get("records"):
-        sf_instance.debug_cleanup(apex_logs=True)
+        sf_instance.debug_cleanup(apex_logs=True, expired_apex_flags=False, all_apex_flags=False)
         apex_logs_after = sf_instance.query("SELECT Id FROM ApexLog LIMIT 1")
         assert len(apex_logs_after.get("records", [])) == 0, (
             "Apex logs were not cleaned up successfully."
@@ -74,13 +116,9 @@ def test_debug_cleanup(sf_instance, already_executed: bool = False):
                 datetime.now(timezone.utc) + timedelta(minutes=5)
             ).isoformat(),
         }
-        resp = sf_instance._create(
+        _ = sf_instance._create(
             sobject="TraceFlag", insert_list=[traceflag_payload], api_type="tooling"
         )
-        with open("debug_payload.json", "w") as f:
-            f.write(json.dumps(resp, indent=2))
-        with open("debug_payload.json", "w") as f:
-            f.write(json.dumps(traceflag_payload, indent=2))
 
         traceflag_query = sf_instance.tooling_query(
             f"SELECT Id FROM TraceFlag WHERE TracedEntityId = '{sf_instance.user_id}' LIMIT 1"
@@ -140,3 +178,4 @@ def test_debug_cleanup(sf_instance, already_executed: bool = False):
 
     sleep(1)  # Race condition mitigation
     return test_debug_cleanup(sf_instance, already_executed=True)
+

sfq-0.0.35/tests/test_debug_cleanup_unit.py

@@ -0,0 +1,57 @@
+import pytest
+from unittest.mock import MagicMock
+from sfq.debug_cleanup import DebugCleanup
+
+class DummySFAuth:
+    def __init__(self):
+        self._crud_client = MagicMock()
+        self.cdelete = MagicMock()
+        self.query = MagicMock()
+        self.tooling_query = MagicMock()
+
+@pytest.fixture
+def dummy_sf_auth():
+    return DummySFAuth()
+
+@pytest.fixture
+def debug_cleanup(dummy_sf_auth):
+    return DebugCleanup(dummy_sf_auth)
+
+def test_debug_cleanup_apex_logs(debug_cleanup, dummy_sf_auth):
+    dummy_sf_auth.query.return_value = {"records": [{"Id": "log1"}, {"Id": "log2"}]}
+    debug_cleanup._debug_cleanup_apex_logs()
+    dummy_sf_auth.cdelete.assert_called_once_with(["log1", "log2"])
+
+def test_debug_cleanup_apex_logs_no_logs(debug_cleanup, dummy_sf_auth):
+    dummy_sf_auth.query.return_value = {"records": []}
+    debug_cleanup._debug_cleanup_apex_logs()
+    dummy_sf_auth.cdelete.assert_not_called()
+
+def test_debug_cleanup_trace_flags_expired(debug_cleanup, dummy_sf_auth):
+    dummy_sf_auth.tooling_query.return_value = {"totalSize": 2, "records": [{"Id": "tf1"}, {"Id": "tf2"}]}
+    debug_cleanup._sf_auth._crud_client.delete.return_value = "deleted"
+    debug_cleanup._debug_cleanup_trace_flags(expired_only=True)
+    dummy_sf_auth._crud_client.delete.assert_called_once_with("TraceFlag", ["tf1", "tf2"], api_type="tooling")
+
+def test_debug_cleanup_trace_flags_none(debug_cleanup, dummy_sf_auth):
+    dummy_sf_auth.tooling_query.return_value = {"totalSize": 0, "records": []}
+    debug_cleanup._debug_cleanup_trace_flags(expired_only=True)
+    dummy_sf_auth._crud_client.delete.assert_not_called()
+
+def test_debug_cleanup_dispatch(debug_cleanup, dummy_sf_auth):
+    # Test all_apex_flags True
+    dummy_sf_auth.tooling_query.return_value = {"totalSize": 1, "records": [{"Id": "tf1"}]}
+    debug_cleanup._sf_auth._crud_client.delete.reset_mock()
+    debug_cleanup.debug_cleanup(apex_logs=False, expired_apex_flags=True, all_apex_flags=True)
+    dummy_sf_auth._crud_client.delete.assert_called_once_with("TraceFlag", ["tf1"], api_type="tooling")
+
+    # Test expired_apex_flags True, all_apex_flags False
+    debug_cleanup._sf_auth._crud_client.delete.reset_mock()
+    debug_cleanup.debug_cleanup(apex_logs=False, expired_apex_flags=True, all_apex_flags=False)
+    dummy_sf_auth._crud_client.delete.assert_called_once_with("TraceFlag", ["tf1"], api_type="tooling")
+
+    # Test apex_logs True
+    dummy_sf_auth.query.return_value = {"records": [{"Id": "log1"}]}
+    debug_cleanup._sf_auth.cdelete.reset_mock()
+    debug_cleanup.debug_cleanup(apex_logs=True, expired_apex_flags=False, all_apex_flags=False)
+    dummy_sf_auth.cdelete.assert_called_once_with(["log1"])

{sfq-0.0.33 → sfq-0.0.35}/tests/test_limits_api.py

@@ -21,7 +21,7 @@ def sf_instance():
     sf = SFAuth(
         instance_url=os.getenv("SF_INSTANCE_URL"),
         client_id=os.getenv("SF_CLIENT_ID"),
-        client_secret=os.getenv("SF_CLIENT_SECRET"),
+        client_secret=os.getenv("SF_CLIENT_SECRET").strip(),
         refresh_token=os.getenv("SF_REFRESH_TOKEN"),
     )
     return sf

{sfq-0.0.33 → sfq-0.0.35}/tests/test_log_trace_redact.py

@@ -23,7 +23,7 @@ def sf_instance():
     sf = SFAuth(
         instance_url=os.getenv("SF_INSTANCE_URL"),
         client_id=os.getenv("SF_CLIENT_ID"),
-        client_secret=os.getenv("SF_CLIENT_SECRET"),
+        client_secret=os.getenv("SF_CLIENT_SECRET").strip(),
         refresh_token=os.getenv("SF_REFRESH_TOKEN"),
     )
     return sf

{sfq-0.0.33 → sfq-0.0.35}/tests/test_open_frontdoor.py

@@ -25,7 +25,7 @@ def sf_instance():
     return SFAuth(
         instance_url=os.getenv("SF_INSTANCE_URL"),
         client_id=os.getenv("SF_CLIENT_ID"),
-        client_secret=os.getenv("SF_CLIENT_SECRET"),
+        client_secret=os.getenv("SF_CLIENT_SECRET").strip(),
         refresh_token=os.getenv("SF_REFRESH_TOKEN"),
     )
 

{sfq-0.0.33 → sfq-0.0.35}/tests/test_query.py

@@ -21,7 +21,7 @@ def sf_instance():
     sf = SFAuth(
         instance_url=os.getenv("SF_INSTANCE_URL"),
         client_id=os.getenv("SF_CLIENT_ID"),
-        client_secret=os.getenv("SF_CLIENT_SECRET"),
+        client_secret=os.getenv("SF_CLIENT_SECRET").strip(),
         refresh_token=os.getenv("SF_REFRESH_TOKEN"),
     )
     return sf

{sfq-0.0.33 → sfq-0.0.35}/tests/test_query_e2e.py

@@ -32,7 +32,7 @@ def auth_manager():
     return AuthManager(
         instance_url=os.getenv("SF_INSTANCE_URL"),
         client_id=os.getenv("SF_CLIENT_ID"),
-        client_secret=os.getenv("SF_CLIENT_SECRET"),
+        client_secret=os.getenv("SF_CLIENT_SECRET").strip(),
         refresh_token=os.getenv("SF_REFRESH_TOKEN"),
     )
 
@@ -52,7 +52,7 @@ def sf_instance():
     sf = SFAuth(
         instance_url=os.getenv("SF_INSTANCE_URL"),
         client_id=os.getenv("SF_CLIENT_ID"),
-        client_secret=os.getenv("SF_CLIENT_SECRET"),
+        client_secret=os.getenv("SF_CLIENT_SECRET").strip(),
         refresh_token=os.getenv("SF_REFRESH_TOKEN"),
     )
     return sf

{sfq-0.0.33 → sfq-0.0.35}/tests/test_query_integration.py

@@ -33,7 +33,7 @@ def sf_auth():
     return SFAuth(
         instance_url=os.getenv("SF_INSTANCE_URL"),
         client_id=os.getenv("SF_CLIENT_ID"),
-        client_secret=os.getenv("SF_CLIENT_SECRET"),
+        client_secret=os.getenv("SF_CLIENT_SECRET").strip(),
         refresh_token=os.getenv("SF_REFRESH_TOKEN"),
     )
 
@@ -55,7 +55,7 @@ def modular_components():
     auth_manager = AuthManager(
         instance_url=os.getenv("SF_INSTANCE_URL"),
         client_id=os.getenv("SF_CLIENT_ID"),
-        client_secret=os.getenv("SF_CLIENT_SECRET"),
+        client_secret=os.getenv("SF_CLIENT_SECRET").strip(),
         refresh_token=os.getenv("SF_REFRESH_TOKEN"),
     )
 

sfq-0.0.35/tests/test_soap_batch_operation.py

@@ -0,0 +1,42 @@
+import pytest
+from sfq.crud import CRUDClient
+from sfq.soap import SOAPClient
+
+class DummyHTTPClient:
+    def __init__(self):
+        self.auth_manager = type('Auth', (), {'access_token': 'dummy_token'})()
+    def get_common_headers(self):
+        return {}
+    def send_request(self, method, endpoint, headers, body):
+        # Simulate a successful SOAP response for create/delete/update
+        if '<create>' in body:
+            return 200, '<result><id>001xx000003DGbEAAW</id><success>true</success></result>'
+        elif '<delete>' in body:
+            return 200, '<result><id>001xx000003DGbEAAW</id><success>true</success></result>'
+        elif '<update>' in body:
+            return 200, '<result><id>001xx000003DGbEAAW</id><success>true</success></result>'
+        return 500, '<faultstring>Error</faultstring>'
+
+@pytest.fixture
+def crud_client():
+    http_client = DummyHTTPClient()
+    soap_client = SOAPClient(http_client)
+    return CRUDClient(http_client, soap_client)
+
+def test_soap_batch_create(crud_client):
+    data = [{"Name": "Test Account"}]
+    result = crud_client._soap_batch_operation("Account", data, "create")
+    assert result is not None
+    assert result[0]["success"] == "true"
+
+def test_soap_batch_delete(crud_client):
+    ids = ["001xx000003DGbEAAW"]
+    result = crud_client._soap_batch_operation("Account", ids, "delete")
+    assert result is not None
+    assert result[0]["success"] == "true"
+
+def test_soap_batch_update(crud_client):
+    data = [{"Id": "001xx000003DGbEAAW", "Name": "Updated Account"}]
+    result = crud_client._soap_batch_operation("Account", data, "update")
+    assert result is not None
+    assert result[0]["success"] == "true"

{sfq-0.0.33 → sfq-0.0.35}/tests/test_static_resources.py

@@ -22,7 +22,7 @@ def sf_instance():
     sf = SFAuth(
         instance_url=os.getenv("SF_INSTANCE_URL"),
         client_id=os.getenv("SF_CLIENT_ID"),
-        client_secret=os.getenv("SF_CLIENT_SECRET"),
+        client_secret=os.getenv("SF_CLIENT_SECRET").strip(),
         refresh_token=os.getenv("SF_REFRESH_TOKEN"),
     )
     return sf

{sfq-0.0.33 → sfq-0.0.35}/uv.lock

@@ -3,5 +3,5 @@ requires-python = ">=3.9"
 
 [[package]]
 name = "sfq"
-version = "0.0.33"
+version = "0.0.35"
 source = { editable = "." }