From b22cded63cbdfddd58d80c9be8dda95b40c32679 Mon Sep 17 00:00:00 2001
From: Colin <colin.rogers@dbtlabs.com>
Date: Tue, 13 Feb 2024 12:13:09 -0800
Subject: [PATCH] Update test requirements in pyproject.toml via bash script

---
 .github/workflows/code-quality.yml      | 21 +++++++++++++++++++
 .github/workflows/integration-tests.yml | 27 +++++++++++++++++++++++++
 2 files changed, 48 insertions(+)

diff --git a/.github/workflows/code-quality.yml b/.github/workflows/code-quality.yml
index dedea94b..1318c159 100644
--- a/.github/workflows/code-quality.yml
+++ b/.github/workflows/code-quality.yml
@@ -7,6 +7,19 @@ on:
       - "*.latest"
   pull_request:
   workflow_dispatch:
+    dbt_adapters_branch:
+      inputs:
+        description: "The branch of dbt-adapters to evaluate"
+        type: string
+        required: true
+        default: "main"
+  workflow_call:
+    inputs:
+      dbt_adapters_branch:
+        description: "The branch of dbt-adapters to evaluate"
+        type: string
+        required: true
+        default: "main"
 
 permissions: read-all
 
@@ -26,6 +39,14 @@ jobs:
         with:
           persist-credentials: false
 
+      - name: Update Adapters and Core branches
+        if: ${{ github.event_name == 'workflow_call' || github.event_name == 'workflow_dispatch'}}
+        shell: bash
+        run: |
+          ./.github/scripts/update_dev_packages.sh \
+            "main" \
+            ${{ github.event.inputs.dbt_adapters_branch }}
+
       - name: Setup `hatch`
         uses: dbt-labs/dbt-adapters/.github/actions/setup-hatch@main
 
diff --git a/.github/workflows/integration-tests.yml b/.github/workflows/integration-tests.yml
index af2383bd..e3d15b1f 100644
--- a/.github/workflows/integration-tests.yml
+++ b/.github/workflows/integration-tests.yml
@@ -7,6 +7,25 @@ on:
       - "*.latest"
   pull_request:
   workflow_dispatch:
+    dbt_adapters_branch:
+      inputs:
+        description: "The branch of dbt-adapters to evaluate"
+        type: string
+        required: true
+        default: "main"
+  workflow_call:
+    inputs:
+      dbt_adapters_branch:
+        description: "The branch of dbt-adapters to use"
+        type: string
+        required: true
+        default: "main"
+      core_branch:
+        description: "The branch of dbt-core to use"
+        type: string
+        required: true
+        default: "main"
+
 
 permissions: read-all
 
@@ -42,6 +61,14 @@ jobs:
       - name: Check out repository
         uses: actions/checkout@v4
 
+      - name: Update Adapters and Core branches
+        if: ${{ github.event_name == 'workflow_call' || github.event_name == 'workflow_dispatch'}}
+        shell: bash
+        run: |
+          ./.github/scripts/update_dev_packages.sh \
+            ${{ github.event.inputs.core_branch }} \
+            ${{ github.event.inputs.dbt_adapters_branch }}
+
       - name: Setup postgres
         shell: bash
         run: psql -f ./scripts/setup_test_database.sql