Skip to content

Commit 226fe8a

Browse files
authored
VertexAI: update testing to clone mock responses from shared repo (#8333)
1 parent e7260e2 commit 226fe8a

24 files changed

+8
-425
lines changed

.gitignore

+2-1
Original file line numberDiff line numberDiff line change
@@ -98,5 +98,6 @@ docs/
9898
*.tfstate
9999
*.tfstate.*
100100

101-
# generated test case text data
101+
# vertexai test data
102+
vertexai-sdk-test-data
102103
mocks-lookup.ts

packages/vertexai/package.json

+2-1
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,8 @@
3333
"build": "rollup -c && yarn api-report",
3434
"build:deps": "lerna run --scope @firebase/vertexai --include-dependencies build",
3535
"dev": "rollup -c -w",
36-
"testsetup": "yarn ts-node ./test-utils/convert-mocks.ts",
36+
"update-responses": "cd test-utils && rm -rf vertexai-sdk-test-data && git clone --depth 1 https://github.com/FirebaseExtended/vertexai-sdk-test-data.git",
37+
"testsetup": "yarn update-responses && yarn ts-node ./test-utils/convert-mocks.ts",
3738
"test": "run-p --npm-path npm lint test:browser",
3839
"test:ci": "yarn testsetup && node ../../scripts/run_tests_in_ci.js -s test",
3940
"test:browser": "yarn testsetup && karma start --single-run",

packages/vertexai/test-utils/convert-mocks.ts

+4-1
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,10 @@ const fs = require('fs');
2525
// eslint-disable-next-line @typescript-eslint/no-require-imports
2626
const { join } = require('path');
2727

28-
const mockResponseDir = join(__dirname, 'mock-responses');
28+
const mockResponseDir = join(
29+
__dirname,
30+
'vertexai-sdk-test-data/mock-responses'
31+
);
2932

3033
async function main(): Promise<void> {
3134
const list = fs.readdirSync(mockResponseDir);

packages/vertexai/test-utils/mock-responses/streaming-failure-empty-content.txt

-2
This file was deleted.

packages/vertexai/test-utils/mock-responses/streaming-failure-finish-reason-safety.txt

-2
This file was deleted.

packages/vertexai/test-utils/mock-responses/streaming-failure-prompt-blocked-safety.txt

-2
This file was deleted.

packages/vertexai/test-utils/mock-responses/streaming-failure-recitation-no-content.txt

-6
This file was deleted.

packages/vertexai/test-utils/mock-responses/streaming-success-basic-reply-long.txt

-12
This file was deleted.

packages/vertexai/test-utils/mock-responses/streaming-success-basic-reply-short.txt

-2
This file was deleted.

packages/vertexai/test-utils/mock-responses/streaming-success-citations.txt

-12
This file was deleted.

packages/vertexai/test-utils/mock-responses/streaming-success-function-call-short.txt

-2
This file was deleted.

0 commit comments

Comments
 (0)