fix(google-sheets): fix auth to use .get(), add integration test, update .env.example

- Fix build_credentials to use context.auth.get("credentials", {}).get("access_token", "")
- Add test_google_sheets_integration.py with GOOGLE_SHEETS_ACCESS_TOKEN skip guard
- Add GOOGLE_SHEETS_ACCESS_TOKEN and GOOGLE_SHEETS_TEST_SPREADSHEET_ID to .env.example
- Apply ruff format to all google-sheets files

Author: Shubhank-Jonnada

.env.example

@@ -59,3 +59,7 @@
 
 # -- Xero --
 # (uses platform OAuth — tokens are short-lived, typically not set here)
+
+# -- Google Sheets --
+# GOOGLE_SHEETS_ACCESS_TOKEN=
+# GOOGLE_SHEETS_TEST_SPREADSHEET_ID=

google-sheets/google_sheets.py

@@ -1,4 +1,10 @@
-from autohive_integrations_sdk import Integration, ExecutionContext, ActionHandler, ActionResult, ActionError
+from autohive_integrations_sdk import (
+    Integration,
+    ExecutionContext,
+    ActionHandler,
+    ActionResult,
+    ActionError,
+)
 from typing import Dict, Any, List
 from googleapiclient.discovery import build
 from googleapiclient.errors import HttpError
@@ -10,8 +16,10 @@
 
 
 def build_credentials(context: ExecutionContext) -> Credentials:
-    access_token = context.auth["credentials"]["access_token"]
-    return Credentials(token=access_token, token_uri="https://oauth2.googleapis.com/token")  # nosec B106
+    access_token = context.auth.get("credentials", {}).get("access_token", "")
+    return Credentials(
+        token=access_token, token_uri="https://oauth2.googleapis.com/token"
+    )  # nosec B106
 
 
 def build_sheets_service(context: ExecutionContext):
@@ -72,7 +80,9 @@ async def execute(self, inputs: Dict[str, Any], context: ExecutionContext):
             service = build_sheets_service(context)
             spreadsheet_id = inputs["spreadsheet_id"]
             include_grid = bool(inputs.get("include_grid_data", False))
-            request = service.spreadsheets().get(spreadsheetId=spreadsheet_id, includeGridData=include_grid)
+            request = service.spreadsheets().get(
+                spreadsheetId=spreadsheet_id, includeGridData=include_grid
+            )
             spreadsheet = request.execute()
             return ActionResult(data={"spreadsheet": spreadsheet}, cost_usd=0.0)
         except HttpError as e:
@@ -120,7 +130,10 @@ async def execute(self, inputs: Dict[str, Any], context: ExecutionContext):
                 params["dateTimeRenderOption"] = dt_render
             result = service.spreadsheets().values().get(**params).execute()
             return ActionResult(
-                data={"range": result.get("range", a1), "values": result.get("values", [])},
+                data={
+                    "range": result.get("range", a1),
+                    "values": result.get("values", []),
+                },
                 cost_usd=0.0,
             )
         except HttpError as e:
@@ -142,7 +155,11 @@ async def execute(self, inputs: Dict[str, Any], context: ExecutionContext):
             if dry_run:
                 # Validate by attempting a read of the target range to ensure spreadsheet exists
                 service = build_sheets_service(context)
-                _ = service.spreadsheets().get(spreadsheetId=spreadsheet_id, includeGridData=False).execute()
+                _ = (
+                    service.spreadsheets()
+                    .get(spreadsheetId=spreadsheet_id, includeGridData=False)
+                    .execute()
+                )
                 # Estimate cells
                 rows = len(values)
                 cols = max((len(r) for r in values), default=0)
@@ -208,7 +225,9 @@ async def execute(self, inputs: Dict[str, Any], context: ExecutionContext):
                 )
                 .execute()
             )
-            return ActionResult(data={"updates": result.get("updates", result)}, cost_usd=0.0)
+            return ActionResult(
+                data={"updates": result.get("updates", result)}, cost_usd=0.0
+            )
         except HttpError as e:
             return ActionError(message=f"Google Sheets API error: {str(e)}")
         except Exception as e:
@@ -235,9 +254,13 @@ async def execute(self, inputs: Dict[str, Any], context: ExecutionContext):
                 }
             ]
             result = (
-                service.spreadsheets().batchUpdate(spreadsheetId=spreadsheet_id, body={"requests": requests}).execute()
+                service.spreadsheets()
+                .batchUpdate(spreadsheetId=spreadsheet_id, body={"requests": requests})
+                .execute()
+            )
+            return ActionResult(
+                data={"replies": result.get("replies", [])}, cost_usd=0.0
             )
