VertexAI: use shared repository for test golden files (#13186)

firebase · Jun 26, 2024 · 18e5d6b · 18e5d6b
1 parent 26fcf04
commit 18e5d6b
Show file tree

Hide file tree

Showing 45 changed files with 35 additions and 879 deletions.
diff --git a/.github/workflows/spm.yml b/.github/workflows/spm.yml
@@ -42,6 +42,8 @@ jobs:
     - uses: mikehardy/buildcache-action@c87cea0ccd718971d6cc39e672c4f26815b6c126
       with:
         cache_key: spm${{ matrix.os }}-${{ matrix.xcode }}-${{ matrix.target }}
+    - name: Clone mock responses for Vertex AI unit tests
+      run: scripts/update_vertexai_responses.sh
     - name: Xcode
       run: sudo xcode-select -s /Applications/${{ matrix.xcode }}.app/Contents/Developer
     - name: Initialize xcodebuild
@@ -74,6 +76,8 @@ jobs:
     - uses: mikehardy/buildcache-action@c87cea0ccd718971d6cc39e672c4f26815b6c126
       with:
         cache_key: device${{ matrix.os }}${{ matrix.xcode }}
+    - name: Clone mock responses for Vertex AI unit tests
+      run: scripts/update_vertexai_responses.sh
     - name: Xcode
       run: sudo xcode-select -s /Applications/${{ matrix.xcode }}.app/Contents/Developer
     - name: Initialize xcodebuild

diff --git a/.github/workflows/vertexai.yml b/.github/workflows/vertexai.yml
@@ -28,6 +28,8 @@ jobs:
       FIREBASECI_USE_LATEST_GOOGLEAPPMEASUREMENT: 1
     steps:
     - uses: actions/checkout@v4
+    - name: Clone mock responses
+      run: scripts/update_vertexai_responses.sh
     - name: Xcode
       run: sudo xcode-select -s /Applications/${{ matrix.xcode }}.app/Contents/Developer
     - name: Initialize xcodebuild

diff --git a/.gitignore b/.gitignore
@@ -162,3 +162,6 @@ FirebaseAppCheck/Apps/AppCheckCustomProvideApp/AppCheckCustomProvideApp/GoogleSe
 .terraform.lock.hcl
 *.tfstate
 *.tfstate.*
+
+# FirebaseVertexAI test data
+vertexai-sdk-test-data
diff --git a/FirebaseVertexAI/Tests/Unit/Responses/streaming-failure-empty-content.txt b/FirebaseVertexAI/Tests/Unit/Responses/streaming-failure-empty-content.txt
diff --git a/FirebaseVertexAI/Tests/Unit/Responses/streaming-failure-error-mid-stream.txt b/FirebaseVertexAI/Tests/Unit/Responses/streaming-failure-error-mid-stream.txt
diff --git a/FirebaseVertexAI/Tests/Unit/Responses/streaming-failure-finish-reason-safety.txt b/FirebaseVertexAI/Tests/Unit/Responses/streaming-failure-finish-reason-safety.txt
diff --git a/FirebaseVertexAI/Tests/Unit/Responses/streaming-failure-invalid-json.txt b/FirebaseVertexAI/Tests/Unit/Responses/streaming-failure-invalid-json.txt
diff --git a/FirebaseVertexAI/Tests/Unit/Responses/streaming-failure-malformed-content.txt b/FirebaseVertexAI/Tests/Unit/Responses/streaming-failure-malformed-content.txt
diff --git a/FirebaseVertexAI/Tests/Unit/Responses/streaming-failure-prompt-blocked-safety.txt b/FirebaseVertexAI/Tests/Unit/Responses/streaming-failure-prompt-blocked-safety.txt
diff --git a/FirebaseVertexAI/Tests/Unit/Responses/streaming-failure-recitation-no-content.txt b/FirebaseVertexAI/Tests/Unit/Responses/streaming-failure-recitation-no-content.txt
diff --git a/FirebaseVertexAI/Tests/Unit/Responses/streaming-failure-unknown-finish-enum.txt b/FirebaseVertexAI/Tests/Unit/Responses/streaming-failure-unknown-finish-enum.txt
diff --git a/FirebaseVertexAI/Tests/Unit/Responses/streaming-success-basic-reply-long.txt b/FirebaseVertexAI/Tests/Unit/Responses/streaming-success-basic-reply-long.txt