Add test setup for 3p packages

.github/workflows/ci.yml

@@ -62,6 +62,8 @@ jobs:
       - name: Bootstrap
         run: ./scripts/bootstrap
 
-      - name: Run tests
-        run: ./scripts/test
+      - name: Build
+        run: ./scripts/build-all
 
+      - name: Run tests
+        run: ./scripts/test-all

jest.config.ts

@@ -16,6 +16,7 @@ const config: JestConfigWithTsJest = {
     '<rootDir>/dist/',
     '<rootDir>/deno/',
     '<rootDir>/deno_tests/',
+    '<rootDir>/packages/',
   ],
   testPathIgnorePatterns: ['scripts'],
 };

package.json

@@ -14,7 +14,7 @@
   ],
   "private": false,
   "scripts": {
-    "test": "./scripts/test",
+    "test": "./scripts/test-all",
     "build": "./scripts/build-all",
     "prepublishOnly": "echo 'to publish, run yarn build && (cd dist; yarn publish)' && exit 1",
     "format": "prettier --write --cache --cache-strategy metadata . !dist",

packages/bedrock-sdk/jest.config.ts

@@ -0,0 +1,17 @@
+import type { JestConfigWithTsJest } from 'ts-jest';
+
+const config: JestConfigWithTsJest = {
+  preset: 'ts-jest/presets/default-esm',
+  testEnvironment: 'node',
+  transform: {
+    '^.+\\.(t|j)sx?$': ['@swc/jest', { sourceMaps: 'inline' }],
+  },
+  moduleNameMapper: {
+    '^@anthropic-ai/bedrock-sdk$': '<rootDir>/src/index.ts',
+    '^@anthropic-ai/bedrock-sdk/(.*)$': '<rootDir>/src/$1',
+  },
+  modulePathIgnorePatterns: ['<rootDir>/dist/', '<rootDir>/deno/', '<rootDir>/deno_tests/'],
+  testPathIgnorePatterns: ['scripts'],
+};
+
+export default config;

packages/bedrock-sdk/package.json

@@ -11,7 +11,7 @@
   "packageManager": "[email protected]",
   "private": false,
   "scripts": {
-    "test": "echo 'no tests defined yet' && exit 1",
+    "test": "jest",
     "build": "bash ./build",
     "prepack": "echo 'to pack, run yarn build && (cd dist; yarn pack)' && exit 1",
     "prepublishOnly": "echo 'to publish, run yarn build && (cd dist; yarn publish)' && exit 1",

packages/bedrock-sdk/scripts/test

@@ -0,0 +1,7 @@
+#!/usr/bin/env bash
+
+set -e
+
+cd "$(dirname "$0")/.."
+
+./node_modules/.bin/jest "$@"

packages/bedrock-sdk/src/client.ts

@@ -117,6 +117,12 @@ export class AnthropicBedrock extends Core.APIClient {
   } {
     options.__streamClass = Stream;
 
+    if (Core.isObj(options.body)) {
+      // create a shallow copy of the request body so that code that mutates it later
+      // doesn't mutate the original user-provided object
+      options.body = { ...options.body };
+    }
+
     if (Core.isObj(options.body)) {
       if (!options.body['anthropic_version']) {
         options.body['anthropic_version'] = DEFAULT_VERSION;

packages/bedrock-sdk/tests/client.test.ts

@@ -0,0 +1,29 @@
+import { AnthropicBedrock } from '@anthropic-ai/bedrock-sdk';
+import Anthropic from '@anthropic-ai/sdk';
+
+test('body params are not mutated', async () => {
+  const client = new AnthropicBedrock({
+    baseURL: 'http://localhost:5000/',
+    awsRegion: 'us-east-1',
+    awsAccessKey: 'placeholder',
+    awsSecretKey: 'placeholder',
+    fetch: (url) => {
+      return Promise.resolve(
+        new Response(JSON.stringify({ url, custom: true }), {
+          headers: { 'Content-Type': 'application/json' },
+        }),
+      );
+    },
+  });
+
+  const params: Anthropic.MessageCreateParamsNonStreaming = {
+    model: 'claude-3-opus-latest',
+    messages: [],
+    max_tokens: 100,
+  };
+  const original = JSON.stringify(params);
+
+  await client.messages.create(params);
+
+  expect(JSON.stringify(params)).toEqual(original);
+});

packages/vertex-sdk/jest.config.ts

@@ -0,0 +1,17 @@
+import type { JestConfigWithTsJest } from 'ts-jest';
+
+const config: JestConfigWithTsJest = {
+  preset: 'ts-jest/presets/default-esm',
+  testEnvironment: 'node',
+  transform: {
+    '^.+\\.(t|j)sx?$': ['@swc/jest', { sourceMaps: 'inline' }],
+  },
+  moduleNameMapper: {
+    '^@anthropic-ai/vertex-sdk$': '<rootDir>/src/index.ts',
+    '^@anthropic-ai/vertex-sdk/(.*)$': '<rootDir>/src/$1',
+  },
+  modulePathIgnorePatterns: ['<rootDir>/dist/', '<rootDir>/deno/', '<rootDir>/deno_tests/'],
+  testPathIgnorePatterns: ['scripts'],
+};
+
+export default config;

packages/vertex-sdk/package.json

@@ -11,7 +11,7 @@
   "packageManager": "[email protected]",
   "private": false,
   "scripts": {
-    "test": "echo 'no tests defined yet' && exit 1",
+    "test": "jest",
     "build": "bash ./build",
     "prepack": "echo 'to pack, run yarn build && (cd dist; yarn pack)' && exit 1",
     "prepublishOnly": "echo 'to publish, run yarn build && (cd dist; yarn publish)' && exit 1",

packages/vertex-sdk/scripts/test

@@ -0,0 +1,7 @@
+#!/usr/bin/env bash
+
+set -e
+
+cd "$(dirname "$0")/.."
+
+./node_modules/.bin/jest "$@"

packages/vertex-sdk/src/client.ts

@@ -114,6 +114,12 @@ export class AnthropicVertex extends Core.APIClient {
     url: string;
     timeout: number;
   } {
+    if (Core.isObj(options.body)) {
+      // create a shallow copy of the request body so that code that mutates it later
+      // doesn't mutate the original user-provided object
+      options.body = { ...options.body };
+    }
+
     if (Core.isObj(options.body)) {
       if (!options.body['anthropic_version']) {
         options.body['anthropic_version'] = DEFAULT_VERSION;

packages/vertex-sdk/tests/client.test.ts

@@ -0,0 +1,29 @@
+import { AnthropicVertex } from '@anthropic-ai/vertex-sdk';
+import Anthropic from '@anthropic-ai/sdk';
+
+test('body params are not mutated', async () => {
+  const client = new AnthropicVertex({
+    baseURL: 'http://localhost:5000/',
+    accessToken: 'placeholder',
+    region: 'us-east-2',
+    projectId: 'placeholder',
+    fetch: (url) => {
+      return Promise.resolve(
+        new Response(JSON.stringify({ url, custom: true }), {
+          headers: { 'Content-Type': 'application/json' },
+        }),
+      );
+    },
+  });
+
+  const params: Anthropic.MessageCreateParamsNonStreaming = {
+    model: 'claude-3-opus-latest',
+    messages: [],
+    max_tokens: 100,
+  };
+  const original = JSON.stringify(params);
+
+  await client.messages.create(params);
+
+  expect(JSON.stringify(params)).toEqual(original);
+});

scripts/test-all

@@ -0,0 +1,12 @@
+#!/usr/bin/env bash
+
+set -e
+
+bash ./scripts/test
+
+for dir in packages/*; do
+    if [ -f "$dir/scripts/test" ]; then
+        echo "==> Running tests in $dir"
+        (cd "$dir" && bash scripts/test)
+    fi
+done