-            return ActionResult(data={"replies": result.get("replies", [])}, cost_usd=0.0)
         except HttpError as e:
             return ActionError(message=f"Google Sheets API error: {str(e)}")
         except Exception as e:
@@ -273,9 +296,13 @@ async def execute(self, inputs: Dict[str, Any], context: ExecutionContext):
             ]
 
             result = (
-                service.spreadsheets().batchUpdate(spreadsheetId=spreadsheet_id, body={"requests": requests}).execute()
+                service.spreadsheets()
+                .batchUpdate(spreadsheetId=spreadsheet_id, body={"requests": requests})
+                .execute()
+            )
+            return ActionResult(
+                data={"replies": result.get("replies", [])}, cost_usd=0.0
             )
-            return ActionResult(data={"replies": result.get("replies", [])}, cost_usd=0.0)
         except HttpError as e:
             return ActionError(message=f"Google Sheets API error: {str(e)}")
         except Exception as e:
@@ -291,18 +318,26 @@ async def execute(self, inputs: Dict[str, Any], context: ExecutionContext):
             dry_run = bool(inputs.get("dry_run", False))
 
             # Basic validation: ensure it's a list of dicts
-            if not isinstance(requests, list) or not all(isinstance(r, dict) for r in requests):
+            if not isinstance(requests, list) or not all(
+                isinstance(r, dict) for r in requests
+            ):
                 return ActionError(message="requests must be an array of objects")
 
             if dry_run:
                 # Validate by fetching spreadsheet metadata
                 service = build_sheets_service(context)
-                _ = service.spreadsheets().get(spreadsheetId=spreadsheet_id, includeGridData=False).execute()
+                _ = (
+                    service.spreadsheets()
+                    .get(spreadsheetId=spreadsheet_id, includeGridData=False)
+                    .execute()
+                )
                 return ActionResult(data={"replies": [], "dryRun": True}, cost_usd=0.0)
 
             service = build_sheets_service(context)
             result = (
-                service.spreadsheets().batchUpdate(spreadsheetId=spreadsheet_id, body={"requests": requests}).execute()
+                service.spreadsheets()
+                .batchUpdate(spreadsheetId=spreadsheet_id, body={"requests": requests})
+                .execute()
             )
             return ActionResult(
                 data={"replies": result.get("replies", []), "dryRun": False},

google-sheets/tests/context.py

@@ -4,6 +4,7 @@
 
 # Add integration root and vendored dependencies to path
 sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), "..")))
-sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), "../dependencies")))
+sys.path.insert(
+    0, os.path.abspath(os.path.join(os.path.dirname(__file__), "../dependencies"))
+)
 
-from google_sheets import google_sheets

google-sheets/tests/test_google_sheets.py

@@ -57,7 +57,9 @@ def decorator(handler_class):
 
         return decorator
 
-    async def execute_action(self, action_name: str, inputs: Dict[str, Any], context: MockExecutionContext):
+    async def execute_action(
+        self, action_name: str, inputs: Dict[str, Any], context: MockExecutionContext
+    ):
         if action_name in self._actions:
             handler = self._actions[action_name]()
             return await handler.execute(inputs, context)
@@ -119,7 +121,9 @@ async def test_list_spreadsheets_success(self, mock_build):
         assert result["nextPageToken"] == "next_token"
 
         # Verify API calls
-        mock_build.assert_called_once_with("drive", "v3", credentials=mock_build.call_args[1]["credentials"])
+        mock_build.assert_called_once_with(
+            "drive", "v3", credentials=mock_build.call_args[1]["credentials"]
+        )
 
     @patch("google_sheets.build")
     async def test_list_spreadsheets_with_filters(self, mock_build):
@@ -333,7 +337,9 @@ async def run_all_tests():
     test_instance = TestGoogleSheetsIntegration()
 
     # List all test methods
-    test_methods = [method for method in dir(test_instance) if method.startswith("test_")]
+    test_methods = [
+        method for method in dir(test_instance) if method.startswith("test_")
+    ]
 
     print("Running Google Sheets integration tests...")
     print("=" * 50)

google-sheets/tests/test_google_sheets_data_unit.py

@@ -11,7 +11,9 @@
 from unittest.mock import AsyncMock, MagicMock, patch  # noqa: E402
 from autohive_integrations_sdk.integration import ResultType  # noqa: E402
 
