From c1ba2467733543076a6e3071dbb3d6c7eb953fec Mon Sep 17 00:00:00 2001
From: Aaron Steers <aj@airbyte.io>
Date: Thu, 8 Aug 2024 19:07:45 -0700
Subject: [PATCH 1/2] chore: add example script to repro bigquery cache issue

---
 examples/run_salesforce_to_bigquery_cache.py | 72 ++++++++++++++++++++
 1 file changed, 72 insertions(+)
 create mode 100644 examples/run_salesforce_to_bigquery_cache.py

diff --git a/examples/run_salesforce_to_bigquery_cache.py b/examples/run_salesforce_to_bigquery_cache.py
new file mode 100644
index 00000000..05e2ffc2
--- /dev/null
+++ b/examples/run_salesforce_to_bigquery_cache.py
@@ -0,0 +1,72 @@
+# Copyright (c) 2023 Airbyte, Inc., all rights reserved.
+"""
+Usage:
+    poetry run python examples/run_salesforce_to_bigquery_cache.py
+"""
+
+from __future__ import annotations
+
+import tempfile
+import warnings
+from typing import cast
+
+import airbyte as ab
+from airbyte.caches.bigquery import BigQueryCache
+from airbyte.secrets.base import SecretString
+from airbyte.secrets.google_gsm import GoogleGSMSecretManager
+
+warnings.filterwarnings("ignore", message="Cannot create BigQuery Storage client")
+
+
+AIRBYTE_INTERNAL_GCP_PROJECT = "dataline-integration-testing"
+BIGQUERY_SECRET_NAME = "SECRET_DESTINATION-BIGQUERY_CREDENTIALS__CREDS"
+
+secret_mgr = GoogleGSMSecretManager(
+    project=AIRBYTE_INTERNAL_GCP_PROJECT,
+    credentials_json=ab.get_secret("GCP_GSM_CREDENTIALS"),
+)
+
+source_salesforce_config: dict = cast(
+    SecretString,
+    secret_mgr.fetch_connector_secret("source-salesforce"),
+).parse_json()
+bigquery_destination_config: dict = cast(
+    SecretString,
+    secret_mgr.get_secret(BIGQUERY_SECRET_NAME),
+).parse_json()
+
+
+def main() -> None:
+    source = ab.get_source(
+        "source-salesforce",
+        config=source_salesforce_config,
+        install_if_missing=True,
+    )
+    source.check()
+    source.select_all_streams()
+
+    with tempfile.NamedTemporaryFile(mode="w+", delete=False, encoding="utf-8") as temp:
+        # Write credentials to the temp file
+        temp.write(bigquery_destination_config["credentials_json"])
+        temp.flush()
+        temp.close()
+
+        cache = BigQueryCache(
+            project_name=bigquery_destination_config["project_id"],
+            dataset_name=bigquery_destination_config.get(
+                "dataset_id", "pyairbyte_integtest"
+            ),
+            credentials_path=temp.name,
+        )
+
+        result = source.read(cache)
+
+        # Read a second time to make sure table swaps and incremental are working.
+        result = source.read(cache)
+
+        for name, records in result.streams.items():
+            print(f"Stream {name}: {len(records)} records")
+
+
+if __name__ == "__main__":
+    main()

From f52237626c6c02127aaaee35fdf5e96ac11342b2 Mon Sep 17 00:00:00 2001
From: Aaron Steers <aj@airbyte.io>
Date: Fri, 9 Aug 2024 11:04:56 -0700
Subject: [PATCH 2/2] add bigquery destination test script

---
 .../run_salesforce_to_bigquery_destination.py | 66 +++++++++++++++++++
 1 file changed, 66 insertions(+)
 create mode 100644 examples/run_salesforce_to_bigquery_destination.py

diff --git a/examples/run_salesforce_to_bigquery_destination.py b/examples/run_salesforce_to_bigquery_destination.py
new file mode 100644
index 00000000..e6342dfc
--- /dev/null
+++ b/examples/run_salesforce_to_bigquery_destination.py
@@ -0,0 +1,66 @@
+# Copyright (c) 2023 Airbyte, Inc., all rights reserved.
+"""
+Usage:
+    poetry run python examples/run_salesforce_to_bigquery_cache.py
+"""
+
+from __future__ import annotations
+
+import tempfile
+import warnings
+from typing import cast
+
+import airbyte as ab
+from airbyte.secrets.base import SecretString
+from airbyte.secrets.google_gsm import GoogleGSMSecretManager
+
+warnings.filterwarnings("ignore", message="Cannot create BigQuery Storage client")
+
+
+AIRBYTE_INTERNAL_GCP_PROJECT = "dataline-integration-testing"
+BIGQUERY_SECRET_NAME = "SECRET_DESTINATION-BIGQUERY_CREDENTIALS__CREDS"
+
+secret_mgr = GoogleGSMSecretManager(
+    project=AIRBYTE_INTERNAL_GCP_PROJECT,
+    credentials_json=ab.get_secret("GCP_GSM_CREDENTIALS"),
+)
+
+source_salesforce_config: dict = cast(
+    SecretString,
+    secret_mgr.fetch_connector_secret("source-salesforce"),
+).parse_json()
+bigquery_destination_config: dict = cast(
+    SecretString,
+    secret_mgr.get_secret(BIGQUERY_SECRET_NAME),
+).parse_json()
+
+
+def main() -> None:
+    source = ab.get_source(
+        "source-salesforce",
+        config=source_salesforce_config,
+        install_if_missing=True,
+    )
+    source.check()
+    source.select_all_streams()
+
+    with tempfile.NamedTemporaryFile(mode="w+", delete=False, encoding="utf-8") as temp:
+        # Write credentials to the temp file
+        temp.write(bigquery_destination_config["credentials_json"])
+        temp.flush()
+        temp.close()
+
+        destination = ab.get_destination(
+            "destination-bigquery",
+            config={**bigquery_destination_config, "dataset_id": "pyairbyte_tests"},
+        )
+
+        destination.write(
+            source,
+            cache=ab.get_default_cache(),
+            # cache=False,  # Toggle comment to test with/without caching
+        )
+
+
+if __name__ == "__main__":
+    main()