update rest to use DataConnectResponseWithMaxAge instead of DataConnectResponse

stephenarosaj · stephenarosaj · commit b99ba7256980 · 2026-03-03T15:02:47.000-08:00
diff --git a/packages/data-connect/src/network/rest/restTransport.ts b/packages/data-connect/src/network/rest/restTransport.ts
@@ -19,6 +19,7 @@ import {
   CallerSdkType,
   CallerSdkTypeEnum,
   DataConnectResponse,
+  DataConnectResponseWithMaxAge,
   AbstractDataConnectTransport,
   SubscribeNotificationHook
 } from '..';
@@ -60,7 +61,10 @@ export class RESTTransport extends AbstractDataConnectTransport {
   invokeQuery: <Data, Variables>(
     queryName: string,
     body?: Variables
-  ) => Promise<DataConnectResponse<Data>> = <Data, Variables = unknown>(
+  ) => Promise<DataConnectResponseWithMaxAge<Data>> = <
+    Data,
+    Variables = unknown
+  >(
     queryName: string,
     body: Variables
   ) => {
diff --git a/packages/data-connect/test/unit/transportManager.test.ts b/packages/data-connect/test/unit/transportManager.test.ts
@@ -23,6 +23,7 @@ import sinonChai from 'sinon-chai';
 import { DataConnectOptions } from '../../src/api/DataConnect';
 import {
   DataConnectResponse,
+  DataConnectResponseWithMaxAge,
   SubscribeNotificationHook
 } from '../../src/network';
 import { DataConnectTransportManager } from '../../src/network/manager';
@@ -61,7 +62,7 @@ describe('DataConnectTransportManager', () => {
         (manager as unknown as { restTransport: RESTTransport }).restTransport,
         'invokeQuery'
       )
-      .resolves({ data: testData } as DataConnectResponse<TestData>);
+      .resolves({ data: testData } as DataConnectResponseWithMaxAge<TestData>);
 
     await manager.invokeQuery<TestData, TestVariables>(
       'testQuery',