-_spec = importlib.util.spec_from_file_location("google_sheets_data_mod", os.path.join(_parent, "google_sheets.py"))
+_spec = importlib.util.spec_from_file_location(
+    "google_sheets_data_mod", os.path.join(_parent, "google_sheets.py")
+)
 _mod = importlib.util.module_from_spec(_spec)
 sys.modules["google_sheets_data_mod"] = _mod
 _spec.loader.exec_module(_mod)
@@ -52,7 +54,9 @@ async def test_happy_path_returns_values(self, mock_build, mock_context):
         }
 
         result = await google_sheets.execute_action(
-            "sheets_read_range", {"spreadsheet_id": "sid", "range": "Sheet1!A1:B2"}, mock_context
+            "sheets_read_range",
+            {"spreadsheet_id": "sid", "range": "Sheet1!A1:B2"},
+            mock_context,
         )
 
         assert result.result.data["range"] == "Sheet1!A1:B2"
@@ -62,10 +66,14 @@ async def test_happy_path_returns_values(self, mock_build, mock_context):
     @patch("google_sheets_data_mod.build")
     async def test_empty_range_returns_empty_values(self, mock_build, mock_context):
         service = make_sheets_service(mock_build)
-        service.spreadsheets().values().get().execute.return_value = {"range": "Sheet1!A1"}
+        service.spreadsheets().values().get().execute.return_value = {
+            "range": "Sheet1!A1"
+        }
 
         result = await google_sheets.execute_action(
-            "sheets_read_range", {"spreadsheet_id": "sid", "range": "Sheet1!A1"}, mock_context
+            "sheets_read_range",
+            {"spreadsheet_id": "sid", "range": "Sheet1!A1"},
+            mock_context,
         )
 
         assert result.result.data["values"] == []
@@ -74,7 +82,10 @@ async def test_empty_range_returns_empty_values(self, mock_build, mock_context):
     @patch("google_sheets_data_mod.build")
     async def test_value_render_option_passed(self, mock_build, mock_context):
         service = make_sheets_service(mock_build)
