Add browser integration tests

Author: aron

integration/browser/.npmrc

@@ -0,0 +1,3 @@
+package-lock=false
+audit=false
+fund=false

integration/browser/index.js

@@ -0,0 +1,22 @@
+import Replicate from "replicate";
+
+/**
+ * @param {string} - token the REPLICATE_API_TOKEN
+ */
+window.main = async (token) => {
+  const replicate = new Replicate({ auth: token });
+  const stream = replicate.stream(
+    "replicate/canary:30e22229542eb3f79d4f945dacb58d32001b02cc313ae6f54eef27904edf3272",
+    {
+      input: {
+        text: "Betty Browser",
+      },
+    }
+  );
+
+  const output = [];
+  for await (const event of stream) {
+    output.push(String(event));
+  }
+  return output.join("");
+};

integration/browser/index.test.js

@@ -0,0 +1,34 @@
+import { test, expect } from "@playwright/test";
+import { build } from "esbuild";
+
+// Convert the source file from commonjs to a browser script.
+const result = await build({
+  entryPoints: ["index.js"],
+  bundle: true,
+  platform: "browser",
+  external: ["node:crypto"],
+  write: false,
+});
+const source = new TextDecoder().decode(result.outputFiles[0].contents);
+
+// https://playwright.dev/docs/network#modify-requests
+
+test("browser", async ({ page }) => {
+  // Patch the API endpoint to work around CORS for now.
+  await page.route(
+    "https://api.replicate.com/v1/predictions",
+    async (route) => {
+      // Fetch original response.
+      const response = await route.fetch();
+      // Add a prefix to the title.
+      return route.fulfill({ response });
+    }
+  );
+
+  await page.addScriptTag({ content: source });
+  const result = await page.evaluate(
+    (token) => window.main(token),
+    [process.env.REPLICATE_API_TOKEN]
+  );
+  expect(result).toBe("hello there, Betty Browser");
+});

integration/browser/package.json

@@ -0,0 +1,19 @@
+{
+  "name": "replicate-app-browser",
+  "private": true,
+  "version": "0.0.0",
+  "description": "",
+  "main": "index.js",
+  "type": "module",
+  "scripts": {
+    "test": "node index.test.js"
+  },
+  "license": "ISC",
+  "dependencies": {
+    "replicate": "../../"
+  },
+  "devDependencies": {
+    "@playwright/test": "^1.42.1",
+    "esbuild": "^0.20.1"
+  }
+}

integration/browser/playwright.config.ts

@@ -0,0 +1,3 @@
+import { defineConfig } from "@playwright/test";
+
+export default defineConfig({});