test: add tests for locateElement

src/createActions.ts

@@ -8,7 +8,7 @@ import { z } from "zod";
 
 export const createActions = (
   page: Page
-): Record<string, RunnableFunctionWithoutParse | RunnableFunctionWithParse<any>> => {
+): Record<string, RunnableFunctionWithParse<any>> => {
   const locatorMap = new Map();
 
   const getLocator = (elementId: string) => {
@@ -23,10 +23,7 @@ export const createActions = (
 
   return {
     locateElement: {
-      function: async (args: {
-        attributeName: string;
-        cssSelector: string;
-      }) => {
+      function: async (args: { cssSelector: string }) => {
         const locator = await page.locator(args.cssSelector);
 
         const elementId = randomUUID();
@@ -606,6 +603,9 @@ export const createActions = (
       function: () => {
         return null;
       },
+      parse: (args: string) => {
+        return z.object({}).parse(JSON.parse(args));
+      },
       description:
         "This function is called at the end when the initial instructions asked to perform an action.",
       name: "resultAction",

tests/actions.spec.ts

@@ -0,0 +1,24 @@
+import { expect, test } from "@playwright/test";
+import { createActions } from "../src/createActions";
+import { ChatCompletionRunner } from "openai/lib/ChatCompletionRunner";
+
+const runner = {} as ChatCompletionRunner;
+
+test("finds element using a CSS locator and returns elementId", async ({
+  page,
+}) => {
+  await page.goto("/");
+
+  const actions = createActions(page);
+
+  const result = await actions.locateElement.function(
+    {
+      cssSelector: "h1",
+    },
+    runner
+  );
+
+  expect(result).toStrictEqual({
+    elementId: expect.any(String),
+  });
+});