Add support for selecting devices when running automated tests

src/mcp/tools/apptesting/devices.ts

@@ -0,0 +1,26 @@
+import { z } from "zod";
+import { tool } from "../../tool";
+import { google } from "googleapis";
+import { toContent } from "../../util";
+import { getAccessToken } from "../../../apiv2";
+
+export const get_devices = tool(
+  "apptesting",
+  {
+    name: "get_devices",
+    description:
+      "Get available devices that can be used for automated tests using the app testing agent",
+    inputSchema: z.object({
+      type: z.enum(["ANDROID"]).describe("The type of device"),
+    }),
+  },
+  async ({ type }) => {
+    const testing = google.testing("v1");
+    return toContent(
+      await testing.testEnvironmentCatalog.get({
+        oauth_token: await getAccessToken(),
+        environmentType: type,
+      }),
+    );
+  },
+);

src/mcp/tools/apptesting/index.ts

@@ -1,4 +1,5 @@
 import type { ServerTool } from "../../tool";
+import { get_devices } from "./devices";
 import { check_test, run_tests } from "./tests";
 
-export const apptestingTools: ServerTool[] = [run_tests, check_test];
+export const apptestingTools: ServerTool[] = [run_tests, check_test, get_devices];