chore: Add system tests for agent engines module

PiperOrigin-RevId: 895367197

Author: sararob

system_test/agent_engine_e2e_test.ts

@@ -9,10 +9,29 @@ import {Client} from '../src/genai/client';
 const PROJECT = process.env['GCLOUD_PROJECT'];
 const LOCATION = 'us-central1';
 
+async function pollOperation(client: Client, operationName: string) {
+  let isDone = false;
+  let pollCount = 0;
+  while (!isDone && pollCount < 40) {
+    const status = await client.agentEnginesInternal.getAgentOperationInternal({
+      operationName,
+    });
+    if (status.done) {
+      return status;
+    }
+    pollCount++;
+    await new Promise((resolve) => setTimeout(resolve, 5000));
+  }
+  throw new Error(
+      `Operation ${operationName} did not complete within the timeout.`,
+  );
+}
+
 describe('agentEnginesInternal', () => {
   let client: Client;
   let agentEngineName: string|undefined;
-  let operationName: string|undefined;
+  let createOperationName: string|undefined;
+  let updateOperationName: string|undefined;
 
   beforeEach(() => {
     jasmine.DEFAULT_TIMEOUT_INTERVAL = 300000;
@@ -25,11 +44,11 @@ describe('agentEnginesInternal', () => {
   afterAll(async () => {
     // If the test failed during the create operation, try to recover
     // the resource name from the operation to cleanup.
-    if (!agentEngineName && operationName) {
+    if (!agentEngineName && createOperationName) {
       try {
         const status =
             await client.agentEnginesInternal.getAgentOperationInternal({
-              operationName,
+              operationName: createOperationName,
             });
         if (status.done && status.response?.name) {
           agentEngineName = status.response.name;
@@ -60,25 +79,42 @@ describe('agentEnginesInternal', () => {
     });
 
     expect(createOp.name).toBeDefined();
-    operationName = createOp.name;
-
-    // Poll for operation completion to get the Agent Engine resource name.
-    let isDone = false;
-    let pollCount = 0;
-    while (!isDone && pollCount < 40) {
-      const status =
-          await client.agentEnginesInternal.getAgentOperationInternal({
-            operationName: operationName!,
-          });
-      if (status.done) {
-        isDone = true;
-        agentEngineName = status.response?.name;
-      } else {
-        pollCount++;
-        await new Promise((resolve) => setTimeout(resolve, 5000));
-      }
-    }
+    createOperationName = createOp.name;
+
+    const status = await pollOperation(client, createOperationName!);
+    agentEngineName = status.response?.name;
 
     expect(agentEngineName).toBeDefined();
   });
+
+  it('should get an Agent Engine', async () => {
+    const agentEngine = await client.agentEnginesInternal.getInternal({
+      name: agentEngineName!,
+    });
+    expect(agentEngine.name).toBeDefined();
+    expect(agentEngine.displayName).toEqual('test-agent-engine-sample');
+    expect(agentEngine.labels).toEqual({'test-label': 'test-value'});
+  });
+
+  it('should update an Agent Engine', async () => {
+    const updateOp = await client.agentEnginesInternal.updateInternal({
+      name: agentEngineName!,
+      config: {
+        displayName: 'test-agent-engine-updated',
+        description: 'Updated from the Vertex JS SDK system tests',
+      },
+    });
+
+    expect(updateOp.name).toBeDefined();
+    updateOperationName = updateOp.name;
+
+    const status = await pollOperation(client, updateOperationName!);
+
+    if (status.response) {
+      expect(status.response.displayName)
+          .toEqual(
+              'test-agent-engine-updated',
+          );
+    }
+  });
 });