-        service.spreadsheets().values().get().execute.return_value = {"range": "A1", "values": []}
+        service.spreadsheets().values().get().execute.return_value = {
+            "range": "A1",
+            "values": [],
+        }
 
         await google_sheets.execute_action(
             "sheets_read_range",
@@ -89,11 +100,18 @@ async def test_value_render_option_passed(self, mock_build, mock_context):
     @patch("google_sheets_data_mod.build")
     async def test_datetime_render_option_passed(self, mock_build, mock_context):
         service = make_sheets_service(mock_build)
-        service.spreadsheets().values().get().execute.return_value = {"range": "A1", "values": []}
+        service.spreadsheets().values().get().execute.return_value = {
+            "range": "A1",
+            "values": [],
+        }
 
         await google_sheets.execute_action(
             "sheets_read_range",
-            {"spreadsheet_id": "sid", "range": "A1", "dateTimeRenderOption": "FORMATTED_STRING"},
+            {
+                "spreadsheet_id": "sid",
+                "range": "A1",
+                "dateTimeRenderOption": "FORMATTED_STRING",
+            },
             mock_context,
         )
 
@@ -109,18 +127,24 @@ async def test_http_error_returns_action_error(self, mock_build, mock_context):
         mock_resp = MagicMock()
         mock_resp.status = 400
         mock_resp.reason = "Bad Request"
-        service.spreadsheets().values().get().execute.side_effect = HttpError(mock_resp, b"Bad Request")
+        service.spreadsheets().values().get().execute.side_effect = HttpError(
+            mock_resp, b"Bad Request"
+        )
 
         result = await google_sheets.execute_action(
-            "sheets_read_range", {"spreadsheet_id": "sid", "range": "Bad!Range"}, mock_context
+            "sheets_read_range",
+            {"spreadsheet_id": "sid", "range": "Bad!Range"},
+            mock_context,
         )
 
         assert result.type == ResultType.ACTION_ERROR
         assert "Google Sheets API error" in result.result.message
 
     @pytest.mark.asyncio
     @patch("google_sheets_data_mod.build")
-    async def test_generic_exception_returns_action_error(self, mock_build, mock_context):
+    async def test_generic_exception_returns_action_error(
+        self, mock_build, mock_context
+    ):
         service = make_sheets_service(mock_build)
         service.spreadsheets().values().get().execute.side_effect = Exception("Timeout")
 
@@ -149,7 +173,11 @@ async def test_happy_path_returns_update_info(self, mock_build, mock_context):
 
         result = await google_sheets.execute_action(
             "sheets_write_range",
-            {"spreadsheet_id": "sid", "range": "Sheet1!A1:B2", "values": [["Name", "Age"], ["Alice", "30"]]},
+            {
+                "spreadsheet_id": "sid",
+                "range": "Sheet1!A1:B2",
+                "values": [["Name", "Age"], ["Alice", "30"]],
+            },
             mock_context,
         )
 
@@ -159,13 +187,20 @@ async def test_happy_path_returns_update_info(self, mock_build, mock_context):
 
     @pytest.mark.asyncio
     @patch("google_sheets_data_mod.build")
-    async def test_dry_run_returns_estimate_without_write(self, mock_build, mock_context):
+    async def test_dry_run_returns_estimate_without_write(
+        self, mock_build, mock_context
+    ):
         service = make_sheets_service(mock_build)
         service.spreadsheets().get().execute.return_value = {"spreadsheetId": "sid"}
 
         result = await google_sheets.execute_action(
             "sheets_write_range",
-            {"spreadsheet_id": "sid", "range": "A1", "values": [["a", "b"], ["c", "d"]], "dry_run": True},
+            {
+                "spreadsheet_id": "sid",
+                "range": "A1",
+                "values": [["a", "b"], ["c", "d"]],
+                "dry_run": True,
+            },
             mock_context,
         )
 
@@ -200,7 +235,9 @@ async def test_http_error_returns_action_error(self, mock_build, mock_context):
         mock_resp = MagicMock()
         mock_resp.status = 403
         mock_resp.reason = "Forbidden"
-        service.spreadsheets().values().update().execute.side_effect = HttpError(mock_resp, b"Forbidden")
+        service.spreadsheets().values().update().execute.side_effect = HttpError(
+            mock_resp, b"Forbidden"
+        )
 
         result = await google_sheets.execute_action(
             "sheets_write_range",
@@ -213,9 +250,13 @@ async def test_http_error_returns_action_error(self, mock_build, mock_context):
 
     @pytest.mark.asyncio
     @patch("google_sheets_data_mod.build")
-    async def test_generic_exception_returns_action_error(self, mock_build, mock_context):
+    async def test_generic_exception_returns_action_error(
+        self, mock_build, mock_context
+    ):
         service = make_sheets_service(mock_build)
-        service.spreadsheets().values().update().execute.side_effect = Exception("Write failed")
+        service.spreadsheets().values().update().execute.side_effect = Exception(
+            "Write failed"
+        )
 
         result = await google_sheets.execute_action(
             "sheets_write_range",
@@ -241,7 +282,11 @@ async def test_happy_path_returns_updates(self, mock_build, mock_context):
 
         result = await google_sheets.execute_action(
             "sheets_append_rows",
-            {"spreadsheet_id": "sid", "range": "Sheet1", "rows": [["a", "b"], ["c", "d"], ["e", "f"]]},
+            {
+                "spreadsheet_id": "sid",
+                "range": "Sheet1",
+                "rows": [["a", "b"], ["c", "d"], ["e", "f"]],
+            },
             mock_context,
         )
 
@@ -286,7 +331,9 @@ async def test_http_error_returns_action_error(self, mock_build, mock_context):
         mock_resp = MagicMock()
         mock_resp.status = 429
         mock_resp.reason = "Too Many Requests"
-        service.spreadsheets().values().append().execute.side_effect = HttpError(mock_resp, b"Rate limited")
+        service.spreadsheets().values().append().execute.side_effect = HttpError(
+            mock_resp, b"Rate limited"
+        )
 
         result = await google_sheets.execute_action(
             "sheets_append_rows",
@@ -299,9 +346,13 @@ async def test_http_error_returns_action_error(self, mock_build, mock_context):
 
     @pytest.mark.asyncio
     @patch("google_sheets_data_mod.build")
-    async def test_generic_exception_returns_action_error(self, mock_build, mock_context):
+    async def test_generic_exception_returns_action_error(
+        self, mock_build, mock_context
+    ):
         service = make_sheets_service(mock_build)
-        service.spreadsheets().values().append().execute.side_effect = Exception("Append failed")
+        service.spreadsheets().values().append().execute.side_effect = Exception(
+            "Append failed"
+        )
 
         result = await google_sheets.execute_action(
             "sheets_append_rows",