@osdk/client 2.2.0-beta.6 → 2.2.0-beta.8

package/build/browser/observable/internal/testUtils.js.map

@@ -1 +1 @@
-{"version":3,"file":"testUtils.js","names":["Chalk","pDefer","invariant","afterEach","beforeEach","expect","vi","vitest","additionalContext","chalk","mockLog","args","console","log","yellow","createTestLogger","bindings","options","colors","debug","cyan","bgCyan","info","green","bgGreen","trace","gray","bgGray","error","red","bgRed","warn","bgYellow","fatal","redBright","bgRedBright","createLogMethod","name","fn","hasData","obj","more","slice","msgPrefix","methodName","magenta","Object","keys","length","child","theseBindings","theseOptions","level","undefined","isLevelEnabled","createClientMockHelper","client","logger","mockImplementation","localLogger","Error","e","where","deadObjectSet","whereArgs","fetchPage","fetchArgs","fetchOne","baseUrl","ontologyRid","objectFactory","objectFactory2","ontologyProvider","getActionDefinition","getInterfaceDefinition","getObjectDefinition","getQueryDefinition","tokenProvider","objectSetFactory","fetch","clientCacheKey","requestContext","mockApplyActionOnce","d","mockReturnValueOnce","applyAction","x","promise","type","addedLinks","addedObjects","deletedLinksCount","deletedObjectsCount","editedObjectTypes","modifiedObjects","mockFetchOneOnce","expectedId","a","process","env","NODE_ENV","r","$primaryKey","mockObjectFactory2Once","mocked","mockFetchPageOnce","objectSet","fetchPageArgs","clause","createDefer","subscriptions","s","unsubscribe","push","expectSingleListCallAndClear","subFn","resolvedList","payloadOptions","isFakeTimers","runOnlyPendingTimers","toHaveBeenCalledExactlyOnceWith","listPayloadContaining","mockClear","expectSingleObjectCallAndClear","object","status","objectContaining","any","String","waitForCall","times","waitFor","toHaveBeenCalledTimes","createSubscriptionHelper","mockSingleSubCallback","ret","assign","expectLoadingAndLoaded","q","toHaveBeenNthCalledWith","loading","loaded","mockListSubCallback","cacheEntryContaining","cacheKey","value","toBeOneOf","anything","lastUpdated","nonOptionalValue","src","key","objectPayloadContaining","isOptimistic","Boolean","fetchMore","Function","hasMore","applyCustomMatchers","extend","toBeGreaterThan","pass","message"],"sources":["testUtils.ts"],"sourcesContent":["/*\n * Copyright 2025 Palantir Technologies, Inc. All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport type {\n  ActionDefinition,\n  ActionEditResponse,\n  ObjectOrInterfaceDefinition,\n  ObjectSet,\n  ObjectTypeDefinition,\n  Osdk,\n  OsdkBase,\n  PageResult,\n  WhereClause,\n} from \"@osdk/api\";\nimport { Chalk } from \"chalk\";\nimport type { DeferredPromise } from \"p-defer\";\nimport pDefer from \"p-defer\";\nimport invariant from \"tiny-invariant\";\nimport type { Mock } from \"vitest\";\nimport { afterEach, beforeEach, expect, vi, vitest } from \"vitest\";\nimport type { ActionSignatureFromDef } from \"../../actions/applyAction.js\";\nimport type { Client } from \"../../Client.js\";\nimport { additionalContext } from \"../../Client.js\";\nimport type { LogFn, Logger } from \"../../Logger.js\";\nimport type { ListPayload } from \"../ListPayload.js\";\nimport type { ObjectPayload } from \"../ObjectPayload.js\";\nimport type { Status, Unsubscribable } from \"../ObservableClient.js\";\nimport type { Entry } from \"./Layer.js\";\n\nconst chalk = new Chalk(); // new Chalk({ level: 3 });\n\nexport interface MockClientHelper {\n  client: Mock<Client> & Client;\n\n  mockApplyActionOnce: () => DeferredPromise<Partial<ActionEditResponse>>;\n\n  mockFetchOneOnce: <Q extends ObjectTypeDefinition>(\n    expectedId?: string | number | boolean,\n  ) => DeferredPromise<\n    Partial<\n      Osdk.Instance<Q>\n    >\n  >;\n\n  mockObjectFactory2Once: () => DeferredPromise<\n    Osdk.Instance<ObjectOrInterfaceDefinition, never, any, {}>[]\n  >;\n\n  mockFetchPageOnce: <\n    Q extends ObjectTypeDefinition,\n  >(\n    expected?: {\n      where: WhereClause<any>;\n      orderBy: Record<string, \"asc\" | \"desc\" | undefined>;\n    },\n  ) => DeferredPromise<\n    PageResult<Partial<Osdk.Instance<Q>>>\n  >;\n}\n\nfunction mockLog(...args: any[]) {\n  // eslint-disable-next-line no-console\n  console.log(\n    chalk.yellow(\"mockClient\"),\n    ...args,\n  );\n}\n// interface LogFn {\n//   (obj: unknown, msg?: string, ...args: any[]): void;\n//   (msg: string, ...args: any[]): void;\n// }\n\nexport function createTestLogger(\n  bindings: Record<string, any>,\n  options?: { level?: string; msgPrefix?: string },\n): Logger {\n  const colors = {\n    debug: [chalk.cyan, chalk.bgCyan],\n    info: [chalk.green, chalk.bgGreen],\n    trace: [chalk.gray, chalk.bgGray],\n    error: [chalk.red, chalk.bgRed],\n    warn: [chalk.yellow, chalk.bgYellow],\n    fatal: [chalk.redBright, chalk.bgRedBright],\n  } as const;\n  function createLogMethod(\n    name: \"debug\" | \"error\" | \"info\" | \"warn\" | \"fatal\" | \"trace\",\n  ) {\n    return vi.fn<LogFn>(\n      (\n        ...args: [\n          obj: unknown,\n          ...args1: any[],\n        ] | [\n          ...args2: any[],\n        ]\n      ) => {\n        const hasData = typeof args[0] !== \"string\";\n        const obj: Record<string, unknown> = hasData ? args[0] as any : {};\n        const more: any[] = hasData ? args.slice(1) : args.slice(0);\n\n        // eslint-disable-next-line no-console\n        console.log(\n          `${colors[name][1](name)}${\n            options?.msgPrefix ? \" \" + colors[name][0](options.msgPrefix) : \"\"\n          }${obj.methodName ? ` .${chalk.magenta(obj.methodName)}()` : \"\"}`,\n          ...more,\n        );\n        if (bindings && Object.keys(bindings).length > 0) {\n          // eslint-disable-next-line no-console\n          console.log(bindings);\n        }\n      },\n    ) as LogFn;\n  }\n  return {\n    debug: createLogMethod(\"debug\"),\n    error: createLogMethod(\"error\"),\n    info: createLogMethod(\"info\"),\n    warn: createLogMethod(\"warn\"),\n    fatal: createLogMethod(\"fatal\"),\n    child: vi.fn<Logger[\"child\"]>((theseBindings, theseOptions) =>\n      createTestLogger({\n        ...bindings,\n        ...theseBindings,\n      }, {\n        level: (theseOptions ?? options)?.level,\n        msgPrefix: options?.msgPrefix || theseOptions?.msgPrefix\n          ? `${options?.msgPrefix ? `${options.msgPrefix} ` : \"\"}${\n            theseOptions?.msgPrefix || \"\"\n          }`\n          : undefined,\n      })\n    ),\n    trace: createLogMethod(\"trace\"),\n    isLevelEnabled: vi.fn((args) => true),\n  };\n}\n\nexport function createClientMockHelper(): MockClientHelper {\n  const client = vitest.fn<typeof client>() as unknown as Mock<Client> & Client;\n\n  const logger = createTestLogger({});\n\n  // this is just a fallback for when there is nothing set so we can track whats up\n  client.mockImplementation((...args: any[]) => {\n    const localLogger = logger.child({}, { msgPrefix: \"fallback mock client\" });\n    try {\n      throw new Error(\"IN THE FALLBACK MOCK CLIENT IMPLEMENTATION\");\n    } catch (e) {\n      localLogger.error(\"Just for the stack trace\", e);\n    }\n\n    let where;\n    const deadObjectSet = {\n      where: (...whereArgs: any[]) => {\n        localLogger.trace(\"where\", whereArgs);\n        where = whereArgs;\n        return deadObjectSet;\n      },\n      fetchPage: (...fetchArgs: any[]) => {\n        localLogger.trace(\"etchPage\", where!, fetchArgs);\n        throw new Error(\"NO\");\n      },\n      fetchOne: (...fetchArgs: any[]) => {\n        localLogger.trace(\"fetchOne\", fetchArgs);\n\n        throw new Error(\"NO\");\n      },\n    };\n    return deadObjectSet;\n  });\n\n  client[additionalContext] = {\n    baseUrl: \"http://localhost:8080\",\n    ontologyRid: \"ri.something\",\n    objectFactory: vitest.fn(),\n    objectFactory2: vitest.fn(),\n    ontologyProvider: {\n      getActionDefinition: vitest.fn(),\n      getInterfaceDefinition: vitest.fn(),\n      getObjectDefinition: vitest.fn(),\n      getQueryDefinition: vitest.fn(),\n    },\n    tokenProvider: vitest.fn(),\n    objectSetFactory: vitest.fn(),\n    fetch: vitest.fn(),\n    clientCacheKey: {} as any,\n    requestContext: {},\n    logger,\n  };\n\n  function mockObjectFactory2Once() {\n    const d = pDefer<\n      Osdk.Instance<ObjectOrInterfaceDefinition, never, any, {}>[]\n    >();\n    vi.mocked(client[additionalContext].objectFactory2).mockReturnValueOnce(\n      d.promise,\n    );\n    return d;\n  }\n\n  function mockFetchPageOnce<\n    X extends PageResult<\n      Osdk.Instance<ObjectTypeDefinition>\n    >,\n  >(): DeferredPromise<X> {\n    const d = pDefer<X>();\n\n    const objectSet: ObjectSet<ObjectTypeDefinition> = {\n      fetchPage: async (fetchPageArgs) => {\n        mockLog(\"fetchPage\", fetchPageArgs);\n        const r = await d.promise;\n        return { ...r, $primaryKey: fetchPageArgs };\n      },\n      where: (clause) => {\n        mockLog(\"where\", clause);\n        return objectSet;\n      },\n    } as Pick<\n      ObjectSet<ObjectTypeDefinition>,\n      \"fetchPage\" | \"where\"\n    > as ObjectSet<ObjectTypeDefinition>;\n\n    client.mockReturnValueOnce(objectSet);\n    return d;\n  }\n\n  function mockFetchOneOnce<\n    X extends Partial<OsdkBase<any>>,\n  >(expectedId?: string | number | boolean): DeferredPromise<X> {\n    const d = pDefer<X>();\n\n    client.mockReturnValueOnce(\n      {\n        fetchOne: async (a) => {\n          mockLog(\"fetchOne\", a);\n          invariant(\n            expectedId === undefined || a === expectedId,\n            \"expected id to match\",\n          );\n          const r = await d.promise;\n          return { ...r, $primaryKey: a };\n        },\n      } as Pick<ObjectSet<ObjectTypeDefinition>, \"fetchOne\">,\n    );\n    return d;\n  }\n\n  function mockApplyActionOnce(): DeferredPromise<\n    Partial<ActionEditResponse>\n  > {\n    const d = pDefer<Partial<ActionEditResponse>>();\n\n    client.mockReturnValueOnce(\n      {\n        applyAction: async (_args): Promise<ActionEditResponse> => {\n          const x = await d.promise;\n          return {\n            type: \"edits\",\n            addedLinks: x.addedLinks ?? [],\n            addedObjects: x.addedObjects ?? [],\n            deletedLinksCount: x.deletedLinksCount ?? 0,\n            deletedObjectsCount: x.deletedObjectsCount ?? 0,\n            editedObjectTypes: x.editedObjectTypes ?? [],\n            modifiedObjects: x.modifiedObjects ?? [],\n          };\n        },\n      } as Pick<\n        ActionSignatureFromDef<ActionDefinition>,\n        \"applyAction\"\n      >,\n    );\n    return d;\n  }\n\n  return {\n    client,\n    mockApplyActionOnce,\n    mockFetchOneOnce,\n    mockObjectFactory2Once,\n    mockFetchPageOnce,\n  };\n}\n\nexport function createDefer() {\n  let subscriptions: Unsubscribable[];\n\n  beforeEach(() => {\n    subscriptions = [];\n  });\n\n  afterEach(() => {\n    for (const s of subscriptions) {\n      s.unsubscribe();\n    }\n    subscriptions = [];\n  });\n\n  return function defer(x: Unsubscribable): Unsubscribable {\n    subscriptions.push(x);\n    return x;\n  };\n}\n\nexport function expectSingleListCallAndClear<T extends ObjectTypeDefinition>(\n  subFn: Mock<(e: ListPayload | undefined) => void>,\n  resolvedList: Osdk.Instance<T>[],\n  payloadOptions: Omit<Partial<ListPayload>, \"resolvedList\"> = {},\n): void {\n  if (vitest.isFakeTimers()) {\n    vitest.runOnlyPendingTimers();\n  }\n  expect(subFn).toHaveBeenCalledExactlyOnceWith(\n    listPayloadContaining({\n      ...payloadOptions,\n      resolvedList,\n    }),\n  );\n  subFn.mockClear();\n}\n\nexport function expectSingleObjectCallAndClear<T extends ObjectTypeDefinition>(\n  subFn: Mock<(e: ObjectPayload | undefined) => void>,\n  object: Osdk.Instance<T>,\n  status?: Status,\n): void {\n  expect(subFn).toHaveBeenCalledExactlyOnceWith(\n    expect.objectContaining({\n      object,\n      status: status ?? expect.any(String),\n    }),\n  );\n  subFn.mockClear();\n}\n\nexport async function waitForCall(\n  subFn: Mock<(e: any) => void>,\n  times: number = 1,\n): Promise<void> {\n  try {\n    await vi.waitFor(() => {\n      expect(subFn).toHaveBeenCalledTimes(times);\n    });\n  } catch (e) {\n    // we don't need the error, it will retrigger on the next line\n    // and that provides better behavior in the vitest vscode\n    // plugin. This places the error in the test itself instead of\n    // only in this file\n  }\n  expect(subFn).toHaveBeenCalledTimes(times);\n}\n\nfunction createSubscriptionHelper() {\n}\n\nexport function mockSingleSubCallback():\n  & Mock<\n    (e: ObjectPayload | undefined) => void\n  >\n  & {\n    // expectLoaded: (value: unknown) => Promise<void>;\n    // expectLoading: (value: unknown) => Promise<void>;\n    expectLoadingAndLoaded: (q: {\n      loading?: unknown;\n      loaded: unknown;\n    }) => Promise<void>;\n  }\n{\n  const ret = vitest.fn((e: ObjectPayload | undefined) => {});\n\n  //   async function expectLoaded(value: unknown) {\n  //     await waitForCall(ret);\n  //     // as long as we get the loaded call we are happy\n  //     expect(ret).toHaveBeenLastCalledWith(\n  //       cacheEntryContaining({\n  //         value,\n  //         status: \"loaded\",\n  //       }),\n  //     );\n  //     ret.mockClear();\n  //   }\n\n  //   async function expectLoading(value: unknown) {\n  //     await waitForCall(ret);\n  //     // as long as we get the loaded call we are happy\n  //     expect(ret).toHaveBeenCalledExactlyOnceWith(\n  //       cacheEntryContaining({\n  //         value,\n  //         status: \"loading\",\n  //       }),\n  //     );\n  //     ret.mockClear();\n  //   }\n\n  return Object.assign(ret, {\n    // expectLoaded,\n    // expectLoading,\n    expectLoadingAndLoaded: async (\n      q: { loading?: unknown; loaded: unknown },\n    ) => {\n      await waitForCall(ret, 2);\n\n      // as long as we get the loaded call we are happy\n      expect(ret).toHaveBeenNthCalledWith(\n        1,\n        q.loading,\n      );\n      expect(ret).toHaveBeenNthCalledWith(\n        2,\n        q.loaded,\n      );\n      expect(ret).toHaveBeenCalledTimes(2);\n      ret.mockClear();\n    },\n  });\n}\n\nexport function mockListSubCallback(): Mock<\n  (x: ListPayload | undefined) => void\n> {\n  return vitest.fn(\n    (x: ListPayload | undefined) => {},\n  );\n}\n\nexport function cacheEntryContaining(x: Partial<Entry<any>>): Entry<any> {\n  return {\n    cacheKey: x.cacheKey ?? expect.any(Object),\n    value: \"value\" in x\n      ? x.value\n      : expect.toBeOneOf([expect.anything(), undefined]),\n    status: x.status ?? expect.anything(),\n    lastUpdated: x.lastUpdated ?? expect.anything(),\n  };\n}\n\nfunction nonOptionalValue<T extends object, K extends keyof T>(\n  src: T,\n  key: K,\n): NonNullable<T[K]> {\n  return key in src\n    ? src[key]\n    : expect.toBeOneOf([expect.anything(), undefined]);\n}\n\nexport function objectPayloadContaining(\n  x: Partial<ObjectPayload>,\n): ObjectPayload {\n  return {\n    object: nonOptionalValue(x, \"object\"),\n    isOptimistic: expect.any(Boolean),\n    status: x.status ?? expect.anything(),\n    lastUpdated: x.lastUpdated ?? expect.anything(),\n  };\n}\n\nexport function listPayloadContaining(\n  x: Partial<ListPayload>,\n): ListPayload {\n  return {\n    fetchMore: x.fetchMore ?? expect.any(Function),\n    hasMore: x.hasMore ?? expect.any(Boolean),\n    resolvedList: x.resolvedList ?? expect.anything(),\n    isOptimistic: expect.any(Boolean),\n    status: x.status ?? expect.anything(),\n    lastUpdated: x.lastUpdated ?? expect.anything(),\n  };\n}\n\nexport function applyCustomMatchers(): void {\n  expect.extend({\n    toBeGreaterThan: (r: number, e: number) => {\n      return {\n        pass: r > e,\n        message: () => `expected ${r} to be greater than ${e} (lastUpdated)`,\n      };\n    },\n  });\n}\n\ninterface CustomMatchers<R = any> {\n  toBeGreaterThan: (n: number) => R;\n}\n\ninterface CustomAsymmetricMatchers<R = any> {\n  toBeGreaterThan: (n: number) => R;\n}\n\ndeclare module \"vitest\" {\n  interface Assertion<T = any> extends CustomMatchers<T> {}\n  interface AsymmetricMatchersContaining extends CustomAsymmetricMatchers {}\n}\n"],"mappings":"AAAA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;;AAaA,SAASA,KAAK,QAAQ,OAAO;AAE7B,OAAOC,MAAM,MAAM,SAAS;AAC5B,OAAOC,SAAS,MAAM,gBAAgB;AAEtC,SAASC,SAAS,EAAEC,UAAU,EAAEC,MAAM,EAAEC,EAAE,EAAEC,MAAM,QAAQ,QAAQ;AAGlE,SAASC,iBAAiB,QAAQ,iBAAiB;AAOnD,MAAMC,KAAK,GAAG,IAAIT,KAAK,CAAC,CAAC,CAAC,CAAC;;AA+B3B,SAASU,OAAOA,CAAC,GAAGC,IAAW,EAAE;EAC/B;EACAC,OAAO,CAACC,GAAG,CACTJ,KAAK,CAACK,MAAM,CAAC,YAAY,CAAC,EAC1B,GAAGH,IACL,CAAC;AACH;AACA;AACA;AACA;AACA;;AAEA,OAAO,SAASI,gBAAgBA,CAC9BC,QAA6B,EAC7BC,OAAgD,EACxC;EACR,MAAMC,MAAM,GAAG;IACbC,KAAK,EAAE,CAACV,KAAK,CAACW,IAAI,EAAEX,KAAK,CAACY,MAAM,CAAC;IACjCC,IAAI,EAAE,CAACb,KAAK,CAACc,KAAK,EAAEd,KAAK,CAACe,OAAO,CAAC;IAClCC,KAAK,EAAE,CAAChB,KAAK,CAACiB,IAAI,EAAEjB,KAAK,CAACkB,MAAM,CAAC;IACjCC,KAAK,EAAE,CAACnB,KAAK,CAACoB,GAAG,EAAEpB,KAAK,CAACqB,KAAK,CAAC;IAC/BC,IAAI,EAAE,CAACtB,KAAK,CAACK,MAAM,EAAEL,KAAK,CAACuB,QAAQ,CAAC;IACpCC,KAAK,EAAE,CAACxB,KAAK,CAACyB,SAAS,EAAEzB,KAAK,CAAC0B,WAAW;EAC5C,CAAU;EACV,SAASC,eAAeA,CACtBC,IAA6D,EAC7D;IACA,OAAO/B,EAAE,CAACgC,EAAE,CACV,CACE,GAAG3B,IAKF,KACE;MACH,MAAM4B,OAAO,GAAG,OAAO5B,IAAI,CAAC,CAAC,CAAC,KAAK,QAAQ;MAC3C,MAAM6B,GAA4B,GAAGD,OAAO,GAAG5B,IAAI,CAAC,CAAC,CAAC,GAAU,CAAC,CAAC;MAClE,MAAM8B,IAAW,GAAGF,OAAO,GAAG5B,IAAI,CAAC+B,KAAK,CAAC,CAAC,CAAC,GAAG/B,IAAI,CAAC+B,KAAK,CAAC,CAAC,CAAC;;MAE3D;MACA9B,OAAO,CAACC,GAAG,CACT,GAAGK,MAAM,CAACmB,IAAI,CAAC,CAAC,CAAC,CAAC,CAACA,IAAI,CAAC,GACtBpB,OAAO,EAAE0B,SAAS,GAAG,GAAG,GAAGzB,MAAM,CAACmB,IAAI,CAAC,CAAC,CAAC,CAAC,CAACpB,OAAO,CAAC0B,SAAS,CAAC,GAAG,EAAE,GACjEH,GAAG,CAACI,UAAU,GAAG,KAAKnC,KAAK,CAACoC,OAAO,CAACL,GAAG,CAACI,UAAU,CAAC,IAAI,GAAG,EAAE,EAAE,EACjE,GAAGH,IACL,CAAC;MACD,IAAIzB,QAAQ,IAAI8B,MAAM,CAACC,IAAI,CAAC/B,QAAQ,CAAC,CAACgC,MAAM,GAAG,CAAC,EAAE;QAChD;QACApC,OAAO,CAACC,GAAG,CAACG,QAAQ,CAAC;MACvB;IACF,CACF,CAAC;EACH;EACA,OAAO;IACLG,KAAK,EAAEiB,eAAe,CAAC,OAAO,CAAC;IAC/BR,KAAK,EAAEQ,eAAe,CAAC,OAAO,CAAC;IAC/Bd,IAAI,EAAEc,eAAe,CAAC,MAAM,CAAC;IAC7BL,IAAI,EAAEK,eAAe,CAAC,MAAM,CAAC;IAC7BH,KAAK,EAAEG,eAAe,CAAC,OAAO,CAAC;IAC/Ba,KAAK,EAAE3C,EAAE,CAACgC,EAAE,CAAkB,CAACY,aAAa,EAAEC,YAAY,KACxDpC,gBAAgB,CAAC;MACf,GAAGC,QAAQ;MACX,GAAGkC;IACL,CAAC,EAAE;MACDE,KAAK,EAAE,CAACD,YAAY,IAAIlC,OAAO,GAAGmC,KAAK;MACvCT,SAAS,EAAE1B,OAAO,EAAE0B,SAAS,IAAIQ,YAAY,EAAER,SAAS,GACpD,GAAG1B,OAAO,EAAE0B,SAAS,GAAG,GAAG1B,OAAO,CAAC0B,SAAS,GAAG,GAAG,EAAE,GACpDQ,YAAY,EAAER,SAAS,IAAI,EAAE,EAC7B,GACAU;IACN,CAAC,CACH,CAAC;IACD5B,KAAK,EAAEW,eAAe,CAAC,OAAO,CAAC;IAC/BkB,cAAc,EAAEhD,EAAE,CAACgC,EAAE,CAAC,MAAU,IAAI;EACtC,CAAC;AACH;AAEA,OAAO,SAASiB,sBAAsBA,CAAA,EAAqB;EACzD,MAAMC,MAAM,GAAGjD,MAAM,CAAC+B,EAAE,CAAgB,CAAqC;EAE7E,MAAMmB,MAAM,GAAG1C,gBAAgB,CAAC,CAAC,CAAC,CAAC;;EAEnC;EACAyC,MAAM,CAACE,kBAAkB,CAAC,CAAC,GAAG/C,IAAW,KAAK;IAC5C,MAAMgD,WAAW,GAAGF,MAAM,CAACR,KAAK,CAAC,CAAC,CAAC,EAAE;MAAEN,SAAS,EAAE;IAAuB,CAAC,CAAC;IAC3E,IAAI;MACF,MAAM,IAAIiB,KAAK,CAAC,4CAA4C,CAAC;IAC/D,CAAC,CAAC,OAAOC,CAAC,EAAE;MACVF,WAAW,CAAC/B,KAAK,CAAC,0BAA0B,EAAEiC,CAAC,CAAC;IAClD;IAEA,IAAIC,KAAK;IACT,MAAMC,aAAa,GAAG;MACpBD,KAAK,EAAEA,CAAC,GAAGE,SAAgB,KAAK;QAC9BL,WAAW,CAAClC,KAAK,CAAC,OAAO,EAAEuC,SAAS,CAAC;QACrCF,KAAK,GAAGE,SAAS;QACjB,OAAOD,aAAa;MACtB,CAAC;MACDE,SAAS,EAAEA,CAAC,GAAGC,SAAgB,KAAK;QAClCP,WAAW,CAAClC,KAAK,CAAC,UAAU,EAAEqC,KAAK,EAAGI,SAAS,CAAC;QAChD,MAAM,IAAIN,KAAK,CAAC,IAAI,CAAC;MACvB,CAAC;MACDO,QAAQ,EAAEA,CAAC,GAAGD,SAAgB,KAAK;QACjCP,WAAW,CAAClC,KAAK,CAAC,UAAU,EAAEyC,SAAS,CAAC;QAExC,MAAM,IAAIN,KAAK,CAAC,IAAI,CAAC;MACvB;IACF,CAAC;IACD,OAAOG,aAAa;EACtB,CAAC,CAAC;EAEFP,MAAM,CAAChD,iBAAiB,CAAC,GAAG;IAC1B4D,OAAO,EAAE,uBAAuB;IAChCC,WAAW,EAAE,cAAc;IAC3BC,aAAa,EAAE/D,MAAM,CAAC+B,EAAE,CAAC,CAAC;IAC1BiC,cAAc,EAAEhE,MAAM,CAAC+B,EAAE,CAAC,CAAC;IAC3BkC,gBAAgB,EAAE;MAChBC,mBAAmB,EAAElE,MAAM,CAAC+B,EAAE,CAAC,CAAC;MAChCoC,sBAAsB,EAAEnE,MAAM,CAAC+B,EAAE,CAAC,CAAC;MACnCqC,mBAAmB,EAAEpE,MAAM,CAAC+B,EAAE,CAAC,CAAC;MAChCsC,kBAAkB,EAAErE,MAAM,CAAC+B,EAAE,CAAC;IAChC,CAAC;IACDuC,aAAa,EAAEtE,MAAM,CAAC+B,EAAE,CAAC,CAAC;IAC1BwC,gBAAgB,EAAEvE,MAAM,CAAC+B,EAAE,CAAC,CAAC;IAC7ByC,KAAK,EAAExE,MAAM,CAAC+B,EAAE,CAAC,CAAC;IAClB0C,cAAc,EAAE,CAAC,CAAQ;IACzBC,cAAc,EAAE,CAAC,CAAC;IAClBxB;EACF,CAAC;EAsFD,OAAO;IACLD,MAAM;IACN0B,mBAAmB,EA7BrB,SAAAA,CAAA,EAEE;MACA,MAAMC,CAAC,GAAGlF,MAAM,CAA8B,CAAC;MAE/CuD,MAAM,CAAC4B,mBAAmB,CACxB;QACEC,WAAW,EAAE,MAAAA,CAAA,KAA8C;UACzD,MAAMC,CAAC,GAAG,MAAMH,CAAC,CAACI,OAAO;UACzB,OAAO;YACLC,IAAI,EAAE,OAAO;YACbC,UAAU,EAAEH,CAAC,CAACG,UAAU,IAAI,EAAE;YAC9BC,YAAY,EAAEJ,CAAC,CAACI,YAAY,IAAI,EAAE;YAClCC,iBAAiB,EAAEL,CAAC,CAACK,iBAAiB,IAAI,CAAC;YAC3CC,mBAAmB,EAAEN,CAAC,CAACM,mBAAmB,IAAI,CAAC;YAC/CC,iBAAiB,EAAEP,CAAC,CAACO,iBAAiB,IAAI,EAAE;YAC5CC,eAAe,EAAER,CAAC,CAACQ,eAAe,IAAI;UACxC,CAAC;QACH;MACF,CAIF,CAAC;MACD,OAAOX,CAAC;IACV,CAIqB;IACnBY,gBAAgB,EAnDlB,SAAAA,CAEEC,UAAsC,EAAsB;MAC5D,MAAMb,CAAC,GAAGlF,MAAM,CAAI,CAAC;MAErBuD,MAAM,CAAC4B,mBAAmB,CACxB;QACEjB,QAAQ,EAAE,MAAO8B,CAAC,IAAK;UACrBvF,OAAO,CAAC,UAAU,EAAEuF,CAAC,CAAC;UACtB,EACED,UAAU,KAAK3C,SAAS,IAAI4C,CAAC,KAAKD,UAAU,IAAAE,OAAA,CAAAC,GAAA,CAAAC,QAAA,oBAD9ClG,SAAS,QAEP,sBAAsB,IAFxBA,SAAS;UAIT,MAAMmG,CAAC,GAAG,MAAMlB,CAAC,CAACI,OAAO;UACzB,OAAO;YAAE,GAAGc,CAAC;YAAEC,WAAW,EAAEL;UAAE,CAAC;QACjC;MACF,CACF,CAAC;MACD,OAAOd,CAAC;IACV,CAgCkB;IAChBoB,sBAAsB,EAxFxB,SAAAA,CAAA,EAAkC;MAChC,MAAMpB,CAAC,GAAGlF,MAAM,CAEd,CAAC;MACHK,EAAE,CAACkG,MAAM,CAAChD,MAAM,CAAChD,iBAAiB,CAAC,CAAC+D,cAAc,CAAC,CAACa,mBAAmB,CACrED,CAAC,CAACI,OACJ,CAAC;MACD,OAAOJ,CAAC;IACV,CAgFwB;IACtBsB,iBAAiB,EA/EnB,SAAAA,CAAA,EAIwB;MACtB,MAAMtB,CAAC,GAAGlF,MAAM,CAAI,CAAC;MAErB,MAAMyG,SAA0C,GAAG;QACjDzC,SAAS,EAAE,MAAO0C,aAAa,IAAK;UAClCjG,OAAO,CAAC,WAAW,EAAEiG,aAAa,CAAC;UACnC,MAAMN,CAAC,GAAG,MAAMlB,CAAC,CAACI,OAAO;UACzB,OAAO;YAAE,GAAGc,CAAC;YAAEC,WAAW,EAAEK;UAAc,CAAC;QAC7C,CAAC;QACD7C,KAAK,EAAG8C,MAAM,IAAK;UACjBlG,OAAO,CAAC,OAAO,EAAEkG,MAAM,CAAC;UACxB,OAAOF,SAAS;QAClB;MACF,CAGoC;MAEpClD,MAAM,CAAC4B,mBAAmB,CAACsB,SAAS,CAAC;MACrC,OAAOvB,CAAC;IACV;EAwDA,CAAC;AACH;AAEA,OAAO,SAAS0B,WAAWA,CAAA,EAAG;EAC5B,IAAIC,aAA+B;EAEnC1G,UAAU,CAAC,MAAM;IACf0G,aAAa,GAAG,EAAE;EACpB,CAAC,CAAC;EAEF3G,SAAS,CAAC,MAAM;IACd,KAAK,MAAM4G,CAAC,IAAID,aAAa,EAAE;MAC7BC,CAAC,CAACC,WAAW,CAAC,CAAC;IACjB;IACAF,aAAa,GAAG,EAAE;EACpB,CAAC,CAAC;EAEF,OAAO,UAAexB,CAAiB,EAAkB;IACvDwB,aAAa,CAACG,IAAI,CAAC3B,CAAC,CAAC;IACrB,OAAOA,CAAC;EACV,CAAC;AACH;AAEA,OAAO,SAAS4B,4BAA4BA,CAC1CC,KAAiD,EACjDC,YAAgC,EAChCC,cAA0D,GAAG,CAAC,CAAC,EACzD;EACN,IAAI9G,MAAM,CAAC+G,YAAY,CAAC,CAAC,EAAE;IACzB/G,MAAM,CAACgH,oBAAoB,CAAC,CAAC;EAC/B;EACAlH,MAAM,CAAC8G,KAAK,CAAC,CAACK,+BAA+B,CAC3CC,qBAAqB,CAAC;IACpB,GAAGJ,cAAc;IACjBD;EACF,CAAC,CACH,CAAC;EACDD,KAAK,CAACO,SAAS,CAAC,CAAC;AACnB;AAEA,OAAO,SAASC,8BAA8BA,CAC5CR,KAAmD,EACnDS,MAAwB,EACxBC,MAAe,EACT;EACNxH,MAAM,CAAC8G,KAAK,CAAC,CAACK,+BAA+B,CAC3CnH,MAAM,CAACyH,gBAAgB,CAAC;IACtBF,MAAM;IACNC,MAAM,EAAEA,MAAM,IAAIxH,MAAM,CAAC0H,GAAG,CAACC,MAAM;EACrC,CAAC,CACH,CAAC;EACDb,KAAK,CAACO,SAAS,CAAC,CAAC;AACnB;AAEA,OAAO,eAAeO,WAAWA,CAC/Bd,KAA6B,EAC7Be,KAAa,GAAG,CAAC,EACF;EACf,IAAI;IACF,MAAM5H,EAAE,CAAC6H,OAAO,CAAC,MAAM;MACrB9H,MAAM,CAAC8G,KAAK,CAAC,CAACiB,qBAAqB,CAACF,KAAK,CAAC;IAC5C,CAAC,CAAC;EACJ,CAAC,CAAC,OAAOrE,CAAC,EAAE;IACV;IACA;IACA;IACA;EAAA;EAEFxD,MAAM,CAAC8G,KAAK,CAAC,CAACiB,qBAAqB,CAACF,KAAK,CAAC;AAC5C;AAEA,SAASG,wBAAwBA,CAAA,EAAG,CACpC;AAEA,OAAO,SAASC,qBAAqBA,CAAA,EAYrC;EACE,MAAMC,GAAG,GAAGhI,MAAM,CAAC+B,EAAE,CAAC,MAAkC,CAAC,CAAC,CAAC;;EAE3D;EACA;EACA;EACA;EACA;EACA;EACA;EACA;EACA;EACA;EACA;;EAEA;EACA;EACA;EACA;EACA;EACA;EACA;EACA;EACA;EACA;EACA;;EAEA,OAAOQ,MAAM,CAAC0F,MAAM,CAACD,GAAG,EAAE;IACxB;IACA;IACAE,sBAAsB,EAAE,MACtBC,CAAyC,IACtC;MACH,MAAMT,WAAW,CAACM,GAAG,EAAE,CAAC,CAAC;;MAEzB;MACAlI,MAAM,CAACkI,GAAG,CAAC,CAACI,uBAAuB,CACjC,CAAC,EACDD,CAAC,CAACE,OACJ,CAAC;MACDvI,MAAM,CAACkI,GAAG,CAAC,CAACI,uBAAuB,CACjC,CAAC,EACDD,CAAC,CAACG,MACJ,CAAC;MACDxI,MAAM,CAACkI,GAAG,CAAC,CAACH,qBAAqB,CAAC,CAAC,CAAC;MACpCG,GAAG,CAACb,SAAS,CAAC,CAAC;IACjB;EACF,CAAC,CAAC;AACJ;AAEA,OAAO,SAASoB,mBAAmBA,CAAA,EAEjC;EACA,OAAOvI,MAAM,CAAC+B,EAAE,CACd,MAAgC,CAAC,CACnC,CAAC;AACH;AAEA,OAAO,SAASyG,oBAAoBA,CAACzD,CAAsB,EAAc;EACvE,OAAO;IACL0D,QAAQ,EAAE1D,CAAC,CAAC0D,QAAQ,IAAI3I,MAAM,CAAC0H,GAAG,CAACjF,MAAM,CAAC;IAC1CmG,KAAK,EAAE,OAAO,IAAI3D,CAAC,GACfA,CAAC,CAAC2D,KAAK,GACP5I,MAAM,CAAC6I,SAAS,CAAC,CAAC7I,MAAM,CAAC8I,QAAQ,CAAC,CAAC,EAAE9F,SAAS,CAAC,CAAC;IACpDwE,MAAM,EAAEvC,CAAC,CAACuC,MAAM,IAAIxH,MAAM,CAAC8I,QAAQ,CAAC,CAAC;IACrCC,WAAW,EAAE9D,CAAC,CAAC8D,WAAW,IAAI/I,MAAM,CAAC8I,QAAQ,CAAC;EAChD,CAAC;AACH;AAEA,SAASE,gBAAgBA,CACvBC,GAAM,EACNC,GAAM,EACa;EACnB,OAAOA,GAAG,IAAID,GAAG,GACbA,GAAG,CAACC,GAAG,CAAC,GACRlJ,MAAM,CAAC6I,SAAS,CAAC,CAAC7I,MAAM,CAAC8I,QAAQ,CAAC,CAAC,EAAE9F,SAAS,CAAC,CAAC;AACtD;AAEA,OAAO,SAASmG,uBAAuBA,CACrClE,CAAyB,EACV;EACf,OAAO;IACLsC,MAAM,EAAEyB,gBAAgB,CAAC/D,CAAC,EAAE,QAAQ,CAAC;IACrCmE,YAAY,EAAEpJ,MAAM,CAAC0H,GAAG,CAAC2B,OAAO,CAAC;IACjC7B,MAAM,EAAEvC,CAAC,CAACuC,MAAM,IAAIxH,MAAM,CAAC8I,QAAQ,CAAC,CAAC;IACrCC,WAAW,EAAE9D,CAAC,CAAC8D,WAAW,IAAI/I,MAAM,CAAC8I,QAAQ,CAAC;EAChD,CAAC;AACH;AAEA,OAAO,SAAS1B,qBAAqBA,CACnCnC,CAAuB,EACV;EACb,OAAO;IACLqE,SAAS,EAAErE,CAAC,CAACqE,SAAS,IAAItJ,MAAM,CAAC0H,GAAG,CAAC6B,QAAQ,CAAC;IAC9CC,OAAO,EAAEvE,CAAC,CAACuE,OAAO,IAAIxJ,MAAM,CAAC0H,GAAG,CAAC2B,OAAO,CAAC;IACzCtC,YAAY,EAAE9B,CAAC,CAAC8B,YAAY,IAAI/G,MAAM,CAAC8I,QAAQ,CAAC,CAAC;IACjDM,YAAY,EAAEpJ,MAAM,CAAC0H,GAAG,CAAC2B,OAAO,CAAC;IACjC7B,MAAM,EAAEvC,CAAC,CAACuC,MAAM,IAAIxH,MAAM,CAAC8I,QAAQ,CAAC,CAAC;IACrCC,WAAW,EAAE9D,CAAC,CAAC8D,WAAW,IAAI/I,MAAM,CAAC8I,QAAQ,CAAC;EAChD,CAAC;AACH;AAEA,OAAO,SAASW,mBAAmBA,CAAA,EAAS;EAC1CzJ,MAAM,CAAC0J,MAAM,CAAC;IACZC,eAAe,EAAEA,CAAC3D,CAAS,EAAExC,CAAS,KAAK;MACzC,OAAO;QACLoG,IAAI,EAAE5D,CAAC,GAAGxC,CAAC;QACXqG,OAAO,EAAEA,CAAA,KAAM,YAAY7D,CAAC,uBAAuBxC,CAAC;MACtD,CAAC;IACH;EACF,CAAC,CAAC;AACJ","ignoreList":[]}
+{"version":3,"file":"testUtils.js","names":["Chalk","pDefer","invariant","afterEach","beforeEach","expect","vi","vitest","additionalContext","chalk","mockLog","args","console","log","yellow","createTestLogger","bindings","options","colors","debug","cyan","bgCyan","info","green","bgGreen","trace","gray","bgGray","error","red","bgRed","warn","bgYellow","fatal","redBright","bgRedBright","createLogMethod","name","fn","hasData","obj","more","slice","msgPrefix","methodName","magenta","Object","keys","length","child","theseBindings","theseOptions","level","undefined","isLevelEnabled","createClientMockHelper","client","logger","mockImplementation","localLogger","Error","e","where","deadObjectSet","whereArgs","fetchPage","fetchArgs","fetchOne","baseUrl","ontologyRid","objectFactory","objectFactory2","ontologyProvider","getActionDefinition","getInterfaceDefinition","getObjectDefinition","getQueryDefinition","tokenProvider","objectSetFactory","fetch","clientCacheKey","requestContext","mockApplyActionOnce","d","mockReturnValueOnce","applyAction","x","promise","type","addedLinks","addedObjects","deletedObjects","deletedLinks","deletedLinksCount","deletedObjectsCount","editedObjectTypes","modifiedObjects","mockFetchOneOnce","expectedId","a","process","env","NODE_ENV","r","$primaryKey","mockObjectFactory2Once","mocked","mockFetchPageOnce","objectSet","fetchPageArgs","clause","createDefer","subscriptions","s","unsubscribe","push","expectSingleListCallAndClear","subFn","resolvedList","payloadOptions","isFakeTimers","runOnlyPendingTimers","toHaveBeenCalledExactlyOnceWith","listPayloadContaining","mockClear","expectSingleObjectCallAndClear","object","status","objectContaining","any","String","waitForCall","times","waitFor","toHaveBeenCalledTimes","createSubscriptionHelper","mockSingleSubCallback","ret","assign","expectLoadingAndLoaded","q","toHaveBeenNthCalledWith","loading","loaded","mockListSubCallback","cacheEntryContaining","cacheKey","value","toBeOneOf","anything","lastUpdated","nonOptionalValue","src","key","objectPayloadContaining","isOptimistic","Boolean","fetchMore","Function","hasMore","applyCustomMatchers","extend","toBeGreaterThan","pass","message"],"sources":["testUtils.ts"],"sourcesContent":["/*\n * Copyright 2025 Palantir Technologies, Inc. All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport type {\n  ActionDefinition,\n  ActionEditResponse,\n  ObjectOrInterfaceDefinition,\n  ObjectSet,\n  ObjectTypeDefinition,\n  Osdk,\n  OsdkBase,\n  PageResult,\n  WhereClause,\n} from \"@osdk/api\";\nimport { Chalk } from \"chalk\";\nimport type { DeferredPromise } from \"p-defer\";\nimport pDefer from \"p-defer\";\nimport invariant from \"tiny-invariant\";\nimport type { Mock } from \"vitest\";\nimport { afterEach, beforeEach, expect, vi, vitest } from \"vitest\";\nimport type { ActionSignatureFromDef } from \"../../actions/applyAction.js\";\nimport type { Client } from \"../../Client.js\";\nimport { additionalContext } from \"../../Client.js\";\nimport type { LogFn, Logger } from \"../../Logger.js\";\nimport type { ListPayload } from \"../ListPayload.js\";\nimport type { ObjectPayload } from \"../ObjectPayload.js\";\nimport type { Status, Unsubscribable } from \"../ObservableClient.js\";\nimport type { Entry } from \"./Layer.js\";\n\nconst chalk = new Chalk(); // new Chalk({ level: 3 });\n\nexport interface MockClientHelper {\n  client: Mock<Client> & Client;\n\n  mockApplyActionOnce: () => DeferredPromise<Partial<ActionEditResponse>>;\n\n  mockFetchOneOnce: <Q extends ObjectTypeDefinition>(\n    expectedId?: string | number | boolean,\n  ) => DeferredPromise<\n    Partial<\n      Osdk.Instance<Q>\n    >\n  >;\n\n  mockObjectFactory2Once: () => DeferredPromise<\n    Osdk.Instance<ObjectOrInterfaceDefinition, never, any, {}>[]\n  >;\n\n  mockFetchPageOnce: <\n    Q extends ObjectTypeDefinition,\n  >(\n    expected?: {\n      where: WhereClause<any>;\n      orderBy: Record<string, \"asc\" | \"desc\" | undefined>;\n    },\n  ) => DeferredPromise<\n    PageResult<Partial<Osdk.Instance<Q>>>\n  >;\n}\n\nfunction mockLog(...args: any[]) {\n  // eslint-disable-next-line no-console\n  console.log(\n    chalk.yellow(\"mockClient\"),\n    ...args,\n  );\n}\n// interface LogFn {\n//   (obj: unknown, msg?: string, ...args: any[]): void;\n//   (msg: string, ...args: any[]): void;\n// }\n\nexport function createTestLogger(\n  bindings: Record<string, any>,\n  options?: { level?: string; msgPrefix?: string },\n): Logger {\n  const colors = {\n    debug: [chalk.cyan, chalk.bgCyan],\n    info: [chalk.green, chalk.bgGreen],\n    trace: [chalk.gray, chalk.bgGray],\n    error: [chalk.red, chalk.bgRed],\n    warn: [chalk.yellow, chalk.bgYellow],\n    fatal: [chalk.redBright, chalk.bgRedBright],\n  } as const;\n  function createLogMethod(\n    name: \"debug\" | \"error\" | \"info\" | \"warn\" | \"fatal\" | \"trace\",\n  ) {\n    return vi.fn<LogFn>(\n      (\n        ...args: [\n          obj: unknown,\n          ...args1: any[],\n        ] | [\n          ...args2: any[],\n        ]\n      ) => {\n        const hasData = typeof args[0] !== \"string\";\n        const obj: Record<string, unknown> = hasData ? args[0] as any : {};\n        const more: any[] = hasData ? args.slice(1) : args.slice(0);\n\n        // eslint-disable-next-line no-console\n        console.log(\n          `${colors[name][1](name)}${\n            options?.msgPrefix ? \" \" + colors[name][0](options.msgPrefix) : \"\"\n          }${obj.methodName ? ` .${chalk.magenta(obj.methodName)}()` : \"\"}`,\n          ...more,\n        );\n        if (bindings && Object.keys(bindings).length > 0) {\n          // eslint-disable-next-line no-console\n          console.log(bindings);\n        }\n      },\n    ) as LogFn;\n  }\n  return {\n    debug: createLogMethod(\"debug\"),\n    error: createLogMethod(\"error\"),\n    info: createLogMethod(\"info\"),\n    warn: createLogMethod(\"warn\"),\n    fatal: createLogMethod(\"fatal\"),\n    child: vi.fn<Logger[\"child\"]>((theseBindings, theseOptions) =>\n      createTestLogger({\n        ...bindings,\n        ...theseBindings,\n      }, {\n        level: (theseOptions ?? options)?.level,\n        msgPrefix: options?.msgPrefix || theseOptions?.msgPrefix\n          ? `${options?.msgPrefix ? `${options.msgPrefix} ` : \"\"}${\n            theseOptions?.msgPrefix || \"\"\n          }`\n          : undefined,\n      })\n    ),\n    trace: createLogMethod(\"trace\"),\n    isLevelEnabled: vi.fn((args) => true),\n  };\n}\n\nexport function createClientMockHelper(): MockClientHelper {\n  const client = vitest.fn<typeof client>() as unknown as Mock<Client> & Client;\n\n  const logger = createTestLogger({});\n\n  // this is just a fallback for when there is nothing set so we can track whats up\n  client.mockImplementation((...args: any[]) => {\n    const localLogger = logger.child({}, { msgPrefix: \"fallback mock client\" });\n    try {\n      throw new Error(\"IN THE FALLBACK MOCK CLIENT IMPLEMENTATION\");\n    } catch (e) {\n      localLogger.error(\"Just for the stack trace\", e);\n    }\n\n    let where;\n    const deadObjectSet = {\n      where: (...whereArgs: any[]) => {\n        localLogger.trace(\"where\", whereArgs);\n        where = whereArgs;\n        return deadObjectSet;\n      },\n      fetchPage: (...fetchArgs: any[]) => {\n        localLogger.trace(\"etchPage\", where!, fetchArgs);\n        throw new Error(\"NO\");\n      },\n      fetchOne: (...fetchArgs: any[]) => {\n        localLogger.trace(\"fetchOne\", fetchArgs);\n\n        throw new Error(\"NO\");\n      },\n    };\n    return deadObjectSet;\n  });\n\n  client[additionalContext] = {\n    baseUrl: \"http://localhost:8080\",\n    ontologyRid: \"ri.something\",\n    objectFactory: vitest.fn(),\n    objectFactory2: vitest.fn(),\n    ontologyProvider: {\n      getActionDefinition: vitest.fn(),\n      getInterfaceDefinition: vitest.fn(),\n      getObjectDefinition: vitest.fn(),\n      getQueryDefinition: vitest.fn(),\n    },\n    tokenProvider: vitest.fn(),\n    objectSetFactory: vitest.fn(),\n    fetch: vitest.fn(),\n    clientCacheKey: {} as any,\n    requestContext: {},\n    logger,\n  };\n\n  function mockObjectFactory2Once() {\n    const d = pDefer<\n      Osdk.Instance<ObjectOrInterfaceDefinition, never, any, {}>[]\n    >();\n    vi.mocked(client[additionalContext].objectFactory2).mockReturnValueOnce(\n      d.promise,\n    );\n    return d;\n  }\n\n  function mockFetchPageOnce<\n    X extends PageResult<\n      Osdk.Instance<ObjectTypeDefinition>\n    >,\n  >(): DeferredPromise<X> {\n    const d = pDefer<X>();\n\n    const objectSet: ObjectSet<ObjectTypeDefinition> = {\n      fetchPage: async (fetchPageArgs) => {\n        mockLog(\"fetchPage\", fetchPageArgs);\n        const r = await d.promise;\n        return { ...r, $primaryKey: fetchPageArgs };\n      },\n      where: (clause) => {\n        mockLog(\"where\", clause);\n        return objectSet;\n      },\n    } as Pick<\n      ObjectSet<ObjectTypeDefinition>,\n      \"fetchPage\" | \"where\"\n    > as ObjectSet<ObjectTypeDefinition>;\n\n    client.mockReturnValueOnce(objectSet);\n    return d;\n  }\n\n  function mockFetchOneOnce<\n    X extends Partial<OsdkBase<any>>,\n  >(expectedId?: string | number | boolean): DeferredPromise<X> {\n    const d = pDefer<X>();\n\n    client.mockReturnValueOnce(\n      {\n        fetchOne: async (a) => {\n          mockLog(\"fetchOne\", a);\n          invariant(\n            expectedId === undefined || a === expectedId,\n            \"expected id to match\",\n          );\n          const r = await d.promise;\n          return { ...r, $primaryKey: a };\n        },\n      } as Pick<ObjectSet<ObjectTypeDefinition>, \"fetchOne\">,\n    );\n    return d;\n  }\n\n  function mockApplyActionOnce(): DeferredPromise<\n    Partial<ActionEditResponse>\n  > {\n    const d = pDefer<Partial<ActionEditResponse>>();\n\n    client.mockReturnValueOnce(\n      {\n        applyAction: async (_args): Promise<ActionEditResponse> => {\n          const x = await d.promise;\n          return {\n            type: \"edits\",\n            addedLinks: x.addedLinks ?? [],\n            addedObjects: x.addedObjects ?? [],\n            deletedObjects: x.deletedObjects ?? [],\n            deletedLinks: x.deletedLinks ?? [],\n            deletedLinksCount: x.deletedLinksCount ?? 0,\n            deletedObjectsCount: x.deletedObjectsCount ?? 0,\n            editedObjectTypes: x.editedObjectTypes ?? [],\n            modifiedObjects: x.modifiedObjects ?? [],\n          };\n        },\n      } as Pick<\n        ActionSignatureFromDef<ActionDefinition>,\n        \"applyAction\"\n      >,\n    );\n    return d;\n  }\n\n  return {\n    client,\n    mockApplyActionOnce,\n    mockFetchOneOnce,\n    mockObjectFactory2Once,\n    mockFetchPageOnce,\n  };\n}\n\nexport function createDefer() {\n  let subscriptions: Unsubscribable[];\n\n  beforeEach(() => {\n    subscriptions = [];\n  });\n\n  afterEach(() => {\n    for (const s of subscriptions) {\n      s.unsubscribe();\n    }\n    subscriptions = [];\n  });\n\n  return function defer(x: Unsubscribable): Unsubscribable {\n    subscriptions.push(x);\n    return x;\n  };\n}\n\nexport function expectSingleListCallAndClear<T extends ObjectTypeDefinition>(\n  subFn: Mock<(e: ListPayload | undefined) => void>,\n  resolvedList: Osdk.Instance<T>[],\n  payloadOptions: Omit<Partial<ListPayload>, \"resolvedList\"> = {},\n): void {\n  if (vitest.isFakeTimers()) {\n    vitest.runOnlyPendingTimers();\n  }\n  expect(subFn).toHaveBeenCalledExactlyOnceWith(\n    listPayloadContaining({\n      ...payloadOptions,\n      resolvedList,\n    }),\n  );\n  subFn.mockClear();\n}\n\nexport function expectSingleObjectCallAndClear<T extends ObjectTypeDefinition>(\n  subFn: Mock<(e: ObjectPayload | undefined) => void>,\n  object: Osdk.Instance<T>,\n  status?: Status,\n): void {\n  expect(subFn).toHaveBeenCalledExactlyOnceWith(\n    expect.objectContaining({\n      object,\n      status: status ?? expect.any(String),\n    }),\n  );\n  subFn.mockClear();\n}\n\nexport async function waitForCall(\n  subFn: Mock<(e: any) => void>,\n  times: number = 1,\n): Promise<void> {\n  try {\n    await vi.waitFor(() => {\n      expect(subFn).toHaveBeenCalledTimes(times);\n    });\n  } catch (e) {\n    // we don't need the error, it will retrigger on the next line\n    // and that provides better behavior in the vitest vscode\n    // plugin. This places the error in the test itself instead of\n    // only in this file\n  }\n  expect(subFn).toHaveBeenCalledTimes(times);\n}\n\nfunction createSubscriptionHelper() {\n}\n\nexport function mockSingleSubCallback():\n  & Mock<\n    (e: ObjectPayload | undefined) => void\n  >\n  & {\n    // expectLoaded: (value: unknown) => Promise<void>;\n    // expectLoading: (value: unknown) => Promise<void>;\n    expectLoadingAndLoaded: (q: {\n      loading?: unknown;\n      loaded: unknown;\n    }) => Promise<void>;\n  }\n{\n  const ret = vitest.fn((e: ObjectPayload | undefined) => {});\n\n  //   async function expectLoaded(value: unknown) {\n  //     await waitForCall(ret);\n  //     // as long as we get the loaded call we are happy\n  //     expect(ret).toHaveBeenLastCalledWith(\n  //       cacheEntryContaining({\n  //         value,\n  //         status: \"loaded\",\n  //       }),\n  //     );\n  //     ret.mockClear();\n  //   }\n\n  //   async function expectLoading(value: unknown) {\n  //     await waitForCall(ret);\n  //     // as long as we get the loaded call we are happy\n  //     expect(ret).toHaveBeenCalledExactlyOnceWith(\n  //       cacheEntryContaining({\n  //         value,\n  //         status: \"loading\",\n  //       }),\n  //     );\n  //     ret.mockClear();\n  //   }\n\n  return Object.assign(ret, {\n    // expectLoaded,\n    // expectLoading,\n    expectLoadingAndLoaded: async (\n      q: { loading?: unknown; loaded: unknown },\n    ) => {\n      await waitForCall(ret, 2);\n\n      // as long as we get the loaded call we are happy\n      expect(ret).toHaveBeenNthCalledWith(\n        1,\n        q.loading,\n      );\n      expect(ret).toHaveBeenNthCalledWith(\n        2,\n        q.loaded,\n      );\n      expect(ret).toHaveBeenCalledTimes(2);\n      ret.mockClear();\n    },\n  });\n}\n\nexport function mockListSubCallback(): Mock<\n  (x: ListPayload | undefined) => void\n> {\n  return vitest.fn(\n    (x: ListPayload | undefined) => {},\n  );\n}\n\nexport function cacheEntryContaining(x: Partial<Entry<any>>): Entry<any> {\n  return {\n    cacheKey: x.cacheKey ?? expect.any(Object),\n    value: \"value\" in x\n      ? x.value\n      : expect.toBeOneOf([expect.anything(), undefined]),\n    status: x.status ?? expect.anything(),\n    lastUpdated: x.lastUpdated ?? expect.anything(),\n  };\n}\n\nfunction nonOptionalValue<T extends object, K extends keyof T>(\n  src: T,\n  key: K,\n): NonNullable<T[K]> {\n  return key in src\n    ? src[key]\n    : expect.toBeOneOf([expect.anything(), undefined]);\n}\n\nexport function objectPayloadContaining(\n  x: Partial<ObjectPayload>,\n): ObjectPayload {\n  return {\n    object: nonOptionalValue(x, \"object\"),\n    isOptimistic: expect.any(Boolean),\n    status: x.status ?? expect.anything(),\n    lastUpdated: x.lastUpdated ?? expect.anything(),\n  };\n}\n\nexport function listPayloadContaining(\n  x: Partial<ListPayload>,\n): ListPayload {\n  return {\n    fetchMore: x.fetchMore ?? expect.any(Function),\n    hasMore: x.hasMore ?? expect.any(Boolean),\n    resolvedList: x.resolvedList ?? expect.anything(),\n    isOptimistic: expect.any(Boolean),\n    status: x.status ?? expect.anything(),\n    lastUpdated: x.lastUpdated ?? expect.anything(),\n  };\n}\n\nexport function applyCustomMatchers(): void {\n  expect.extend({\n    toBeGreaterThan: (r: number, e: number) => {\n      return {\n        pass: r > e,\n        message: () => `expected ${r} to be greater than ${e} (lastUpdated)`,\n      };\n    },\n  });\n}\n\ninterface CustomMatchers<R = any> {\n  toBeGreaterThan: (n: number) => R;\n}\n\ninterface CustomAsymmetricMatchers<R = any> {\n  toBeGreaterThan: (n: number) => R;\n}\n\ndeclare module \"vitest\" {\n  interface Assertion<T = any> extends CustomMatchers<T> {}\n  interface AsymmetricMatchersContaining extends CustomAsymmetricMatchers {}\n}\n"],"mappings":"AAAA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;;AAaA,SAASA,KAAK,QAAQ,OAAO;AAE7B,OAAOC,MAAM,MAAM,SAAS;AAC5B,OAAOC,SAAS,MAAM,gBAAgB;AAEtC,SAASC,SAAS,EAAEC,UAAU,EAAEC,MAAM,EAAEC,EAAE,EAAEC,MAAM,QAAQ,QAAQ;AAGlE,SAASC,iBAAiB,QAAQ,iBAAiB;AAOnD,MAAMC,KAAK,GAAG,IAAIT,KAAK,CAAC,CAAC,CAAC,CAAC;;AA+B3B,SAASU,OAAOA,CAAC,GAAGC,IAAW,EAAE;EAC/B;EACAC,OAAO,CAACC,GAAG,CACTJ,KAAK,CAACK,MAAM,CAAC,YAAY,CAAC,EAC1B,GAAGH,IACL,CAAC;AACH;AACA;AACA;AACA;AACA;;AAEA,OAAO,SAASI,gBAAgBA,CAC9BC,QAA6B,EAC7BC,OAAgD,EACxC;EACR,MAAMC,MAAM,GAAG;IACbC,KAAK,EAAE,CAACV,KAAK,CAACW,IAAI,EAAEX,KAAK,CAACY,MAAM,CAAC;IACjCC,IAAI,EAAE,CAACb,KAAK,CAACc,KAAK,EAAEd,KAAK,CAACe,OAAO,CAAC;IAClCC,KAAK,EAAE,CAAChB,KAAK,CAACiB,IAAI,EAAEjB,KAAK,CAACkB,MAAM,CAAC;IACjCC,KAAK,EAAE,CAACnB,KAAK,CAACoB,GAAG,EAAEpB,KAAK,CAACqB,KAAK,CAAC;IAC/BC,IAAI,EAAE,CAACtB,KAAK,CAACK,MAAM,EAAEL,KAAK,CAACuB,QAAQ,CAAC;IACpCC,KAAK,EAAE,CAACxB,KAAK,CAACyB,SAAS,EAAEzB,KAAK,CAAC0B,WAAW;EAC5C,CAAU;EACV,SAASC,eAAeA,CACtBC,IAA6D,EAC7D;IACA,OAAO/B,EAAE,CAACgC,EAAE,CACV,CACE,GAAG3B,IAKF,KACE;MACH,MAAM4B,OAAO,GAAG,OAAO5B,IAAI,CAAC,CAAC,CAAC,KAAK,QAAQ;MAC3C,MAAM6B,GAA4B,GAAGD,OAAO,GAAG5B,IAAI,CAAC,CAAC,CAAC,GAAU,CAAC,CAAC;MAClE,MAAM8B,IAAW,GAAGF,OAAO,GAAG5B,IAAI,CAAC+B,KAAK,CAAC,CAAC,CAAC,GAAG/B,IAAI,CAAC+B,KAAK,CAAC,CAAC,CAAC;;MAE3D;MACA9B,OAAO,CAACC,GAAG,CACT,GAAGK,MAAM,CAACmB,IAAI,CAAC,CAAC,CAAC,CAAC,CAACA,IAAI,CAAC,GACtBpB,OAAO,EAAE0B,SAAS,GAAG,GAAG,GAAGzB,MAAM,CAACmB,IAAI,CAAC,CAAC,CAAC,CAAC,CAACpB,OAAO,CAAC0B,SAAS,CAAC,GAAG,EAAE,GACjEH,GAAG,CAACI,UAAU,GAAG,KAAKnC,KAAK,CAACoC,OAAO,CAACL,GAAG,CAACI,UAAU,CAAC,IAAI,GAAG,EAAE,EAAE,EACjE,GAAGH,IACL,CAAC;MACD,IAAIzB,QAAQ,IAAI8B,MAAM,CAACC,IAAI,CAAC/B,QAAQ,CAAC,CAACgC,MAAM,GAAG,CAAC,EAAE;QAChD;QACApC,OAAO,CAACC,GAAG,CAACG,QAAQ,CAAC;MACvB;IACF,CACF,CAAC;EACH;EACA,OAAO;IACLG,KAAK,EAAEiB,eAAe,CAAC,OAAO,CAAC;IAC/BR,KAAK,EAAEQ,eAAe,CAAC,OAAO,CAAC;IAC/Bd,IAAI,EAAEc,eAAe,CAAC,MAAM,CAAC;IAC7BL,IAAI,EAAEK,eAAe,CAAC,MAAM,CAAC;IAC7BH,KAAK,EAAEG,eAAe,CAAC,OAAO,CAAC;IAC/Ba,KAAK,EAAE3C,EAAE,CAACgC,EAAE,CAAkB,CAACY,aAAa,EAAEC,YAAY,KACxDpC,gBAAgB,CAAC;MACf,GAAGC,QAAQ;MACX,GAAGkC;IACL,CAAC,EAAE;MACDE,KAAK,EAAE,CAACD,YAAY,IAAIlC,OAAO,GAAGmC,KAAK;MACvCT,SAAS,EAAE1B,OAAO,EAAE0B,SAAS,IAAIQ,YAAY,EAAER,SAAS,GACpD,GAAG1B,OAAO,EAAE0B,SAAS,GAAG,GAAG1B,OAAO,CAAC0B,SAAS,GAAG,GAAG,EAAE,GACpDQ,YAAY,EAAER,SAAS,IAAI,EAAE,EAC7B,GACAU;IACN,CAAC,CACH,CAAC;IACD5B,KAAK,EAAEW,eAAe,CAAC,OAAO,CAAC;IAC/BkB,cAAc,EAAEhD,EAAE,CAACgC,EAAE,CAAC,MAAU,IAAI;EACtC,CAAC;AACH;AAEA,OAAO,SAASiB,sBAAsBA,CAAA,EAAqB;EACzD,MAAMC,MAAM,GAAGjD,MAAM,CAAC+B,EAAE,CAAgB,CAAqC;EAE7E,MAAMmB,MAAM,GAAG1C,gBAAgB,CAAC,CAAC,CAAC,CAAC;;EAEnC;EACAyC,MAAM,CAACE,kBAAkB,CAAC,CAAC,GAAG/C,IAAW,KAAK;IAC5C,MAAMgD,WAAW,GAAGF,MAAM,CAACR,KAAK,CAAC,CAAC,CAAC,EAAE;MAAEN,SAAS,EAAE;IAAuB,CAAC,CAAC;IAC3E,IAAI;MACF,MAAM,IAAIiB,KAAK,CAAC,4CAA4C,CAAC;IAC/D,CAAC,CAAC,OAAOC,CAAC,EAAE;MACVF,WAAW,CAAC/B,KAAK,CAAC,0BAA0B,EAAEiC,CAAC,CAAC;IAClD;IAEA,IAAIC,KAAK;IACT,MAAMC,aAAa,GAAG;MACpBD,KAAK,EAAEA,CAAC,GAAGE,SAAgB,KAAK;QAC9BL,WAAW,CAAClC,KAAK,CAAC,OAAO,EAAEuC,SAAS,CAAC;QACrCF,KAAK,GAAGE,SAAS;QACjB,OAAOD,aAAa;MACtB,CAAC;MACDE,SAAS,EAAEA,CAAC,GAAGC,SAAgB,KAAK;QAClCP,WAAW,CAAClC,KAAK,CAAC,UAAU,EAAEqC,KAAK,EAAGI,SAAS,CAAC;QAChD,MAAM,IAAIN,KAAK,CAAC,IAAI,CAAC;MACvB,CAAC;MACDO,QAAQ,EAAEA,CAAC,GAAGD,SAAgB,KAAK;QACjCP,WAAW,CAAClC,KAAK,CAAC,UAAU,EAAEyC,SAAS,CAAC;QAExC,MAAM,IAAIN,KAAK,CAAC,IAAI,CAAC;MACvB;IACF,CAAC;IACD,OAAOG,aAAa;EACtB,CAAC,CAAC;EAEFP,MAAM,CAAChD,iBAAiB,CAAC,GAAG;IAC1B4D,OAAO,EAAE,uBAAuB;IAChCC,WAAW,EAAE,cAAc;IAC3BC,aAAa,EAAE/D,MAAM,CAAC+B,EAAE,CAAC,CAAC;IAC1BiC,cAAc,EAAEhE,MAAM,CAAC+B,EAAE,CAAC,CAAC;IAC3BkC,gBAAgB,EAAE;MAChBC,mBAAmB,EAAElE,MAAM,CAAC+B,EAAE,CAAC,CAAC;MAChCoC,sBAAsB,EAAEnE,MAAM,CAAC+B,EAAE,CAAC,CAAC;MACnCqC,mBAAmB,EAAEpE,MAAM,CAAC+B,EAAE,CAAC,CAAC;MAChCsC,kBAAkB,EAAErE,MAAM,CAAC+B,EAAE,CAAC;IAChC,CAAC;IACDuC,aAAa,EAAEtE,MAAM,CAAC+B,EAAE,CAAC,CAAC;IAC1BwC,gBAAgB,EAAEvE,MAAM,CAAC+B,EAAE,CAAC,CAAC;IAC7ByC,KAAK,EAAExE,MAAM,CAAC+B,EAAE,CAAC,CAAC;IAClB0C,cAAc,EAAE,CAAC,CAAQ;IACzBC,cAAc,EAAE,CAAC,CAAC;IAClBxB;EACF,CAAC;EAwFD,OAAO;IACLD,MAAM;IACN0B,mBAAmB,EA/BrB,SAAAA,CAAA,EAEE;MACA,MAAMC,CAAC,GAAGlF,MAAM,CAA8B,CAAC;MAE/CuD,MAAM,CAAC4B,mBAAmB,CACxB;QACEC,WAAW,EAAE,MAAAA,CAAA,KAA8C;UACzD,MAAMC,CAAC,GAAG,MAAMH,CAAC,CAACI,OAAO;UACzB,OAAO;YACLC,IAAI,EAAE,OAAO;YACbC,UAAU,EAAEH,CAAC,CAACG,UAAU,IAAI,EAAE;YAC9BC,YAAY,EAAEJ,CAAC,CAACI,YAAY,IAAI,EAAE;YAClCC,cAAc,EAAEL,CAAC,CAACK,cAAc,IAAI,EAAE;YACtCC,YAAY,EAAEN,CAAC,CAACM,YAAY,IAAI,EAAE;YAClCC,iBAAiB,EAAEP,CAAC,CAACO,iBAAiB,IAAI,CAAC;YAC3CC,mBAAmB,EAAER,CAAC,CAACQ,mBAAmB,IAAI,CAAC;YAC/CC,iBAAiB,EAAET,CAAC,CAACS,iBAAiB,IAAI,EAAE;YAC5CC,eAAe,EAAEV,CAAC,CAACU,eAAe,IAAI;UACxC,CAAC;QACH;MACF,CAIF,CAAC;MACD,OAAOb,CAAC;IACV,CAIqB;IACnBc,gBAAgB,EArDlB,SAAAA,CAEEC,UAAsC,EAAsB;MAC5D,MAAMf,CAAC,GAAGlF,MAAM,CAAI,CAAC;MAErBuD,MAAM,CAAC4B,mBAAmB,CACxB;QACEjB,QAAQ,EAAE,MAAOgC,CAAC,IAAK;UACrBzF,OAAO,CAAC,UAAU,EAAEyF,CAAC,CAAC;UACtB,EACED,UAAU,KAAK7C,SAAS,IAAI8C,CAAC,KAAKD,UAAU,IAAAE,OAAA,CAAAC,GAAA,CAAAC,QAAA,oBAD9CpG,SAAS,QAEP,sBAAsB,IAFxBA,SAAS;UAIT,MAAMqG,CAAC,GAAG,MAAMpB,CAAC,CAACI,OAAO;UACzB,OAAO;YAAE,GAAGgB,CAAC;YAAEC,WAAW,EAAEL;UAAE,CAAC;QACjC;MACF,CACF,CAAC;MACD,OAAOhB,CAAC;IACV,CAkCkB;IAChBsB,sBAAsB,EA1FxB,SAAAA,CAAA,EAAkC;MAChC,MAAMtB,CAAC,GAAGlF,MAAM,CAEd,CAAC;MACHK,EAAE,CAACoG,MAAM,CAAClD,MAAM,CAAChD,iBAAiB,CAAC,CAAC+D,cAAc,CAAC,CAACa,mBAAmB,CACrED,CAAC,CAACI,OACJ,CAAC;MACD,OAAOJ,CAAC;IACV,CAkFwB;IACtBwB,iBAAiB,EAjFnB,SAAAA,CAAA,EAIwB;MACtB,MAAMxB,CAAC,GAAGlF,MAAM,CAAI,CAAC;MAErB,MAAM2G,SAA0C,GAAG;QACjD3C,SAAS,EAAE,MAAO4C,aAAa,IAAK;UAClCnG,OAAO,CAAC,WAAW,EAAEmG,aAAa,CAAC;UACnC,MAAMN,CAAC,GAAG,MAAMpB,CAAC,CAACI,OAAO;UACzB,OAAO;YAAE,GAAGgB,CAAC;YAAEC,WAAW,EAAEK;UAAc,CAAC;QAC7C,CAAC;QACD/C,KAAK,EAAGgD,MAAM,IAAK;UACjBpG,OAAO,CAAC,OAAO,EAAEoG,MAAM,CAAC;UACxB,OAAOF,SAAS;QAClB;MACF,CAGoC;MAEpCpD,MAAM,CAAC4B,mBAAmB,CAACwB,SAAS,CAAC;MACrC,OAAOzB,CAAC;IACV;EA0DA,CAAC;AACH;AAEA,OAAO,SAAS4B,WAAWA,CAAA,EAAG;EAC5B,IAAIC,aAA+B;EAEnC5G,UAAU,CAAC,MAAM;IACf4G,aAAa,GAAG,EAAE;EACpB,CAAC,CAAC;EAEF7G,SAAS,CAAC,MAAM;IACd,KAAK,MAAM8G,CAAC,IAAID,aAAa,EAAE;MAC7BC,CAAC,CAACC,WAAW,CAAC,CAAC;IACjB;IACAF,aAAa,GAAG,EAAE;EACpB,CAAC,CAAC;EAEF,OAAO,UAAe1B,CAAiB,EAAkB;IACvD0B,aAAa,CAACG,IAAI,CAAC7B,CAAC,CAAC;IACrB,OAAOA,CAAC;EACV,CAAC;AACH;AAEA,OAAO,SAAS8B,4BAA4BA,CAC1CC,KAAiD,EACjDC,YAAgC,EAChCC,cAA0D,GAAG,CAAC,CAAC,EACzD;EACN,IAAIhH,MAAM,CAACiH,YAAY,CAAC,CAAC,EAAE;IACzBjH,MAAM,CAACkH,oBAAoB,CAAC,CAAC;EAC/B;EACApH,MAAM,CAACgH,KAAK,CAAC,CAACK,+BAA+B,CAC3CC,qBAAqB,CAAC;IACpB,GAAGJ,cAAc;IACjBD;EACF,CAAC,CACH,CAAC;EACDD,KAAK,CAACO,SAAS,CAAC,CAAC;AACnB;AAEA,OAAO,SAASC,8BAA8BA,CAC5CR,KAAmD,EACnDS,MAAwB,EACxBC,MAAe,EACT;EACN1H,MAAM,CAACgH,KAAK,CAAC,CAACK,+BAA+B,CAC3CrH,MAAM,CAAC2H,gBAAgB,CAAC;IACtBF,MAAM;IACNC,MAAM,EAAEA,MAAM,IAAI1H,MAAM,CAAC4H,GAAG,CAACC,MAAM;EACrC,CAAC,CACH,CAAC;EACDb,KAAK,CAACO,SAAS,CAAC,CAAC;AACnB;AAEA,OAAO,eAAeO,WAAWA,CAC/Bd,KAA6B,EAC7Be,KAAa,GAAG,CAAC,EACF;EACf,IAAI;IACF,MAAM9H,EAAE,CAAC+H,OAAO,CAAC,MAAM;MACrBhI,MAAM,CAACgH,KAAK,CAAC,CAACiB,qBAAqB,CAACF,KAAK,CAAC;IAC5C,CAAC,CAAC;EACJ,CAAC,CAAC,OAAOvE,CAAC,EAAE;IACV;IACA;IACA;IACA;EAAA;EAEFxD,MAAM,CAACgH,KAAK,CAAC,CAACiB,qBAAqB,CAACF,KAAK,CAAC;AAC5C;AAEA,SAASG,wBAAwBA,CAAA,EAAG,CACpC;AAEA,OAAO,SAASC,qBAAqBA,CAAA,EAYrC;EACE,MAAMC,GAAG,GAAGlI,MAAM,CAAC+B,EAAE,CAAC,MAAkC,CAAC,CAAC,CAAC;;EAE3D;EACA;EACA;EACA;EACA;EACA;EACA;EACA;EACA;EACA;EACA;;EAEA;EACA;EACA;EACA;EACA;EACA;EACA;EACA;EACA;EACA;EACA;;EAEA,OAAOQ,MAAM,CAAC4F,MAAM,CAACD,GAAG,EAAE;IACxB;IACA;IACAE,sBAAsB,EAAE,MACtBC,CAAyC,IACtC;MACH,MAAMT,WAAW,CAACM,GAAG,EAAE,CAAC,CAAC;;MAEzB;MACApI,MAAM,CAACoI,GAAG,CAAC,CAACI,uBAAuB,CACjC,CAAC,EACDD,CAAC,CAACE,OACJ,CAAC;MACDzI,MAAM,CAACoI,GAAG,CAAC,CAACI,uBAAuB,CACjC,CAAC,EACDD,CAAC,CAACG,MACJ,CAAC;MACD1I,MAAM,CAACoI,GAAG,CAAC,CAACH,qBAAqB,CAAC,CAAC,CAAC;MACpCG,GAAG,CAACb,SAAS,CAAC,CAAC;IACjB;EACF,CAAC,CAAC;AACJ;AAEA,OAAO,SAASoB,mBAAmBA,CAAA,EAEjC;EACA,OAAOzI,MAAM,CAAC+B,EAAE,CACd,MAAgC,CAAC,CACnC,CAAC;AACH;AAEA,OAAO,SAAS2G,oBAAoBA,CAAC3D,CAAsB,EAAc;EACvE,OAAO;IACL4D,QAAQ,EAAE5D,CAAC,CAAC4D,QAAQ,IAAI7I,MAAM,CAAC4H,GAAG,CAACnF,MAAM,CAAC;IAC1CqG,KAAK,EAAE,OAAO,IAAI7D,CAAC,GACfA,CAAC,CAAC6D,KAAK,GACP9I,MAAM,CAAC+I,SAAS,CAAC,CAAC/I,MAAM,CAACgJ,QAAQ,CAAC,CAAC,EAAEhG,SAAS,CAAC,CAAC;IACpD0E,MAAM,EAAEzC,CAAC,CAACyC,MAAM,IAAI1H,MAAM,CAACgJ,QAAQ,CAAC,CAAC;IACrCC,WAAW,EAAEhE,CAAC,CAACgE,WAAW,IAAIjJ,MAAM,CAACgJ,QAAQ,CAAC;EAChD,CAAC;AACH;AAEA,SAASE,gBAAgBA,CACvBC,GAAM,EACNC,GAAM,EACa;EACnB,OAAOA,GAAG,IAAID,GAAG,GACbA,GAAG,CAACC,GAAG,CAAC,GACRpJ,MAAM,CAAC+I,SAAS,CAAC,CAAC/I,MAAM,CAACgJ,QAAQ,CAAC,CAAC,EAAEhG,SAAS,CAAC,CAAC;AACtD;AAEA,OAAO,SAASqG,uBAAuBA,CACrCpE,CAAyB,EACV;EACf,OAAO;IACLwC,MAAM,EAAEyB,gBAAgB,CAACjE,CAAC,EAAE,QAAQ,CAAC;IACrCqE,YAAY,EAAEtJ,MAAM,CAAC4H,GAAG,CAAC2B,OAAO,CAAC;IACjC7B,MAAM,EAAEzC,CAAC,CAACyC,MAAM,IAAI1H,MAAM,CAACgJ,QAAQ,CAAC,CAAC;IACrCC,WAAW,EAAEhE,CAAC,CAACgE,WAAW,IAAIjJ,MAAM,CAACgJ,QAAQ,CAAC;EAChD,CAAC;AACH;AAEA,OAAO,SAAS1B,qBAAqBA,CACnCrC,CAAuB,EACV;EACb,OAAO;IACLuE,SAAS,EAAEvE,CAAC,CAACuE,SAAS,IAAIxJ,MAAM,CAAC4H,GAAG,CAAC6B,QAAQ,CAAC;IAC9CC,OAAO,EAAEzE,CAAC,CAACyE,OAAO,IAAI1J,MAAM,CAAC4H,GAAG,CAAC2B,OAAO,CAAC;IACzCtC,YAAY,EAAEhC,CAAC,CAACgC,YAAY,IAAIjH,MAAM,CAACgJ,QAAQ,CAAC,CAAC;IACjDM,YAAY,EAAEtJ,MAAM,CAAC4H,GAAG,CAAC2B,OAAO,CAAC;IACjC7B,MAAM,EAAEzC,CAAC,CAACyC,MAAM,IAAI1H,MAAM,CAACgJ,QAAQ,CAAC,CAAC;IACrCC,WAAW,EAAEhE,CAAC,CAACgE,WAAW,IAAIjJ,MAAM,CAACgJ,QAAQ,CAAC;EAChD,CAAC;AACH;AAEA,OAAO,SAASW,mBAAmBA,CAAA,EAAS;EAC1C3J,MAAM,CAAC4J,MAAM,CAAC;IACZC,eAAe,EAAEA,CAAC3D,CAAS,EAAE1C,CAAS,KAAK;MACzC,OAAO;QACLsG,IAAI,EAAE5D,CAAC,GAAG1C,CAAC;QACXuG,OAAO,EAAEA,CAAA,KAAM,YAAY7D,CAAC,uBAAuB1C,CAAC;MACtD,CAAC;IACH;EACF,CAAC,CAAC;AACJ","ignoreList":[]}

package/build/browser/queries/applyQuery.js

@@ -15,6 +15,8 @@
  */
 
 import * as OntologiesV2 from "@osdk/foundry.ontologies";
+import invariant from "tiny-invariant";
+import { createObjectSpecifierFromPrimaryKey } from "../object/createObjectSpecifierFromPrimaryKey.js";
 import { createObjectSet } from "../objectSet/createObjectSet.js";
 import { hydrateAttachmentFromRidInternal } from "../public-utils/hydrateAttachmentFromRid.js";
 import { addUserAgentAndRequestContextHeaders } from "../util/addUserAgentAndRequestContextHeaders.js";
@@ -111,6 +113,19 @@ async function remapQueryResponse(client, responseDataType, responseValue, defin
         }
         return responseValue;
       }
+    case "map":
+      {
+        const map = {};
+        !Array.isArray(responseValue) ? process.env.NODE_ENV !== "production" ? invariant(false, "Expected array entry") : invariant(false) : void 0;
+        for (const entry of responseValue) {
+          !entry.key ? process.env.NODE_ENV !== "production" ? invariant(false, "Expected key") : invariant(false) : void 0;
+          !entry.value ? process.env.NODE_ENV !== "production" ? invariant(false, "Expected value") : invariant(false) : void 0;
+          const key = responseDataType.keyType.type === "object" ? getObjectSpecifier(entry.key, responseDataType.keyType.object, definitions) : entry.key;
+          const value = await remapQueryResponse(client, responseDataType.valueType, entry.value, definitions);
+          map[key] = value;
+        }
+        return map;
+      }
     case "twoDimensionalAggregation":
       {
         const result = [];
@@ -171,6 +186,15 @@ async function getRequiredDefinitions(dataType, client) {
       {
         return getRequiredDefinitions(dataType.set, client);
       }
+    case "map":
+      {
+        for (const value of [dataType.keyType, dataType.valueType]) {
+          for (const [type, objectDef] of await getRequiredDefinitions(value, client)) {
+            result.set(type, objectDef);
+          }
+        }
+        break;
+      }
     case "struct":
       {
         for (const value of Object.values(dataType.struct)) {
@@ -223,12 +247,20 @@ function requiresConversion(dataType) {
       return false;
   }
 }
+function getObjectSpecifier(primaryKey, objectTypeApiName, definitions) {
+  const def = definitions.get(objectTypeApiName);
+  if (!def) {
+    throw new Error(`Missing definition for ${objectTypeApiName}`);
+  }
+  return createObjectSpecifierFromPrimaryKey(def, primaryKey);
+}
 export function createQueryObjectResponse(primaryKey, objectDef) {
   return {
     $apiName: objectDef.apiName,
     $title: undefined,
     $objectType: objectDef.apiName,
-    $primaryKey: primaryKey
+    $primaryKey: primaryKey,
+    $objectSpecifier: createObjectSpecifierFromPrimaryKey(objectDef, primaryKey)
   };
 }
 //# sourceMappingURL=applyQuery.js.map

package/build/browser/queries/applyQuery.js.map

@@ -1 +1 @@
-{"version":3,"file":"applyQuery.js","names":["OntologiesV2","createObjectSet","hydrateAttachmentFromRidInternal","addUserAgentAndRequestContextHeaders","augmentRequestContext","toDataValueQueries","applyQuery","client","query","params","qd","ontologyProvider","getQueryDefinition","apiName","response","Queries","execute","finalMethodCall","ontologyRid","parameters","remapQueryParams","objectOutputDefs","getRequiredDefinitions","output","remappedResponse","remapQueryResponse","value","paramTypes","parameterMap","key","Object","entries","responseDataType","responseValue","definitions","nullable","undefined","Error","multiplicity","withoutMultiplicity","i","length","type","set","def","get","object","createQueryObjectResponse","objectSet","objectSets","objectType","reference","subtype","struct","requiresConversion","result","groups","push","subResult","subKey","dataType","Map","objectDef","getObjectDefinition","values","some","primaryKey","$apiName","$title","$objectType","$primaryKey"],"sources":["applyQuery.ts"],"sourcesContent":["/*\n * Copyright 2024 Palantir Technologies, Inc. All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport type {\n  AllowedBucketKeyTypes,\n  AllowedBucketTypes,\n  CompileTimeMetadata,\n  ObjectOrInterfaceDefinition,\n  ObjectTypeDefinition,\n  OsdkBase,\n  PrimaryKeyType,\n  QueryDataTypeDefinition,\n  QueryDefinition,\n  QueryParameterDefinition,\n} from \"@osdk/api\";\nimport type { DataValue } from \"@osdk/foundry.ontologies\";\nimport * as OntologiesV2 from \"@osdk/foundry.ontologies\";\nimport type { MinimalClient } from \"../MinimalClientContext.js\";\nimport { createObjectSet } from \"../objectSet/createObjectSet.js\";\nimport { hydrateAttachmentFromRidInternal } from \"../public-utils/hydrateAttachmentFromRid.js\";\nimport { addUserAgentAndRequestContextHeaders } from \"../util/addUserAgentAndRequestContextHeaders.js\";\nimport { augmentRequestContext } from \"../util/augmentRequestContext.js\";\nimport { toDataValueQueries } from \"../util/toDataValueQueries.js\";\nimport type { QueryParameterType, QueryReturnType } from \"./types.js\";\n\nexport async function applyQuery<\n  QD extends QueryDefinition<any>,\n  P extends QueryParameterType<CompileTimeMetadata<QD>[\"parameters\"]>,\n>(\n  client: MinimalClient,\n  query: QD,\n  params?: P,\n): Promise<\n  QueryReturnType<CompileTimeMetadata<QD>[\"output\"]>\n> {\n  const qd = await client.ontologyProvider.getQueryDefinition(query.apiName);\n\n  const response = await OntologiesV2.Queries.execute(\n    addUserAgentAndRequestContextHeaders(\n      augmentRequestContext(client, _ => ({ finalMethodCall: \"applyQuery\" })),\n      query,\n    ),\n    await client.ontologyRid,\n    query.apiName,\n    {\n      parameters: params\n        ? await remapQueryParams(\n          params as { [parameterId: string]: any },\n          client,\n          qd.parameters,\n        )\n        : {},\n    },\n  );\n  const objectOutputDefs = await getRequiredDefinitions(qd.output, client);\n  const remappedResponse = await remapQueryResponse(\n    client,\n    qd.output,\n    response.value,\n    objectOutputDefs,\n  );\n  return remappedResponse as QueryReturnType<CompileTimeMetadata<QD>[\"output\"]>;\n}\n\nasync function remapQueryParams(\n  params: { [parameterId: string]: any },\n  client: MinimalClient,\n  paramTypes: Record<string, QueryParameterDefinition<any>>,\n): Promise<{ [parameterId: string]: any }> {\n  const parameterMap: { [parameterName: string]: unknown } = {};\n  for (const [key, value] of Object.entries(params)) {\n    parameterMap[key] = await toDataValueQueries(\n      value,\n      client,\n      paramTypes[key],\n    );\n  }\n  return parameterMap;\n}\n\nasync function remapQueryResponse<\n  Q extends ObjectTypeDefinition,\n  T extends QueryDataTypeDefinition<Q | never>,\n>(\n  client: MinimalClient,\n  responseDataType: T,\n  responseValue: DataValue,\n  definitions: Map<string, ObjectOrInterfaceDefinition>,\n): Promise<QueryReturnType<T>> {\n  // handle null responses\n  if (responseValue == null) {\n    if (responseDataType.nullable) {\n      return undefined as unknown as QueryReturnType<typeof responseDataType>;\n    } else {\n      throw new Error(\"Got null response when nullable was not allowed\");\n    }\n  }\n\n  if (\n    responseDataType.multiplicity != null\n    && responseDataType.multiplicity\n  ) {\n    const withoutMultiplicity = { ...responseDataType, multiplicity: false };\n    for (let i = 0; i < responseValue.length; i++) {\n      responseValue[i] = await remapQueryResponse(\n        client,\n        withoutMultiplicity,\n        responseValue[i],\n        definitions,\n      );\n    }\n    return responseValue as QueryReturnType<typeof responseDataType>;\n  }\n\n  switch (responseDataType.type) {\n    case \"union\": {\n      throw new Error(\"Union return types are not yet supported\");\n    }\n\n    case \"set\": {\n      for (let i = 0; i < responseValue.length; i++) {\n        responseValue[i] = await remapQueryResponse(\n          client,\n          responseDataType.set,\n          responseValue[i],\n          definitions,\n        );\n      }\n\n      return responseValue as QueryReturnType<typeof responseDataType>;\n    }\n\n    case \"attachment\": {\n      return hydrateAttachmentFromRidInternal(\n        client,\n        responseValue,\n      ) as QueryReturnType<\n        typeof responseDataType\n      >;\n    }\n    case \"object\": {\n      const def = definitions.get(responseDataType.object);\n      if (!def) {\n        throw new Error(\n          `Missing definition for ${responseDataType.object}`,\n        );\n      }\n      return createQueryObjectResponse(\n        responseValue,\n        def,\n      ) as QueryReturnType<\n        typeof responseDataType\n      >;\n    }\n\n    case \"objectSet\": {\n      const def = definitions.get(responseDataType.objectSet);\n      if (!def) {\n        throw new Error(\n          `Missing definition for ${responseDataType.objectSet}`,\n        );\n      }\n      if (typeof responseValue === \"string\") {\n        return createObjectSet(def, client, {\n          type: \"intersect\",\n          objectSets: [\n            { type: \"base\", objectType: responseDataType.objectSet },\n            { type: \"reference\", reference: responseValue },\n          ],\n        }) as QueryReturnType<typeof responseDataType>;\n      }\n\n      return createObjectSet(\n        def,\n        client,\n        responseValue,\n      ) as QueryReturnType<\n        typeof responseDataType\n      >;\n    }\n    case \"struct\": {\n      // figure out what keys need to be fixed up\n      for (const [key, subtype] of Object.entries(responseDataType.struct)) {\n        if (requiresConversion(subtype)) {\n          responseValue[key] = await remapQueryResponse(\n            client,\n            subtype,\n            responseValue[key],\n            definitions,\n          );\n        }\n      }\n\n      return responseValue as QueryReturnType<typeof responseDataType>;\n    }\n    case \"twoDimensionalAggregation\": {\n      const result: {\n        key: AllowedBucketKeyTypes;\n        value: AllowedBucketTypes;\n      }[] = [];\n      for (const { key, value } of responseValue.groups) {\n        result.push({ key, value });\n      }\n      return result as QueryReturnType<typeof responseDataType>;\n    }\n\n    case \"threeDimensionalAggregation\": {\n      const result: {\n        key: AllowedBucketKeyTypes;\n        groups: { key: AllowedBucketKeyTypes; value: AllowedBucketTypes }[];\n      }[] = [];\n      for (const { key, groups } of responseValue.groups) {\n        const subResult: { key: any; value: any }[] = [];\n        for (const { key: subKey, value } of groups) {\n          subResult.push({ key: subKey, value });\n        }\n        result.push({ key, groups: subResult });\n      }\n      return result as QueryReturnType<typeof responseDataType>;\n    }\n  }\n\n  return responseValue as QueryReturnType<typeof responseDataType>;\n}\n\nasync function getRequiredDefinitions(\n  dataType: QueryDataTypeDefinition,\n  client: MinimalClient,\n): Promise<Map<string, ObjectOrInterfaceDefinition>> {\n  const result = new Map<string, ObjectOrInterfaceDefinition>();\n  switch (dataType.type) {\n    case \"objectSet\": {\n      const objectDef = await client.ontologyProvider.getObjectDefinition(\n        dataType.objectSet,\n      );\n      result.set(dataType.objectSet, objectDef);\n      break;\n    }\n    case \"object\": {\n      const objectDef = await client.ontologyProvider.getObjectDefinition(\n        dataType.object,\n      );\n      result.set(dataType.object, objectDef);\n      break;\n    }\n\n    case \"set\": {\n      return getRequiredDefinitions(dataType.set, client);\n    }\n\n    case \"struct\": {\n      for (const value of Object.values(dataType.struct)) {\n        for (\n          const [type, objectDef] of await getRequiredDefinitions(value, client)\n        ) {\n          result.set(type, objectDef);\n        }\n      }\n      break;\n    }\n    case \"attachment\":\n    case \"boolean\":\n    case \"date\":\n    case \"double\":\n    case \"float\":\n    case \"integer\":\n    case \"long\":\n    case \"string\":\n    case \"threeDimensionalAggregation\":\n    case \"timestamp\":\n    case \"twoDimensionalAggregation\":\n    case \"union\":\n      break;\n  }\n\n  return result;\n}\nfunction requiresConversion(dataType: QueryDataTypeDefinition) {\n  switch (dataType.type) {\n    case \"boolean\":\n    case \"date\":\n    case \"double\":\n    case \"float\":\n    case \"integer\":\n    case \"long\":\n    case \"string\":\n    case \"timestamp\":\n      return false;\n\n    case \"union\":\n      return true;\n\n    case \"struct\":\n      return Object.values(dataType.struct).some(requiresConversion);\n\n    case \"set\":\n      return requiresConversion(dataType.set);\n\n    case \"attachment\":\n    case \"objectSet\":\n    case \"twoDimensionalAggregation\":\n    case \"threeDimensionalAggregation\":\n    case \"object\":\n      return true;\n\n    default:\n      return false;\n  }\n}\n\nexport function createQueryObjectResponse<\n  Q extends ObjectOrInterfaceDefinition,\n>(\n  primaryKey: PrimaryKeyType<Q>,\n  objectDef: Q,\n): OsdkBase<Q> {\n  return {\n    $apiName: objectDef.apiName,\n    $title: undefined,\n    $objectType: objectDef.apiName,\n    $primaryKey: primaryKey,\n  };\n}\n"],"mappings":"AAAA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;;AAeA,OAAO,KAAKA,YAAY,MAAM,0BAA0B;AAExD,SAASC,eAAe,QAAQ,iCAAiC;AACjE,SAASC,gCAAgC,QAAQ,6CAA6C;AAC9F,SAASC,oCAAoC,QAAQ,iDAAiD;AACtG,SAASC,qBAAqB,QAAQ,kCAAkC;AACxE,SAASC,kBAAkB,QAAQ,+BAA+B;AAGlE,OAAO,eAAeC,UAAUA,CAI9BC,MAAqB,EACrBC,KAAS,EACTC,MAAU,EAGV;EACA,MAAMC,EAAE,GAAG,MAAMH,MAAM,CAACI,gBAAgB,CAACC,kBAAkB,CAACJ,KAAK,CAACK,OAAO,CAAC;EAE1E,MAAMC,QAAQ,GAAG,MAAMd,YAAY,CAACe,OAAO,CAACC,OAAO,CACjDb,oCAAoC,CAClCC,qBAAqB,CAACG,MAAM,EAAE,OAAM;IAAEU,eAAe,EAAE;EAAa,CAAC,CAAC,CAAC,EACvET,KACF,CAAC,EACD,MAAMD,MAAM,CAACW,WAAW,EACxBV,KAAK,CAACK,OAAO,EACb;IACEM,UAAU,EAAEV,MAAM,GACd,MAAMW,gBAAgB,CACtBX,MAAM,EACNF,MAAM,EACNG,EAAE,CAACS,UACL,CAAC,GACC,CAAC;EACP,CACF,CAAC;EACD,MAAME,gBAAgB,GAAG,MAAMC,sBAAsB,CAACZ,EAAE,CAACa,MAAM,EAAEhB,MAAM,CAAC;EACxE,MAAMiB,gBAAgB,GAAG,MAAMC,kBAAkB,CAC/ClB,MAAM,EACNG,EAAE,CAACa,MAAM,EACTT,QAAQ,CAACY,KAAK,EACdL,gBACF,CAAC;EACD,OAAOG,gBAAgB;AACzB;AAEA,eAAeJ,gBAAgBA,CAC7BX,MAAsC,EACtCF,MAAqB,EACrBoB,UAAyD,EAChB;EACzC,MAAMC,YAAkD,GAAG,CAAC,CAAC;EAC7D,KAAK,MAAM,CAACC,GAAG,EAAEH,KAAK,CAAC,IAAII,MAAM,CAACC,OAAO,CAACtB,MAAM,CAAC,EAAE;IACjDmB,YAAY,CAACC,GAAG,CAAC,GAAG,MAAMxB,kBAAkB,CAC1CqB,KAAK,EACLnB,MAAM,EACNoB,UAAU,CAACE,GAAG,CAChB,CAAC;EACH;EACA,OAAOD,YAAY;AACrB;AAEA,eAAeH,kBAAkBA,CAI/BlB,MAAqB,EACrByB,gBAAmB,EACnBC,aAAwB,EACxBC,WAAqD,EACxB;EAC7B;EACA,IAAID,aAAa,IAAI,IAAI,EAAE;IACzB,IAAID,gBAAgB,CAACG,QAAQ,EAAE;MAC7B,OAAOC,SAAS;IAClB,CAAC,MAAM;MACL,MAAM,IAAIC,KAAK,CAAC,iDAAiD,CAAC;IACpE;EACF;EAEA,IACEL,gBAAgB,CAACM,YAAY,IAAI,IAAI,IAClCN,gBAAgB,CAACM,YAAY,EAChC;IACA,MAAMC,mBAAmB,GAAG;MAAE,GAAGP,gBAAgB;MAAEM,YAAY,EAAE;IAAM,CAAC;IACxE,KAAK,IAAIE,CAAC,GAAG,CAAC,EAAEA,CAAC,GAAGP,aAAa,CAACQ,MAAM,EAAED,CAAC,EAAE,EAAE;MAC7CP,aAAa,CAACO,CAAC,CAAC,GAAG,MAAMf,kBAAkB,CACzClB,MAAM,EACNgC,mBAAmB,EACnBN,aAAa,CAACO,CAAC,CAAC,EAChBN,WACF,CAAC;IACH;IACA,OAAOD,aAAa;EACtB;EAEA,QAAQD,gBAAgB,CAACU,IAAI;IAC3B,KAAK,OAAO;MAAE;QACZ,MAAM,IAAIL,KAAK,CAAC,0CAA0C,CAAC;MAC7D;IAEA,KAAK,KAAK;MAAE;QACV,KAAK,IAAIG,CAAC,GAAG,CAAC,EAAEA,CAAC,GAAGP,aAAa,CAACQ,MAAM,EAAED,CAAC,EAAE,EAAE;UAC7CP,aAAa,CAACO,CAAC,CAAC,GAAG,MAAMf,kBAAkB,CACzClB,MAAM,EACNyB,gBAAgB,CAACW,GAAG,EACpBV,aAAa,CAACO,CAAC,CAAC,EAChBN,WACF,CAAC;QACH;QAEA,OAAOD,aAAa;MACtB;IAEA,KAAK,YAAY;MAAE;QACjB,OAAO/B,gCAAgC,CACrCK,MAAM,EACN0B,aACF,CAAC;MAGH;IACA,KAAK,QAAQ;MAAE;QACb,MAAMW,GAAG,GAAGV,WAAW,CAACW,GAAG,CAACb,gBAAgB,CAACc,MAAM,CAAC;QACpD,IAAI,CAACF,GAAG,EAAE;UACR,MAAM,IAAIP,KAAK,CACb,0BAA0BL,gBAAgB,CAACc,MAAM,EACnD,CAAC;QACH;QACA,OAAOC,yBAAyB,CAC9Bd,aAAa,EACbW,GACF,CAAC;MAGH;IAEA,KAAK,WAAW;MAAE;QAChB,MAAMA,GAAG,GAAGV,WAAW,CAACW,GAAG,CAACb,gBAAgB,CAACgB,SAAS,CAAC;QACvD,IAAI,CAACJ,GAAG,EAAE;UACR,MAAM,IAAIP,KAAK,CACb,0BAA0BL,gBAAgB,CAACgB,SAAS,EACtD,CAAC;QACH;QACA,IAAI,OAAOf,aAAa,KAAK,QAAQ,EAAE;UACrC,OAAOhC,eAAe,CAAC2C,GAAG,EAAErC,MAAM,EAAE;YAClCmC,IAAI,EAAE,WAAW;YACjBO,UAAU,EAAE,CACV;cAAEP,IAAI,EAAE,MAAM;cAAEQ,UAAU,EAAElB,gBAAgB,CAACgB;YAAU,CAAC,EACxD;cAAEN,IAAI,EAAE,WAAW;cAAES,SAAS,EAAElB;YAAc,CAAC;UAEnD,CAAC,CAAC;QACJ;QAEA,OAAOhC,eAAe,CACpB2C,GAAG,EACHrC,MAAM,EACN0B,aACF,CAAC;MAGH;IACA,KAAK,QAAQ;MAAE;QACb;QACA,KAAK,MAAM,CAACJ,GAAG,EAAEuB,OAAO,CAAC,IAAItB,MAAM,CAACC,OAAO,CAACC,gBAAgB,CAACqB,MAAM,CAAC,EAAE;UACpE,IAAIC,kBAAkB,CAACF,OAAO,CAAC,EAAE;YAC/BnB,aAAa,CAACJ,GAAG,CAAC,GAAG,MAAMJ,kBAAkB,CAC3ClB,MAAM,EACN6C,OAAO,EACPnB,aAAa,CAACJ,GAAG,CAAC,EAClBK,WACF,CAAC;UACH;QACF;QAEA,OAAOD,aAAa;MACtB;IACA,KAAK,2BAA2B;MAAE;QAChC,MAAMsB,MAGH,GAAG,EAAE;QACR,KAAK,MAAM;UAAE1B,GAAG;UAAEH;QAAM,CAAC,IAAIO,aAAa,CAACuB,MAAM,EAAE;UACjDD,MAAM,CAACE,IAAI,CAAC;YAAE5B,GAAG;YAAEH;UAAM,CAAC,CAAC;QAC7B;QACA,OAAO6B,MAAM;MACf;IAEA,KAAK,6BAA6B;MAAE;QAClC,MAAMA,MAGH,GAAG,EAAE;QACR,KAAK,MAAM;UAAE1B,GAAG;UAAE2B;QAAO,CAAC,IAAIvB,aAAa,CAACuB,MAAM,EAAE;UAClD,MAAME,SAAqC,GAAG,EAAE;UAChD,KAAK,MAAM;YAAE7B,GAAG,EAAE8B,MAAM;YAAEjC;UAAM,CAAC,IAAI8B,MAAM,EAAE;YAC3CE,SAAS,CAACD,IAAI,CAAC;cAAE5B,GAAG,EAAE8B,MAAM;cAAEjC;YAAM,CAAC,CAAC;UACxC;UACA6B,MAAM,CAACE,IAAI,CAAC;YAAE5B,GAAG;YAAE2B,MAAM,EAAEE;UAAU,CAAC,CAAC;QACzC;QACA,OAAOH,MAAM;MACf;EACF;EAEA,OAAOtB,aAAa;AACtB;AAEA,eAAeX,sBAAsBA,CACnCsC,QAAiC,EACjCrD,MAAqB,EAC8B;EACnD,MAAMgD,MAAM,GAAG,IAAIM,GAAG,CAAsC,CAAC;EAC7D,QAAQD,QAAQ,CAAClB,IAAI;IACnB,KAAK,WAAW;MAAE;QAChB,MAAMoB,SAAS,GAAG,MAAMvD,MAAM,CAACI,gBAAgB,CAACoD,mBAAmB,CACjEH,QAAQ,CAACZ,SACX,CAAC;QACDO,MAAM,CAACZ,GAAG,CAACiB,QAAQ,CAACZ,SAAS,EAAEc,SAAS,CAAC;QACzC;MACF;IACA,KAAK,QAAQ;MAAE;QACb,MAAMA,SAAS,GAAG,MAAMvD,MAAM,CAACI,gBAAgB,CAACoD,mBAAmB,CACjEH,QAAQ,CAACd,MACX,CAAC;QACDS,MAAM,CAACZ,GAAG,CAACiB,QAAQ,CAACd,MAAM,EAAEgB,SAAS,CAAC;QACtC;MACF;IAEA,KAAK,KAAK;MAAE;QACV,OAAOxC,sBAAsB,CAACsC,QAAQ,CAACjB,GAAG,EAAEpC,MAAM,CAAC;MACrD;IAEA,KAAK,QAAQ;MAAE;QACb,KAAK,MAAMmB,KAAK,IAAII,MAAM,CAACkC,MAAM,CAACJ,QAAQ,CAACP,MAAM,CAAC,EAAE;UAClD,KACE,MAAM,CAACX,IAAI,EAAEoB,SAAS,CAAC,IAAI,MAAMxC,sBAAsB,CAACI,KAAK,EAAEnB,MAAM,CAAC,EACtE;YACAgD,MAAM,CAACZ,GAAG,CAACD,IAAI,EAAEoB,SAAS,CAAC;UAC7B;QACF;QACA;MACF;IACA,KAAK,YAAY;IACjB,KAAK,SAAS;IACd,KAAK,MAAM;IACX,KAAK,QAAQ;IACb,KAAK,OAAO;IACZ,KAAK,SAAS;IACd,KAAK,MAAM;IACX,KAAK,QAAQ;IACb,KAAK,6BAA6B;IAClC,KAAK,WAAW;IAChB,KAAK,2BAA2B;IAChC,KAAK,OAAO;MACV;EACJ;EAEA,OAAOP,MAAM;AACf;AACA,SAASD,kBAAkBA,CAACM,QAAiC,EAAE;EAC7D,QAAQA,QAAQ,CAAClB,IAAI;IACnB,KAAK,SAAS;IACd,KAAK,MAAM;IACX,KAAK,QAAQ;IACb,KAAK,OAAO;IACZ,KAAK,SAAS;IACd,KAAK,MAAM;IACX,KAAK,QAAQ;IACb,KAAK,WAAW;MACd,OAAO,KAAK;IAEd,KAAK,OAAO;MACV,OAAO,IAAI;IAEb,KAAK,QAAQ;MACX,OAAOZ,MAAM,CAACkC,MAAM,CAACJ,QAAQ,CAACP,MAAM,CAAC,CAACY,IAAI,CAACX,kBAAkB,CAAC;IAEhE,KAAK,KAAK;MACR,OAAOA,kBAAkB,CAACM,QAAQ,CAACjB,GAAG,CAAC;IAEzC,KAAK,YAAY;IACjB,KAAK,WAAW;IAChB,KAAK,2BAA2B;IAChC,KAAK,6BAA6B;IAClC,KAAK,QAAQ;MACX,OAAO,IAAI;IAEb;MACE,OAAO,KAAK;EAChB;AACF;AAEA,OAAO,SAASI,yBAAyBA,CAGvCmB,UAA6B,EAC7BJ,SAAY,EACC;EACb,OAAO;IACLK,QAAQ,EAAEL,SAAS,CAACjD,OAAO;IAC3BuD,MAAM,EAAEhC,SAAS;IACjBiC,WAAW,EAAEP,SAAS,CAACjD,OAAO;IAC9ByD,WAAW,EAAEJ;EACf,CAAC;AACH","ignoreList":[]}
+{"version":3,"file":"applyQuery.js","names":["OntologiesV2","invariant","createObjectSpecifierFromPrimaryKey","createObjectSet","hydrateAttachmentFromRidInternal","addUserAgentAndRequestContextHeaders","augmentRequestContext","toDataValueQueries","applyQuery","client","query","params","qd","ontologyProvider","getQueryDefinition","apiName","response","Queries","execute","finalMethodCall","ontologyRid","parameters","remapQueryParams","objectOutputDefs","getRequiredDefinitions","output","remappedResponse","remapQueryResponse","value","paramTypes","parameterMap","key","Object","entries","responseDataType","responseValue","definitions","nullable","undefined","Error","multiplicity","withoutMultiplicity","i","length","type","set","def","get","object","createQueryObjectResponse","objectSet","objectSets","objectType","reference","subtype","struct","requiresConversion","map","Array","isArray","process","env","NODE_ENV","entry","keyType","getObjectSpecifier","valueType","result","groups","push","subResult","subKey","dataType","Map","objectDef","getObjectDefinition","values","some","primaryKey","objectTypeApiName","$apiName","$title","$objectType","$primaryKey","$objectSpecifier"],"sources":["applyQuery.ts"],"sourcesContent":["/*\n * Copyright 2024 Palantir Technologies, Inc. All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport type {\n  AllowedBucketKeyTypes,\n  AllowedBucketTypes,\n  CompileTimeMetadata,\n  ObjectOrInterfaceDefinition,\n  ObjectTypeDefinition,\n  OsdkBase,\n  PrimaryKeyType,\n  QueryDataTypeDefinition,\n  QueryDefinition,\n  QueryParameterDefinition,\n} from \"@osdk/api\";\nimport type { DataValue } from \"@osdk/foundry.ontologies\";\nimport * as OntologiesV2 from \"@osdk/foundry.ontologies\";\nimport invariant from \"tiny-invariant\";\nimport type { MinimalClient } from \"../MinimalClientContext.js\";\nimport { createObjectSpecifierFromPrimaryKey } from \"../object/createObjectSpecifierFromPrimaryKey.js\";\nimport { createObjectSet } from \"../objectSet/createObjectSet.js\";\nimport { hydrateAttachmentFromRidInternal } from \"../public-utils/hydrateAttachmentFromRid.js\";\nimport { addUserAgentAndRequestContextHeaders } from \"../util/addUserAgentAndRequestContextHeaders.js\";\nimport { augmentRequestContext } from \"../util/augmentRequestContext.js\";\nimport { toDataValueQueries } from \"../util/toDataValueQueries.js\";\nimport type { QueryParameterType, QueryReturnType } from \"./types.js\";\n\nexport async function applyQuery<\n  QD extends QueryDefinition<any>,\n  P extends QueryParameterType<CompileTimeMetadata<QD>[\"parameters\"]>,\n>(\n  client: MinimalClient,\n  query: QD,\n  params?: P,\n): Promise<\n  QueryReturnType<CompileTimeMetadata<QD>[\"output\"]>\n> {\n  const qd = await client.ontologyProvider.getQueryDefinition(query.apiName);\n\n  const response = await OntologiesV2.Queries.execute(\n    addUserAgentAndRequestContextHeaders(\n      augmentRequestContext(client, _ => ({ finalMethodCall: \"applyQuery\" })),\n      query,\n    ),\n    await client.ontologyRid,\n    query.apiName,\n    {\n      parameters: params\n        ? await remapQueryParams(\n          params as { [parameterId: string]: any },\n          client,\n          qd.parameters,\n        )\n        : {},\n    },\n  );\n  const objectOutputDefs = await getRequiredDefinitions(qd.output, client);\n  const remappedResponse = await remapQueryResponse(\n    client,\n    qd.output,\n    response.value,\n    objectOutputDefs,\n  );\n  return remappedResponse as QueryReturnType<CompileTimeMetadata<QD>[\"output\"]>;\n}\n\nasync function remapQueryParams(\n  params: { [parameterId: string]: any },\n  client: MinimalClient,\n  paramTypes: Record<string, QueryParameterDefinition<any>>,\n): Promise<{ [parameterId: string]: any }> {\n  const parameterMap: { [parameterName: string]: unknown } = {};\n  for (const [key, value] of Object.entries(params)) {\n    parameterMap[key] = await toDataValueQueries(\n      value,\n      client,\n      paramTypes[key],\n    );\n  }\n  return parameterMap;\n}\n\nasync function remapQueryResponse<\n  Q extends ObjectTypeDefinition,\n  T extends QueryDataTypeDefinition<Q | never>,\n>(\n  client: MinimalClient,\n  responseDataType: T,\n  responseValue: DataValue,\n  definitions: Map<string, ObjectOrInterfaceDefinition>,\n): Promise<QueryReturnType<T>> {\n  // handle null responses\n  if (responseValue == null) {\n    if (responseDataType.nullable) {\n      return undefined as unknown as QueryReturnType<typeof responseDataType>;\n    } else {\n      throw new Error(\"Got null response when nullable was not allowed\");\n    }\n  }\n\n  if (\n    responseDataType.multiplicity != null\n    && responseDataType.multiplicity\n  ) {\n    const withoutMultiplicity = { ...responseDataType, multiplicity: false };\n    for (let i = 0; i < responseValue.length; i++) {\n      responseValue[i] = await remapQueryResponse(\n        client,\n        withoutMultiplicity,\n        responseValue[i],\n        definitions,\n      );\n    }\n    return responseValue as QueryReturnType<typeof responseDataType>;\n  }\n\n  switch (responseDataType.type) {\n    case \"union\": {\n      throw new Error(\"Union return types are not yet supported\");\n    }\n\n    case \"set\": {\n      for (let i = 0; i < responseValue.length; i++) {\n        responseValue[i] = await remapQueryResponse(\n          client,\n          responseDataType.set,\n          responseValue[i],\n          definitions,\n        );\n      }\n\n      return responseValue as QueryReturnType<typeof responseDataType>;\n    }\n\n    case \"attachment\": {\n      return hydrateAttachmentFromRidInternal(\n        client,\n        responseValue,\n      ) as QueryReturnType<\n        typeof responseDataType\n      >;\n    }\n    case \"object\": {\n      const def = definitions.get(responseDataType.object);\n      if (!def) {\n        throw new Error(\n          `Missing definition for ${responseDataType.object}`,\n        );\n      }\n      return createQueryObjectResponse(\n        responseValue,\n        def,\n      ) as QueryReturnType<\n        typeof responseDataType\n      >;\n    }\n\n    case \"objectSet\": {\n      const def = definitions.get(responseDataType.objectSet);\n      if (!def) {\n        throw new Error(\n          `Missing definition for ${responseDataType.objectSet}`,\n        );\n      }\n      if (typeof responseValue === \"string\") {\n        return createObjectSet(def, client, {\n          type: \"intersect\",\n          objectSets: [\n            { type: \"base\", objectType: responseDataType.objectSet },\n            { type: \"reference\", reference: responseValue },\n          ],\n        }) as QueryReturnType<typeof responseDataType>;\n      }\n\n      return createObjectSet(\n        def,\n        client,\n        responseValue,\n      ) as QueryReturnType<\n        typeof responseDataType\n      >;\n    }\n    case \"struct\": {\n      // figure out what keys need to be fixed up\n      for (const [key, subtype] of Object.entries(responseDataType.struct)) {\n        if (requiresConversion(subtype)) {\n          responseValue[key] = await remapQueryResponse(\n            client,\n            subtype,\n            responseValue[key],\n            definitions,\n          );\n        }\n      }\n\n      return responseValue as QueryReturnType<typeof responseDataType>;\n    }\n\n    case \"map\": {\n      const map = {} as any;\n\n      invariant(Array.isArray(responseValue), \"Expected array entry\");\n      for (const entry of responseValue) {\n        invariant(entry.key, \"Expected key\");\n        invariant(entry.value, \"Expected value\");\n        const key = responseDataType.keyType.type === \"object\"\n          ? getObjectSpecifier(\n            entry.key,\n            responseDataType.keyType.object,\n            definitions,\n          )\n          : entry.key;\n        const value = await remapQueryResponse(\n          client,\n          responseDataType.valueType,\n          entry.value,\n          definitions,\n        );\n        map[key] = value;\n      }\n      return map;\n    }\n\n    case \"twoDimensionalAggregation\": {\n      const result: {\n        key: AllowedBucketKeyTypes;\n        value: AllowedBucketTypes;\n      }[] = [];\n      for (const { key, value } of responseValue.groups) {\n        result.push({ key, value });\n      }\n      return result as QueryReturnType<typeof responseDataType>;\n    }\n\n    case \"threeDimensionalAggregation\": {\n      const result: {\n        key: AllowedBucketKeyTypes;\n        groups: { key: AllowedBucketKeyTypes; value: AllowedBucketTypes }[];\n      }[] = [];\n      for (const { key, groups } of responseValue.groups) {\n        const subResult: { key: any; value: any }[] = [];\n        for (const { key: subKey, value } of groups) {\n          subResult.push({ key: subKey, value });\n        }\n        result.push({ key, groups: subResult });\n      }\n      return result as QueryReturnType<typeof responseDataType>;\n    }\n  }\n\n  return responseValue as QueryReturnType<typeof responseDataType>;\n}\n\nasync function getRequiredDefinitions(\n  dataType: QueryDataTypeDefinition,\n  client: MinimalClient,\n): Promise<Map<string, ObjectOrInterfaceDefinition>> {\n  const result = new Map<string, ObjectOrInterfaceDefinition>();\n  switch (dataType.type) {\n    case \"objectSet\": {\n      const objectDef = await client.ontologyProvider.getObjectDefinition(\n        dataType.objectSet,\n      );\n      result.set(dataType.objectSet, objectDef);\n      break;\n    }\n    case \"object\": {\n      const objectDef = await client.ontologyProvider.getObjectDefinition(\n        dataType.object,\n      );\n      result.set(dataType.object, objectDef);\n      break;\n    }\n\n    case \"set\": {\n      return getRequiredDefinitions(dataType.set, client);\n    }\n\n    case \"map\": {\n      for (const value of [dataType.keyType, dataType.valueType]) {\n        for (\n          const [type, objectDef] of await getRequiredDefinitions(value, client)\n        ) {\n          result.set(type, objectDef);\n        }\n      }\n      break;\n    }\n\n    case \"struct\": {\n      for (const value of Object.values(dataType.struct)) {\n        for (\n          const [type, objectDef] of await getRequiredDefinitions(value, client)\n        ) {\n          result.set(type, objectDef);\n        }\n      }\n      break;\n    }\n    case \"attachment\":\n    case \"boolean\":\n    case \"date\":\n    case \"double\":\n    case \"float\":\n    case \"integer\":\n    case \"long\":\n    case \"string\":\n    case \"threeDimensionalAggregation\":\n    case \"timestamp\":\n    case \"twoDimensionalAggregation\":\n    case \"union\":\n      break;\n  }\n\n  return result;\n}\nfunction requiresConversion(dataType: QueryDataTypeDefinition) {\n  switch (dataType.type) {\n    case \"boolean\":\n    case \"date\":\n    case \"double\":\n    case \"float\":\n    case \"integer\":\n    case \"long\":\n    case \"string\":\n    case \"timestamp\":\n      return false;\n\n    case \"union\":\n      return true;\n\n    case \"struct\":\n      return Object.values(dataType.struct).some(requiresConversion);\n\n    case \"set\":\n      return requiresConversion(dataType.set);\n\n    case \"attachment\":\n    case \"objectSet\":\n    case \"twoDimensionalAggregation\":\n    case \"threeDimensionalAggregation\":\n    case \"object\":\n      return true;\n\n    default:\n      return false;\n  }\n}\n\nfunction getObjectSpecifier(\n  primaryKey: any,\n  objectTypeApiName: string,\n  definitions: Map<string, ObjectOrInterfaceDefinition>,\n): string {\n  const def = definitions.get(objectTypeApiName);\n  if (!def) {\n    throw new Error(\n      `Missing definition for ${objectTypeApiName}`,\n    );\n  }\n  return createObjectSpecifierFromPrimaryKey(\n    def,\n    primaryKey,\n  );\n}\n\nexport function createQueryObjectResponse<\n  Q extends ObjectOrInterfaceDefinition,\n>(\n  primaryKey: PrimaryKeyType<Q>,\n  objectDef: Q,\n): OsdkBase<Q> {\n  return {\n    $apiName: objectDef.apiName,\n    $title: undefined,\n    $objectType: objectDef.apiName,\n    $primaryKey: primaryKey,\n    $objectSpecifier: createObjectSpecifierFromPrimaryKey(\n      objectDef,\n      primaryKey,\n    ),\n  };\n}\n"],"mappings":"AAAA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;;AAeA,OAAO,KAAKA,YAAY,MAAM,0BAA0B;AACxD,OAAOC,SAAS,MAAM,gBAAgB;AAEtC,SAASC,mCAAmC,QAAQ,kDAAkD;AACtG,SAASC,eAAe,QAAQ,iCAAiC;AACjE,SAASC,gCAAgC,QAAQ,6CAA6C;AAC9F,SAASC,oCAAoC,QAAQ,iDAAiD;AACtG,SAASC,qBAAqB,QAAQ,kCAAkC;AACxE,SAASC,kBAAkB,QAAQ,+BAA+B;AAGlE,OAAO,eAAeC,UAAUA,CAI9BC,MAAqB,EACrBC,KAAS,EACTC,MAAU,EAGV;EACA,MAAMC,EAAE,GAAG,MAAMH,MAAM,CAACI,gBAAgB,CAACC,kBAAkB,CAACJ,KAAK,CAACK,OAAO,CAAC;EAE1E,MAAMC,QAAQ,GAAG,MAAMhB,YAAY,CAACiB,OAAO,CAACC,OAAO,CACjDb,oCAAoC,CAClCC,qBAAqB,CAACG,MAAM,EAAE,OAAM;IAAEU,eAAe,EAAE;EAAa,CAAC,CAAC,CAAC,EACvET,KACF,CAAC,EACD,MAAMD,MAAM,CAACW,WAAW,EACxBV,KAAK,CAACK,OAAO,EACb;IACEM,UAAU,EAAEV,MAAM,GACd,MAAMW,gBAAgB,CACtBX,MAAM,EACNF,MAAM,EACNG,EAAE,CAACS,UACL,CAAC,GACC,CAAC;EACP,CACF,CAAC;EACD,MAAME,gBAAgB,GAAG,MAAMC,sBAAsB,CAACZ,EAAE,CAACa,MAAM,EAAEhB,MAAM,CAAC;EACxE,MAAMiB,gBAAgB,GAAG,MAAMC,kBAAkB,CAC/ClB,MAAM,EACNG,EAAE,CAACa,MAAM,EACTT,QAAQ,CAACY,KAAK,EACdL,gBACF,CAAC;EACD,OAAOG,gBAAgB;AACzB;AAEA,eAAeJ,gBAAgBA,CAC7BX,MAAsC,EACtCF,MAAqB,EACrBoB,UAAyD,EAChB;EACzC,MAAMC,YAAkD,GAAG,CAAC,CAAC;EAC7D,KAAK,MAAM,CAACC,GAAG,EAAEH,KAAK,CAAC,IAAII,MAAM,CAACC,OAAO,CAACtB,MAAM,CAAC,EAAE;IACjDmB,YAAY,CAACC,GAAG,CAAC,GAAG,MAAMxB,kBAAkB,CAC1CqB,KAAK,EACLnB,MAAM,EACNoB,UAAU,CAACE,GAAG,CAChB,CAAC;EACH;EACA,OAAOD,YAAY;AACrB;AAEA,eAAeH,kBAAkBA,CAI/BlB,MAAqB,EACrByB,gBAAmB,EACnBC,aAAwB,EACxBC,WAAqD,EACxB;EAC7B;EACA,IAAID,aAAa,IAAI,IAAI,EAAE;IACzB,IAAID,gBAAgB,CAACG,QAAQ,EAAE;MAC7B,OAAOC,SAAS;IAClB,CAAC,MAAM;MACL,MAAM,IAAIC,KAAK,CAAC,iDAAiD,CAAC;IACpE;EACF;EAEA,IACEL,gBAAgB,CAACM,YAAY,IAAI,IAAI,IAClCN,gBAAgB,CAACM,YAAY,EAChC;IACA,MAAMC,mBAAmB,GAAG;MAAE,GAAGP,gBAAgB;MAAEM,YAAY,EAAE;IAAM,CAAC;IACxE,KAAK,IAAIE,CAAC,GAAG,CAAC,EAAEA,CAAC,GAAGP,aAAa,CAACQ,MAAM,EAAED,CAAC,EAAE,EAAE;MAC7CP,aAAa,CAACO,CAAC,CAAC,GAAG,MAAMf,kBAAkB,CACzClB,MAAM,EACNgC,mBAAmB,EACnBN,aAAa,CAACO,CAAC,CAAC,EAChBN,WACF,CAAC;IACH;IACA,OAAOD,aAAa;EACtB;EAEA,QAAQD,gBAAgB,CAACU,IAAI;IAC3B,KAAK,OAAO;MAAE;QACZ,MAAM,IAAIL,KAAK,CAAC,0CAA0C,CAAC;MAC7D;IAEA,KAAK,KAAK;MAAE;QACV,KAAK,IAAIG,CAAC,GAAG,CAAC,EAAEA,CAAC,GAAGP,aAAa,CAACQ,MAAM,EAAED,CAAC,EAAE,EAAE;UAC7CP,aAAa,CAACO,CAAC,CAAC,GAAG,MAAMf,kBAAkB,CACzClB,MAAM,EACNyB,gBAAgB,CAACW,GAAG,EACpBV,aAAa,CAACO,CAAC,CAAC,EAChBN,WACF,CAAC;QACH;QAEA,OAAOD,aAAa;MACtB;IAEA,KAAK,YAAY;MAAE;QACjB,OAAO/B,gCAAgC,CACrCK,MAAM,EACN0B,aACF,CAAC;MAGH;IACA,KAAK,QAAQ;MAAE;QACb,MAAMW,GAAG,GAAGV,WAAW,CAACW,GAAG,CAACb,gBAAgB,CAACc,MAAM,CAAC;QACpD,IAAI,CAACF,GAAG,EAAE;UACR,MAAM,IAAIP,KAAK,CACb,0BAA0BL,gBAAgB,CAACc,MAAM,EACnD,CAAC;QACH;QACA,OAAOC,yBAAyB,CAC9Bd,aAAa,EACbW,GACF,CAAC;MAGH;IAEA,KAAK,WAAW;MAAE;QAChB,MAAMA,GAAG,GAAGV,WAAW,CAACW,GAAG,CAACb,gBAAgB,CAACgB,SAAS,CAAC;QACvD,IAAI,CAACJ,GAAG,EAAE;UACR,MAAM,IAAIP,KAAK,CACb,0BAA0BL,gBAAgB,CAACgB,SAAS,EACtD,CAAC;QACH;QACA,IAAI,OAAOf,aAAa,KAAK,QAAQ,EAAE;UACrC,OAAOhC,eAAe,CAAC2C,GAAG,EAAErC,MAAM,EAAE;YAClCmC,IAAI,EAAE,WAAW;YACjBO,UAAU,EAAE,CACV;cAAEP,IAAI,EAAE,MAAM;cAAEQ,UAAU,EAAElB,gBAAgB,CAACgB;YAAU,CAAC,EACxD;cAAEN,IAAI,EAAE,WAAW;cAAES,SAAS,EAAElB;YAAc,CAAC;UAEnD,CAAC,CAAC;QACJ;QAEA,OAAOhC,eAAe,CACpB2C,GAAG,EACHrC,MAAM,EACN0B,aACF,CAAC;MAGH;IACA,KAAK,QAAQ;MAAE;QACb;QACA,KAAK,MAAM,CAACJ,GAAG,EAAEuB,OAAO,CAAC,IAAItB,MAAM,CAACC,OAAO,CAACC,gBAAgB,CAACqB,MAAM,CAAC,EAAE;UACpE,IAAIC,kBAAkB,CAACF,OAAO,CAAC,EAAE;YAC/BnB,aAAa,CAACJ,GAAG,CAAC,GAAG,MAAMJ,kBAAkB,CAC3ClB,MAAM,EACN6C,OAAO,EACPnB,aAAa,CAACJ,GAAG,CAAC,EAClBK,WACF,CAAC;UACH;QACF;QAEA,OAAOD,aAAa;MACtB;IAEA,KAAK,KAAK;MAAE;QACV,MAAMsB,GAAG,GAAG,CAAC,CAAQ;QAErB,CAAUC,KAAK,CAACC,OAAO,CAACxB,aAAa,CAAC,GAAAyB,OAAA,CAAAC,GAAA,CAAAC,QAAA,oBAAtC7D,SAAS,QAA+B,sBAAsB,IAA9DA,SAAS;QACT,KAAK,MAAM8D,KAAK,IAAI5B,aAAa,EAAE;UACjC,CAAU4B,KAAK,CAAChC,GAAG,GAAA6B,OAAA,CAAAC,GAAA,CAAAC,QAAA,oBAAnB7D,SAAS,QAAY,cAAc,IAAnCA,SAAS;UACT,CAAU8D,KAAK,CAACnC,KAAK,GAAAgC,OAAA,CAAAC,GAAA,CAAAC,QAAA,oBAArB7D,SAAS,QAAc,gBAAgB,IAAvCA,SAAS;UACT,MAAM8B,GAAG,GAAGG,gBAAgB,CAAC8B,OAAO,CAACpB,IAAI,KAAK,QAAQ,GAClDqB,kBAAkB,CAClBF,KAAK,CAAChC,GAAG,EACTG,gBAAgB,CAAC8B,OAAO,CAAChB,MAAM,EAC/BZ,WACF,CAAC,GACC2B,KAAK,CAAChC,GAAG;UACb,MAAMH,KAAK,GAAG,MAAMD,kBAAkB,CACpClB,MAAM,EACNyB,gBAAgB,CAACgC,SAAS,EAC1BH,KAAK,CAACnC,KAAK,EACXQ,WACF,CAAC;UACDqB,GAAG,CAAC1B,GAAG,CAAC,GAAGH,KAAK;QAClB;QACA,OAAO6B,GAAG;MACZ;IAEA,KAAK,2BAA2B;MAAE;QAChC,MAAMU,MAGH,GAAG,EAAE;QACR,KAAK,MAAM;UAAEpC,GAAG;UAAEH;QAAM,CAAC,IAAIO,aAAa,CAACiC,MAAM,EAAE;UACjDD,MAAM,CAACE,IAAI,CAAC;YAAEtC,GAAG;YAAEH;UAAM,CAAC,CAAC;QAC7B;QACA,OAAOuC,MAAM;MACf;IAEA,KAAK,6BAA6B;MAAE;QAClC,MAAMA,MAGH,GAAG,EAAE;QACR,KAAK,MAAM;UAAEpC,GAAG;UAAEqC;QAAO,CAAC,IAAIjC,aAAa,CAACiC,MAAM,EAAE;UAClD,MAAME,SAAqC,GAAG,EAAE;UAChD,KAAK,MAAM;YAAEvC,GAAG,EAAEwC,MAAM;YAAE3C;UAAM,CAAC,IAAIwC,MAAM,EAAE;YAC3CE,SAAS,CAACD,IAAI,CAAC;cAAEtC,GAAG,EAAEwC,MAAM;cAAE3C;YAAM,CAAC,CAAC;UACxC;UACAuC,MAAM,CAACE,IAAI,CAAC;YAAEtC,GAAG;YAAEqC,MAAM,EAAEE;UAAU,CAAC,CAAC;QACzC;QACA,OAAOH,MAAM;MACf;EACF;EAEA,OAAOhC,aAAa;AACtB;AAEA,eAAeX,sBAAsBA,CACnCgD,QAAiC,EACjC/D,MAAqB,EAC8B;EACnD,MAAM0D,MAAM,GAAG,IAAIM,GAAG,CAAsC,CAAC;EAC7D,QAAQD,QAAQ,CAAC5B,IAAI;IACnB,KAAK,WAAW;MAAE;QAChB,MAAM8B,SAAS,GAAG,MAAMjE,MAAM,CAACI,gBAAgB,CAAC8D,mBAAmB,CACjEH,QAAQ,CAACtB,SACX,CAAC;QACDiB,MAAM,CAACtB,GAAG,CAAC2B,QAAQ,CAACtB,SAAS,EAAEwB,SAAS,CAAC;QACzC;MACF;IACA,KAAK,QAAQ;MAAE;QACb,MAAMA,SAAS,GAAG,MAAMjE,MAAM,CAACI,gBAAgB,CAAC8D,mBAAmB,CACjEH,QAAQ,CAACxB,MACX,CAAC;QACDmB,MAAM,CAACtB,GAAG,CAAC2B,QAAQ,CAACxB,MAAM,EAAE0B,SAAS,CAAC;QACtC;MACF;IAEA,KAAK,KAAK;MAAE;QACV,OAAOlD,sBAAsB,CAACgD,QAAQ,CAAC3B,GAAG,EAAEpC,MAAM,CAAC;MACrD;IAEA,KAAK,KAAK;MAAE;QACV,KAAK,MAAMmB,KAAK,IAAI,CAAC4C,QAAQ,CAACR,OAAO,EAAEQ,QAAQ,CAACN,SAAS,CAAC,EAAE;UAC1D,KACE,MAAM,CAACtB,IAAI,EAAE8B,SAAS,CAAC,IAAI,MAAMlD,sBAAsB,CAACI,KAAK,EAAEnB,MAAM,CAAC,EACtE;YACA0D,MAAM,CAACtB,GAAG,CAACD,IAAI,EAAE8B,SAAS,CAAC;UAC7B;QACF;QACA;MACF;IAEA,KAAK,QAAQ;MAAE;QACb,KAAK,MAAM9C,KAAK,IAAII,MAAM,CAAC4C,MAAM,CAACJ,QAAQ,CAACjB,MAAM,CAAC,EAAE;UAClD,KACE,MAAM,CAACX,IAAI,EAAE8B,SAAS,CAAC,IAAI,MAAMlD,sBAAsB,CAACI,KAAK,EAAEnB,MAAM,CAAC,EACtE;YACA0D,MAAM,CAACtB,GAAG,CAACD,IAAI,EAAE8B,SAAS,CAAC;UAC7B;QACF;QACA;MACF;IACA,KAAK,YAAY;IACjB,KAAK,SAAS;IACd,KAAK,MAAM;IACX,KAAK,QAAQ;IACb,KAAK,OAAO;IACZ,KAAK,SAAS;IACd,KAAK,MAAM;IACX,KAAK,QAAQ;IACb,KAAK,6BAA6B;IAClC,KAAK,WAAW;IAChB,KAAK,2BAA2B;IAChC,KAAK,OAAO;MACV;EACJ;EAEA,OAAOP,MAAM;AACf;AACA,SAASX,kBAAkBA,CAACgB,QAAiC,EAAE;EAC7D,QAAQA,QAAQ,CAAC5B,IAAI;IACnB,KAAK,SAAS;IACd,KAAK,MAAM;IACX,KAAK,QAAQ;IACb,KAAK,OAAO;IACZ,KAAK,SAAS;IACd,KAAK,MAAM;IACX,KAAK,QAAQ;IACb,KAAK,WAAW;MACd,OAAO,KAAK;IAEd,KAAK,OAAO;MACV,OAAO,IAAI;IAEb,KAAK,QAAQ;MACX,OAAOZ,MAAM,CAAC4C,MAAM,CAACJ,QAAQ,CAACjB,MAAM,CAAC,CAACsB,IAAI,CAACrB,kBAAkB,CAAC;IAEhE,KAAK,KAAK;MACR,OAAOA,kBAAkB,CAACgB,QAAQ,CAAC3B,GAAG,CAAC;IAEzC,KAAK,YAAY;IACjB,KAAK,WAAW;IAChB,KAAK,2BAA2B;IAChC,KAAK,6BAA6B;IAClC,KAAK,QAAQ;MACX,OAAO,IAAI;IAEb;MACE,OAAO,KAAK;EAChB;AACF;AAEA,SAASoB,kBAAkBA,CACzBa,UAAe,EACfC,iBAAyB,EACzB3C,WAAqD,EAC7C;EACR,MAAMU,GAAG,GAAGV,WAAW,CAACW,GAAG,CAACgC,iBAAiB,CAAC;EAC9C,IAAI,CAACjC,GAAG,EAAE;IACR,MAAM,IAAIP,KAAK,CACb,0BAA0BwC,iBAAiB,EAC7C,CAAC;EACH;EACA,OAAO7E,mCAAmC,CACxC4C,GAAG,EACHgC,UACF,CAAC;AACH;AAEA,OAAO,SAAS7B,yBAAyBA,CAGvC6B,UAA6B,EAC7BJ,SAAY,EACC;EACb,OAAO;IACLM,QAAQ,EAAEN,SAAS,CAAC3D,OAAO;IAC3BkE,MAAM,EAAE3C,SAAS;IACjB4C,WAAW,EAAER,SAAS,CAAC3D,OAAO;IAC9BoE,WAAW,EAAEL,UAAU;IACvBM,gBAAgB,EAAElF,mCAAmC,CACnDwE,SAAS,EACTI,UACF;EACF,CAAC;AACH","ignoreList":[]}

package/build/browser/queries/queries.test.js

@@ -14,7 +14,7 @@
  * limitations under the License.
  */
 
-import { $ontologyRid, $Queries, acceptsThreeDimensionalAggregationFunction, acceptsTwoDimensionalAggregationFunction, addOne, Employee, incrementPersonAge, incrementPersonAgeComplex, queryAcceptsObject, queryAcceptsObjectSets, returnsDate, returnsTimestamp, threeDimensionalAggregationFunction, twoDimensionalAggregationFunction } from "@osdk/client.test.ontology";
+import { $ontologyRid, $Queries, acceptsThreeDimensionalAggregationFunction, acceptsTwoDimensionalAggregationFunction, addOne, Employee, incrementPersonAge, incrementPersonAgeComplex, queryAcceptsObject, queryAcceptsObjectSets, queryTypeReturnsMap, returnsDate, returnsTimestamp, threeDimensionalAggregationFunction, twoDimensionalAggregationFunction } from "@osdk/client.test.ontology";
 import { apiServer } from "@osdk/shared.test";
 import { afterAll, beforeAll, describe, expect, expectTypeOf, it } from "vitest";
 import { createClient } from "../createClient.js";
@@ -41,7 +41,8 @@ describe("queries", () => {
     expect(result).toEqual({
       $apiName: "Employee",
       $objectType: "Employee",
-      $primaryKey: 50031
+      $primaryKey: 50031,
+      $objectSpecifier: "Employee:50031"
     });
 
     // Should also accept primary keys
@@ -51,7 +52,8 @@ describe("queries", () => {
     expect(result2).toEqual({
       $apiName: "Employee",
       $objectType: "Employee",
-      $primaryKey: 50031
+      $primaryKey: 50031,
+      $objectSpecifier: "Employee:50031"
     });
   });
   it("accepts objectSets", async () => {
@@ -98,7 +100,8 @@ describe("queries", () => {
       object: {
         $apiName: "Employee",
         $objectType: "Employee",
-        $primaryKey: 50031
+        $primaryKey: 50031,
+        $objectSpecifier: "Employee:50031"
       }
     });
   });
@@ -193,6 +196,24 @@ describe("queries", () => {
       groups: []
     }]);
   });
+  it("map type request and response works", async () => {
+    client(queryTypeReturnsMap).executeFunction;
+    expectTypeOf().toMatchTypeOf();
+    const result = await client(queryTypeReturnsMap).executeFunction({
+      peopleMap: {
+        ["Employee:person1"]: "hi"
+      }
+    });
+    expect(result).toMatchInlineSnapshot(`
+      {
+        "Employee:50030": "bye",
+      }
+    `);
+    expectTypeOf().toMatchTypeOf();
+    const object = await client(Employee).fetchOne(50030);
+    const value = result[object.$objectSpecifier];
+    expect(value).toBe("bye");
+  });
   it("accepts and returns objects", async () => {
     const employeeObjectSet = client(Employee);
     await client(queryAcceptsObjectSets).executeFunction({
@@ -202,7 +223,7 @@ describe("queries", () => {
   });
   it("queries are enumerable", async () => {
     const queries = Object.keys($Queries);
-    expect(queries).toStrictEqual(["acceptsThreeDimensionalAggregationFunction", "acceptsTwoDimensionalAggregationFunction", "addOne", "incrementPersonAge", "incrementPersonAgeComplex", "queryAcceptsObject", "queryAcceptsObjectSets", "queryTypeReturnsArray", "returnsDate", "returnsObject", "returnsTimestamp", "threeDimensionalAggregationFunction", "twoDimensionalAggregationFunction"]);
+    expect(queries).toStrictEqual(["acceptsThreeDimensionalAggregationFunction", "acceptsTwoDimensionalAggregationFunction", "addOne", "incrementPersonAge", "incrementPersonAgeComplex", "queryAcceptsObject", "queryAcceptsObjectSets", "queryTypeReturnsArray", "queryTypeReturnsMap", "returnsDate", "returnsObject", "returnsTimestamp", "threeDimensionalAggregationFunction", "twoDimensionalAggregationFunction"]);
   });
   it("queries work with arrays", async () => {
     const result = await client($Queries.queryTypeReturnsArray).executeFunction({

package/build/browser/queries/queries.test.js.map

@@ -1 +1 @@
-{"version":3,"file":"queries.test.js","names":["$ontologyRid","$Queries","acceptsThreeDimensionalAggregationFunction","acceptsTwoDimensionalAggregationFunction","addOne","Employee","incrementPersonAge","incrementPersonAgeComplex","queryAcceptsObject","queryAcceptsObjectSets","returnsDate","returnsTimestamp","threeDimensionalAggregationFunction","twoDimensionalAggregationFunction","apiServer","afterAll","beforeAll","describe","expect","expectTypeOf","it","createClient","client","listen","close","result","executeFunction","n","toBe","employee","fetchOne","object","toEqual","$apiName","$objectType","$primaryKey","result2","employeeObjectSet","objectSet","toMatchTypeOf","resultWithTimestamp","resultWithDate","person","firstName","lastName","age","toEqualTypeOf","key","value","aggFunction","groups","startValue","endValue","fail","e","message","toMatch","queries","Object","keys","toStrictEqual","queryTypeReturnsArray","people"],"sources":["queries.test.ts"],"sourcesContent":["/*\n * Copyright 2023 Palantir Technologies, Inc. All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport type { ObjectSet } from \"@osdk/api\";\nimport {\n  $ontologyRid,\n  $Queries,\n  acceptsThreeDimensionalAggregationFunction,\n  acceptsTwoDimensionalAggregationFunction,\n  addOne,\n  Employee,\n  incrementPersonAge,\n  incrementPersonAgeComplex,\n  queryAcceptsObject,\n  queryAcceptsObjectSets,\n  returnsDate,\n  returnsTimestamp,\n  threeDimensionalAggregationFunction,\n  twoDimensionalAggregationFunction,\n} from \"@osdk/client.test.ontology\";\nimport { apiServer } from \"@osdk/shared.test\";\nimport {\n  afterAll,\n  beforeAll,\n  describe,\n  expect,\n  expectTypeOf,\n  it,\n} from \"vitest\";\nimport type { Client } from \"../Client.js\";\nimport { createClient } from \"../createClient.js\";\n\ndescribe(\"queries\", () => {\n  let client: Client;\n\n  beforeAll(async () => {\n    apiServer.listen();\n    client = createClient(\n      \"https://stack.palantir.com\",\n      $ontologyRid,\n      async () => \"myAccessToken\",\n    );\n  });\n\n  afterAll(() => {\n    apiServer.close();\n  });\n\n  it(\"simple query works\", async () => {\n    const result = await client(addOne).executeFunction({ n: 2 });\n    expect(result).toBe(3);\n  });\n\n  it(\"accepts objects\", async () => {\n    const employee = await client(Employee).fetchOne(\n      50030,\n    );\n    const result = await client(queryAcceptsObject).executeFunction({\n      object: employee,\n    });\n    expect(result).toEqual({\n      $apiName: \"Employee\",\n      $objectType: \"Employee\",\n      $primaryKey: 50031,\n    });\n\n    // Should also accept primary keys\n    const result2 = await client(queryAcceptsObject).executeFunction({\n      object: 50030,\n    });\n    expect(result2).toEqual({\n      $apiName: \"Employee\",\n      $objectType: \"Employee\",\n      $primaryKey: 50031,\n    });\n  });\n\n  it(\"accepts objectSets\", async () => {\n    const employeeObjectSet = client(Employee);\n    const result = await client(queryAcceptsObjectSets).executeFunction({\n      objectSet: employeeObjectSet,\n    });\n\n    expectTypeOf<ObjectSet<Employee>>().toMatchTypeOf<typeof result>();\n  });\n\n  it(\"no params work\", async () => {\n    const resultWithTimestamp = await client(returnsTimestamp)\n      .executeFunction();\n    expect(resultWithTimestamp).toBe(\"2019-01-01T00:00:00.000Z\");\n\n    const resultWithDate = await client(returnsDate).executeFunction();\n    expect(resultWithDate).toBe(\"2019-01-01\");\n  });\n\n  it(\"returns and accepts structs property\", async () => {\n    const result = await client(incrementPersonAge).executeFunction({\n      person: { firstName: \"John\", lastName: \"Doe\", age: 42 },\n    });\n    expect(result).toEqual({\n      firstName: \"John\",\n      lastName: \"Doe\",\n      age: 43,\n    });\n  });\n\n  it(\"returns and accepts complex structs property\", async () => {\n    const employee = await client(Employee).fetchOne(\n      50030,\n    );\n    const result = await client(incrementPersonAgeComplex).executeFunction({\n      person: { firstName: \"John\", lastName: \"Doe\", age: 42, object: employee },\n    });\n    expect(result).toEqual({\n      firstName: \"John\",\n      lastName: \"Doe\",\n      age: 43,\n      object: {\n        $apiName: \"Employee\",\n        $objectType: \"Employee\",\n        $primaryKey: 50031,\n      },\n    });\n  });\n\n  it(\"two dimensional aggregation response works\", async () => {\n    const result = await client(twoDimensionalAggregationFunction)\n      .executeFunction();\n    expectTypeOf<typeof result>().toEqualTypeOf<\n      { key: string; value: number }[]\n    >;\n    expect(result).toEqual([{ key: \"Q-AFN\", value: 1 }, {\n      key: \"Q-AFO\",\n      value: 2,\n    }]);\n  });\n\n  it(\"two dimensional aggregation request/response works\", async () => {\n    const clientBoundQueryFunction =\n      client(acceptsTwoDimensionalAggregationFunction).executeFunction;\n    type InferredParamType = Parameters<\n      typeof clientBoundQueryFunction\n    >[0];\n\n    expectTypeOf<{ aggFunction: { key: string; value: number }[] }>()\n      .toMatchTypeOf<InferredParamType>();\n\n    const result = await client(acceptsTwoDimensionalAggregationFunction)\n      .executeFunction({\n        aggFunction: [\n          {\n            key: \"testKey1\",\n            value: 1,\n          },\n          {\n            key: \"testKey2\",\n            value: 2,\n          },\n        ],\n      });\n    expectTypeOf<typeof result>().toEqualTypeOf<\n      { key: string; value: number }[]\n    >;\n\n    expect(result).toEqual([{ key: \"responseKey1\", value: 3 }, {\n      key: \"responseKey2\",\n      value: 4,\n    }]);\n  });\n\n  it(\"three dimensional aggregation response works\", async () => {\n    const result = await client(threeDimensionalAggregationFunction)\n      .executeFunction();\n\n    expectTypeOf<typeof result>().toEqualTypeOf<\n      {\n        key: string;\n        groups: {\n          key: { startValue: string | undefined; endValue: string | undefined };\n          value: number;\n        }[];\n      }[]\n    >;\n    expect(result).toEqual([{\n      key: \"Q-AFN\",\n      groups: [{\n        key: {\n          startValue: \"2010-10-01T00:00:00Z\",\n          endValue: \"2010-10-02T00:00:00Z\",\n        },\n        value: 65.0,\n      }],\n    }, { key: \"Q-AFO\", groups: [] }]);\n  });\n\n  it(\"throws when response is null and response is non-nullable\", async () => {\n    try {\n      const result = await client(addOne).executeFunction({ n: 3 });\n      expect.fail(\"Should not reach here\");\n    } catch (e) {\n      expect((e as Error).message).toMatch(\n        `Got null response when nullable was not allowed`,\n      );\n    }\n  });\n\n  it(\"three dimensional aggregation request/response works\", async () => {\n    const clientBoundQueryFunction =\n      client(acceptsThreeDimensionalAggregationFunction).executeFunction;\n    type InferredParamType = Parameters<\n      typeof clientBoundQueryFunction\n    >[0];\n\n    expectTypeOf<{\n      aggFunction: {\n        key: string;\n        groups: {\n          key: { startValue: string | undefined; endValue: string };\n          value: number;\n        }[];\n      }[];\n    }>()\n      .toMatchTypeOf<InferredParamType>();\n\n    const result = await client(acceptsThreeDimensionalAggregationFunction)\n      .executeFunction({\n        aggFunction: [\n          {\n            key: \"testKey1\",\n            groups: [\n              {\n                key: {\n                  startValue: \"2010-10-01T00:00:00Z\",\n                  endValue: \"2010-10-02T00:00:00Z\",\n                },\n                value: 65.0,\n              },\n            ],\n          },\n          {\n            key: \"testKey2\",\n            groups: [],\n          },\n        ],\n      });\n\n    expect(result).toEqual([\n      {\n        key: \"Q-AFN\",\n        groups: [\n          {\n            key: {\n              startValue: \"2010-10-01T00:00:00Z\",\n              endValue: \"2010-10-02T00:00:00Z\",\n            },\n            value: 65.0,\n          },\n        ],\n      },\n      {\n        key: \"Q-AFO\",\n        groups: [],\n      },\n    ]);\n  });\n\n  it(\"accepts and returns objects\", async () => {\n    const employeeObjectSet = client(Employee);\n    const result = await client(queryAcceptsObjectSets).executeFunction({\n      objectSet: employeeObjectSet,\n    });\n\n    expectTypeOf<ObjectSet<Employee>>().toMatchTypeOf<typeof result>();\n  });\n  it(\"queries are enumerable\", async () => {\n    const queries = Object.keys($Queries);\n    expect(queries).toStrictEqual([\n      \"acceptsThreeDimensionalAggregationFunction\",\n      \"acceptsTwoDimensionalAggregationFunction\",\n      \"addOne\",\n      \"incrementPersonAge\",\n      \"incrementPersonAgeComplex\",\n      \"queryAcceptsObject\",\n      \"queryAcceptsObjectSets\",\n      \"queryTypeReturnsArray\",\n      \"returnsDate\",\n      \"returnsObject\",\n      \"returnsTimestamp\",\n      \"threeDimensionalAggregationFunction\",\n      \"twoDimensionalAggregationFunction\",\n    ]);\n  });\n\n  it(\"queries work with arrays\", async () => {\n    const result = await client($Queries.queryTypeReturnsArray).executeFunction(\n      { people: [\"Brad\", \"George\", \"Ryan\"] },\n    );\n    expect(result).toEqual([\"Pitt\", \"Clooney\", \"Reynolds\"]);\n  });\n});\n"],"mappings":"AAAA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;;AAGA,SACEA,YAAY,EACZC,QAAQ,EACRC,0CAA0C,EAC1CC,wCAAwC,EACxCC,MAAM,EACNC,QAAQ,EACRC,kBAAkB,EAClBC,yBAAyB,EACzBC,kBAAkB,EAClBC,sBAAsB,EACtBC,WAAW,EACXC,gBAAgB,EAChBC,mCAAmC,EACnCC,iCAAiC,QAC5B,4BAA4B;AACnC,SAASC,SAAS,QAAQ,mBAAmB;AAC7C,SACEC,QAAQ,EACRC,SAAS,EACTC,QAAQ,EACRC,MAAM,EACNC,YAAY,EACZC,EAAE,QACG,QAAQ;AAEf,SAASC,YAAY,QAAQ,oBAAoB;AAEjDJ,QAAQ,CAAC,SAAS,EAAE,MAAM;EACxB,IAAIK,MAAc;EAElBN,SAAS,CAAC,YAAY;IACpBF,SAAS,CAACS,MAAM,CAAC,CAAC;IAClBD,MAAM,GAAGD,YAAY,CACnB,4BAA4B,EAC5BrB,YAAY,EACZ,YAAY,eACd,CAAC;EACH,CAAC,CAAC;EAEFe,QAAQ,CAAC,MAAM;IACbD,SAAS,CAACU,KAAK,CAAC,CAAC;EACnB,CAAC,CAAC;EAEFJ,EAAE,CAAC,oBAAoB,EAAE,YAAY;IACnC,MAAMK,MAAM,GAAG,MAAMH,MAAM,CAAClB,MAAM,CAAC,CAACsB,eAAe,CAAC;MAAEC,CAAC,EAAE;IAAE,CAAC,CAAC;IAC7DT,MAAM,CAACO,MAAM,CAAC,CAACG,IAAI,CAAC,CAAC,CAAC;EACxB,CAAC,CAAC;EAEFR,EAAE,CAAC,iBAAiB,EAAE,YAAY;IAChC,MAAMS,QAAQ,GAAG,MAAMP,MAAM,CAACjB,QAAQ,CAAC,CAACyB,QAAQ,CAC9C,KACF,CAAC;IACD,MAAML,MAAM,GAAG,MAAMH,MAAM,CAACd,kBAAkB,CAAC,CAACkB,eAAe,CAAC;MAC9DK,MAAM,EAAEF;IACV,CAAC,CAAC;IACFX,MAAM,CAACO,MAAM,CAAC,CAACO,OAAO,CAAC;MACrBC,QAAQ,EAAE,UAAU;MACpBC,WAAW,EAAE,UAAU;MACvBC,WAAW,EAAE;IACf,CAAC,CAAC;;IAEF;IACA,MAAMC,OAAO,GAAG,MAAMd,MAAM,CAACd,kBAAkB,CAAC,CAACkB,eAAe,CAAC;MAC/DK,MAAM,EAAE;IACV,CAAC,CAAC;IACFb,MAAM,CAACkB,OAAO,CAAC,CAACJ,OAAO,CAAC;MACtBC,QAAQ,EAAE,UAAU;MACpBC,WAAW,EAAE,UAAU;MACvBC,WAAW,EAAE;IACf,CAAC,CAAC;EACJ,CAAC,CAAC;EAEFf,EAAE,CAAC,oBAAoB,EAAE,YAAY;IACnC,MAAMiB,iBAAiB,GAAGf,MAAM,CAACjB,QAAQ,CAAC;IAC3B,MAAMiB,MAAM,CAACb,sBAAsB,CAAC,CAACiB,eAAe,CAAC;MAClEY,SAAS,EAAED;IACb,CAAC,CAAC;IAEFlB,YAAY,CAAsB,CAAC,CAACoB,aAAa,CAAgB,CAAC;EACpE,CAAC,CAAC;EAEFnB,EAAE,CAAC,gBAAgB,EAAE,YAAY;IAC/B,MAAMoB,mBAAmB,GAAG,MAAMlB,MAAM,CAACX,gBAAgB,CAAC,CACvDe,eAAe,CAAC,CAAC;IACpBR,MAAM,CAACsB,mBAAmB,CAAC,CAACZ,IAAI,CAAC,0BAA0B,CAAC;IAE5D,MAAMa,cAAc,GAAG,MAAMnB,MAAM,CAACZ,WAAW,CAAC,CAACgB,eAAe,CAAC,CAAC;IAClER,MAAM,CAACuB,cAAc,CAAC,CAACb,IAAI,CAAC,YAAY,CAAC;EAC3C,CAAC,CAAC;EAEFR,EAAE,CAAC,sCAAsC,EAAE,YAAY;IACrD,MAAMK,MAAM,GAAG,MAAMH,MAAM,CAAChB,kBAAkB,CAAC,CAACoB,eAAe,CAAC;MAC9DgB,MAAM,EAAE;QAAEC,SAAS,EAAE,MAAM;QAAEC,QAAQ,EAAE,KAAK;QAAEC,GAAG,EAAE;MAAG;IACxD,CAAC,CAAC;IACF3B,MAAM,CAACO,MAAM,CAAC,CAACO,OAAO,CAAC;MACrBW,SAAS,EAAE,MAAM;MACjBC,QAAQ,EAAE,KAAK;MACfC,GAAG,EAAE;IACP,CAAC,CAAC;EACJ,CAAC,CAAC;EAEFzB,EAAE,CAAC,8CAA8C,EAAE,YAAY;IAC7D,MAAMS,QAAQ,GAAG,MAAMP,MAAM,CAACjB,QAAQ,CAAC,CAACyB,QAAQ,CAC9C,KACF,CAAC;IACD,MAAML,MAAM,GAAG,MAAMH,MAAM,CAACf,yBAAyB,CAAC,CAACmB,eAAe,CAAC;MACrEgB,MAAM,EAAE;QAAEC,SAAS,EAAE,MAAM;QAAEC,QAAQ,EAAE,KAAK;QAAEC,GAAG,EAAE,EAAE;QAAEd,MAAM,EAAEF;MAAS;IAC1E,CAAC,CAAC;IACFX,MAAM,CAACO,MAAM,CAAC,CAACO,OAAO,CAAC;MACrBW,SAAS,EAAE,MAAM;MACjBC,QAAQ,EAAE,KAAK;MACfC,GAAG,EAAE,EAAE;MACPd,MAAM,EAAE;QACNE,QAAQ,EAAE,UAAU;QACpBC,WAAW,EAAE,UAAU;QACvBC,WAAW,EAAE;MACf;IACF,CAAC,CAAC;EACJ,CAAC,CAAC;EAEFf,EAAE,CAAC,4CAA4C,EAAE,YAAY;IAC3D,MAAMK,MAAM,GAAG,MAAMH,MAAM,CAACT,iCAAiC,CAAC,CAC3Da,eAAe,CAAC,CAAC;IACpBP,YAAY,CAAgB,CAAC,CAAC2B,aAAa;IAG3C5B,MAAM,CAACO,MAAM,CAAC,CAACO,OAAO,CAAC,CAAC;MAAEe,GAAG,EAAE,OAAO;MAAEC,KAAK,EAAE;IAAE,CAAC,EAAE;MAClDD,GAAG,EAAE,OAAO;MACZC,KAAK,EAAE;IACT,CAAC,CAAC,CAAC;EACL,CAAC,CAAC;EAEF5B,EAAE,CAAC,oDAAoD,EAAE,YAAY;IAEjEE,MAAM,CAACnB,wCAAwC,CAAC,CAACuB,eAAe;IAKlEP,YAAY,CAAoD,CAAC,CAC9DoB,aAAa,CAAoB,CAAC;IAErC,MAAMd,MAAM,GAAG,MAAMH,MAAM,CAACnB,wCAAwC,CAAC,CAClEuB,eAAe,CAAC;MACfuB,WAAW,EAAE,CACX;QACEF,GAAG,EAAE,UAAU;QACfC,KAAK,EAAE;MACT,CAAC,EACD;QACED,GAAG,EAAE,UAAU;QACfC,KAAK,EAAE;MACT,CAAC;IAEL,CAAC,CAAC;IACJ7B,YAAY,CAAgB,CAAC,CAAC2B,aAAa;IAI3C5B,MAAM,CAACO,MAAM,CAAC,CAACO,OAAO,CAAC,CAAC;MAAEe,GAAG,EAAE,cAAc;MAAEC,KAAK,EAAE;IAAE,CAAC,EAAE;MACzDD,GAAG,EAAE,cAAc;MACnBC,KAAK,EAAE;IACT,CAAC,CAAC,CAAC;EACL,CAAC,CAAC;EAEF5B,EAAE,CAAC,8CAA8C,EAAE,YAAY;IAC7D,MAAMK,MAAM,GAAG,MAAMH,MAAM,CAACV,mCAAmC,CAAC,CAC7Dc,eAAe,CAAC,CAAC;IAEpBP,YAAY,CAAgB,CAAC,CAAC2B,aAAa;IAS3C5B,MAAM,CAACO,MAAM,CAAC,CAACO,OAAO,CAAC,CAAC;MACtBe,GAAG,EAAE,OAAO;MACZG,MAAM,EAAE,CAAC;QACPH,GAAG,EAAE;UACHI,UAAU,EAAE,sBAAsB;UAClCC,QAAQ,EAAE;QACZ,CAAC;QACDJ,KAAK,EAAE;MACT,CAAC;IACH,CAAC,EAAE;MAAED,GAAG,EAAE,OAAO;MAAEG,MAAM,EAAE;IAAG,CAAC,CAAC,CAAC;EACnC,CAAC,CAAC;EAEF9B,EAAE,CAAC,2DAA2D,EAAE,YAAY;IAC1E,IAAI;MACa,MAAME,MAAM,CAAClB,MAAM,CAAC,CAACsB,eAAe,CAAC;QAAEC,CAAC,EAAE;MAAE,CAAC,CAAC;MAC7DT,MAAM,CAACmC,IAAI,CAAC,uBAAuB,CAAC;IACtC,CAAC,CAAC,OAAOC,CAAC,EAAE;MACVpC,MAAM,CAAEoC,CAAC,CAAWC,OAAO,CAAC,CAACC,OAAO,CAClC,iDACF,CAAC;IACH;EACF,CAAC,CAAC;EAEFpC,EAAE,CAAC,sDAAsD,EAAE,YAAY;IAEnEE,MAAM,CAACpB,0CAA0C,CAAC,CAACwB,eAAe;IAKpEP,YAAY,CAQT,CAAC,CACDoB,aAAa,CAAoB,CAAC;IAErC,MAAMd,MAAM,GAAG,MAAMH,MAAM,CAACpB,0CAA0C,CAAC,CACpEwB,eAAe,CAAC;MACfuB,WAAW,EAAE,CACX;QACEF,GAAG,EAAE,UAAU;QACfG,MAAM,EAAE,CACN;UACEH,GAAG,EAAE;YACHI,UAAU,EAAE,sBAAsB;YAClCC,QAAQ,EAAE;UACZ,CAAC;UACDJ,KAAK,EAAE;QACT,CAAC;MAEL,CAAC,EACD;QACED,GAAG,EAAE,UAAU;QACfG,MAAM,EAAE;MACV,CAAC;IAEL,CAAC,CAAC;IAEJhC,MAAM,CAACO,MAAM,CAAC,CAACO,OAAO,CAAC,CACrB;MACEe,GAAG,EAAE,OAAO;MACZG,MAAM,EAAE,CACN;QACEH,GAAG,EAAE;UACHI,UAAU,EAAE,sBAAsB;UAClCC,QAAQ,EAAE;QACZ,CAAC;QACDJ,KAAK,EAAE;MACT,CAAC;IAEL,CAAC,EACD;MACED,GAAG,EAAE,OAAO;MACZG,MAAM,EAAE;IACV,CAAC,CACF,CAAC;EACJ,CAAC,CAAC;EAEF9B,EAAE,CAAC,6BAA6B,EAAE,YAAY;IAC5C,MAAMiB,iBAAiB,GAAGf,MAAM,CAACjB,QAAQ,CAAC;IAC3B,MAAMiB,MAAM,CAACb,sBAAsB,CAAC,CAACiB,eAAe,CAAC;MAClEY,SAAS,EAAED;IACb,CAAC,CAAC;IAEFlB,YAAY,CAAsB,CAAC,CAACoB,aAAa,CAAgB,CAAC;EACpE,CAAC,CAAC;EACFnB,EAAE,CAAC,wBAAwB,EAAE,YAAY;IACvC,MAAMqC,OAAO,GAAGC,MAAM,CAACC,IAAI,CAAC1D,QAAQ,CAAC;IACrCiB,MAAM,CAACuC,OAAO,CAAC,CAACG,aAAa,CAAC,CAC5B,4CAA4C,EAC5C,0CAA0C,EAC1C,QAAQ,EACR,oBAAoB,EACpB,2BAA2B,EAC3B,oBAAoB,EACpB,wBAAwB,EACxB,uBAAuB,EACvB,aAAa,EACb,eAAe,EACf,kBAAkB,EAClB,qCAAqC,EACrC,mCAAmC,CACpC,CAAC;EACJ,CAAC,CAAC;EAEFxC,EAAE,CAAC,0BAA0B,EAAE,YAAY;IACzC,MAAMK,MAAM,GAAG,MAAMH,MAAM,CAACrB,QAAQ,CAAC4D,qBAAqB,CAAC,CAACnC,eAAe,CACzE;MAAEoC,MAAM,EAAE,CAAC,MAAM,EAAE,QAAQ,EAAE,MAAM;IAAE,CACvC,CAAC;IACD5C,MAAM,CAACO,MAAM,CAAC,CAACO,OAAO,CAAC,CAAC,MAAM,EAAE,SAAS,EAAE,UAAU,CAAC,CAAC;EACzD,CAAC,CAAC;AACJ,CAAC,CAAC","ignoreList":[]}
+{"version":3,"file":"queries.test.js","names":["$ontologyRid","$Queries","acceptsThreeDimensionalAggregationFunction","acceptsTwoDimensionalAggregationFunction","addOne","Employee","incrementPersonAge","incrementPersonAgeComplex","queryAcceptsObject","queryAcceptsObjectSets","queryTypeReturnsMap","returnsDate","returnsTimestamp","threeDimensionalAggregationFunction","twoDimensionalAggregationFunction","apiServer","afterAll","beforeAll","describe","expect","expectTypeOf","it","createClient","client","listen","close","result","executeFunction","n","toBe","employee","fetchOne","object","toEqual","$apiName","$objectType","$primaryKey","$objectSpecifier","result2","employeeObjectSet","objectSet","toMatchTypeOf","resultWithTimestamp","resultWithDate","person","firstName","lastName","age","toEqualTypeOf","key","value","aggFunction","groups","startValue","endValue","fail","e","message","toMatch","peopleMap","toMatchInlineSnapshot","queries","Object","keys","toStrictEqual","queryTypeReturnsArray","people"],"sources":["queries.test.ts"],"sourcesContent":["/*\n * Copyright 2023 Palantir Technologies, Inc. All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport type { ObjectSet, ObjectSpecifier } from \"@osdk/api\";\nimport {\n  $ontologyRid,\n  $Queries,\n  acceptsThreeDimensionalAggregationFunction,\n  acceptsTwoDimensionalAggregationFunction,\n  addOne,\n  Employee,\n  incrementPersonAge,\n  incrementPersonAgeComplex,\n  queryAcceptsObject,\n  queryAcceptsObjectSets,\n  queryTypeReturnsMap,\n  returnsDate,\n  returnsTimestamp,\n  threeDimensionalAggregationFunction,\n  twoDimensionalAggregationFunction,\n} from \"@osdk/client.test.ontology\";\nimport { apiServer } from \"@osdk/shared.test\";\nimport {\n  afterAll,\n  beforeAll,\n  describe,\n  expect,\n  expectTypeOf,\n  it,\n} from \"vitest\";\nimport type { Client } from \"../Client.js\";\nimport { createClient } from \"../createClient.js\";\n\ndescribe(\"queries\", () => {\n  let client: Client;\n\n  beforeAll(async () => {\n    apiServer.listen();\n    client = createClient(\n      \"https://stack.palantir.com\",\n      $ontologyRid,\n      async () => \"myAccessToken\",\n    );\n  });\n\n  afterAll(() => {\n    apiServer.close();\n  });\n\n  it(\"simple query works\", async () => {\n    const result = await client(addOne).executeFunction({ n: 2 });\n    expect(result).toBe(3);\n  });\n\n  it(\"accepts objects\", async () => {\n    const employee = await client(Employee).fetchOne(\n      50030,\n    );\n    const result = await client(queryAcceptsObject).executeFunction({\n      object: employee,\n    });\n    expect(result).toEqual({\n      $apiName: \"Employee\",\n      $objectType: \"Employee\",\n      $primaryKey: 50031,\n      $objectSpecifier: \"Employee:50031\",\n    });\n\n    // Should also accept primary keys\n    const result2 = await client(queryAcceptsObject).executeFunction({\n      object: 50030,\n    });\n    expect(result2).toEqual({\n      $apiName: \"Employee\",\n      $objectType: \"Employee\",\n      $primaryKey: 50031,\n      $objectSpecifier: \"Employee:50031\",\n    });\n  });\n\n  it(\"accepts objectSets\", async () => {\n    const employeeObjectSet = client(Employee);\n    const result = await client(queryAcceptsObjectSets).executeFunction({\n      objectSet: employeeObjectSet,\n    });\n\n    expectTypeOf<ObjectSet<Employee>>().toMatchTypeOf<typeof result>();\n  });\n\n  it(\"no params work\", async () => {\n    const resultWithTimestamp = await client(returnsTimestamp)\n      .executeFunction();\n    expect(resultWithTimestamp).toBe(\"2019-01-01T00:00:00.000Z\");\n\n    const resultWithDate = await client(returnsDate).executeFunction();\n    expect(resultWithDate).toBe(\"2019-01-01\");\n  });\n\n  it(\"returns and accepts structs property\", async () => {\n    const result = await client(incrementPersonAge).executeFunction({\n      person: { firstName: \"John\", lastName: \"Doe\", age: 42 },\n    });\n    expect(result).toEqual({\n      firstName: \"John\",\n      lastName: \"Doe\",\n      age: 43,\n    });\n  });\n\n  it(\"returns and accepts complex structs property\", async () => {\n    const employee = await client(Employee).fetchOne(\n      50030,\n    );\n    const result = await client(incrementPersonAgeComplex).executeFunction({\n      person: { firstName: \"John\", lastName: \"Doe\", age: 42, object: employee },\n    });\n    expect(result).toEqual({\n      firstName: \"John\",\n      lastName: \"Doe\",\n      age: 43,\n      object: {\n        $apiName: \"Employee\",\n        $objectType: \"Employee\",\n        $primaryKey: 50031,\n        $objectSpecifier: \"Employee:50031\",\n      },\n    });\n  });\n\n  it(\"two dimensional aggregation response works\", async () => {\n    const result = await client(twoDimensionalAggregationFunction)\n      .executeFunction();\n    expectTypeOf<typeof result>().toEqualTypeOf<\n      { key: string; value: number }[]\n    >;\n    expect(result).toEqual([{ key: \"Q-AFN\", value: 1 }, {\n      key: \"Q-AFO\",\n      value: 2,\n    }]);\n  });\n\n  it(\"two dimensional aggregation request/response works\", async () => {\n    const clientBoundQueryFunction =\n      client(acceptsTwoDimensionalAggregationFunction).executeFunction;\n    type InferredParamType = Parameters<\n      typeof clientBoundQueryFunction\n    >[0];\n\n    expectTypeOf<{ aggFunction: { key: string; value: number }[] }>()\n      .toMatchTypeOf<InferredParamType>();\n\n    const result = await client(acceptsTwoDimensionalAggregationFunction)\n      .executeFunction({\n        aggFunction: [\n          {\n            key: \"testKey1\",\n            value: 1,\n          },\n          {\n            key: \"testKey2\",\n            value: 2,\n          },\n        ],\n      });\n    expectTypeOf<typeof result>().toEqualTypeOf<\n      { key: string; value: number }[]\n    >;\n\n    expect(result).toEqual([{ key: \"responseKey1\", value: 3 }, {\n      key: \"responseKey2\",\n      value: 4,\n    }]);\n  });\n\n  it(\"three dimensional aggregation response works\", async () => {\n    const result = await client(threeDimensionalAggregationFunction)\n      .executeFunction();\n\n    expectTypeOf<typeof result>().toEqualTypeOf<\n      {\n        key: string;\n        groups: {\n          key: { startValue: string | undefined; endValue: string | undefined };\n          value: number;\n        }[];\n      }[]\n    >;\n    expect(result).toEqual([{\n      key: \"Q-AFN\",\n      groups: [{\n        key: {\n          startValue: \"2010-10-01T00:00:00Z\",\n          endValue: \"2010-10-02T00:00:00Z\",\n        },\n        value: 65.0,\n      }],\n    }, { key: \"Q-AFO\", groups: [] }]);\n  });\n\n  it(\"throws when response is null and response is non-nullable\", async () => {\n    try {\n      const result = await client(addOne).executeFunction({ n: 3 });\n      expect.fail(\"Should not reach here\");\n    } catch (e) {\n      expect((e as Error).message).toMatch(\n        `Got null response when nullable was not allowed`,\n      );\n    }\n  });\n\n  it(\"three dimensional aggregation request/response works\", async () => {\n    const clientBoundQueryFunction =\n      client(acceptsThreeDimensionalAggregationFunction).executeFunction;\n    type InferredParamType = Parameters<\n      typeof clientBoundQueryFunction\n    >[0];\n\n    expectTypeOf<{\n      aggFunction: {\n        key: string;\n        groups: {\n          key: { startValue: string | undefined; endValue: string };\n          value: number;\n        }[];\n      }[];\n    }>()\n      .toMatchTypeOf<InferredParamType>();\n\n    const result = await client(acceptsThreeDimensionalAggregationFunction)\n      .executeFunction({\n        aggFunction: [\n          {\n            key: \"testKey1\",\n            groups: [\n              {\n                key: {\n                  startValue: \"2010-10-01T00:00:00Z\",\n                  endValue: \"2010-10-02T00:00:00Z\",\n                },\n                value: 65.0,\n              },\n            ],\n          },\n          {\n            key: \"testKey2\",\n            groups: [],\n          },\n        ],\n      });\n\n    expect(result).toEqual([\n      {\n        key: \"Q-AFN\",\n        groups: [\n          {\n            key: {\n              startValue: \"2010-10-01T00:00:00Z\",\n              endValue: \"2010-10-02T00:00:00Z\",\n            },\n            value: 65.0,\n          },\n        ],\n      },\n      {\n        key: \"Q-AFO\",\n        groups: [],\n      },\n    ]);\n  });\n\n  it(\"map type request and response works\", async () => {\n    const clientBoundQueryFunction =\n      client(queryTypeReturnsMap).executeFunction;\n    type InferredParamType = Parameters<\n      typeof clientBoundQueryFunction\n    >[0];\n\n    expectTypeOf<InferredParamType>()\n      .toMatchTypeOf<\n        { peopleMap: Record<ObjectSpecifier<Employee>, string> }\n      >();\n\n    const myMap: Record<ObjectSpecifier<Employee>, string> = {\n      [\"Employee:person1\" as any]: \"hi\",\n    };\n    const result = await client(queryTypeReturnsMap).executeFunction({\n      peopleMap: myMap,\n    });\n\n    expect(result).toMatchInlineSnapshot(`\n      {\n        \"Employee:50030\": \"bye\",\n      }\n    `);\n\n    expectTypeOf<typeof result>().toMatchTypeOf<\n      Record<ObjectSpecifier<Employee>, number>\n    >();\n\n    const object = await client(Employee).fetchOne(50030);\n\n    const value = result[object.$objectSpecifier];\n\n    expect(value).toBe(\"bye\");\n  });\n\n  it(\"accepts and returns objects\", async () => {\n    const employeeObjectSet = client(Employee);\n    const result = await client(queryAcceptsObjectSets).executeFunction({\n      objectSet: employeeObjectSet,\n    });\n\n    expectTypeOf<ObjectSet<Employee>>().toMatchTypeOf<typeof result>();\n  });\n  it(\"queries are enumerable\", async () => {\n    const queries = Object.keys($Queries);\n    expect(queries).toStrictEqual([\n      \"acceptsThreeDimensionalAggregationFunction\",\n      \"acceptsTwoDimensionalAggregationFunction\",\n      \"addOne\",\n      \"incrementPersonAge\",\n      \"incrementPersonAgeComplex\",\n      \"queryAcceptsObject\",\n      \"queryAcceptsObjectSets\",\n      \"queryTypeReturnsArray\",\n      \"queryTypeReturnsMap\",\n      \"returnsDate\",\n      \"returnsObject\",\n      \"returnsTimestamp\",\n      \"threeDimensionalAggregationFunction\",\n      \"twoDimensionalAggregationFunction\",\n    ]);\n  });\n\n  it(\"queries work with arrays\", async () => {\n    const result = await client($Queries.queryTypeReturnsArray).executeFunction(\n      { people: [\"Brad\", \"George\", \"Ryan\"] },\n    );\n    expect(result).toEqual([\"Pitt\", \"Clooney\", \"Reynolds\"]);\n  });\n});\n"],"mappings":"AAAA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;;AAGA,SACEA,YAAY,EACZC,QAAQ,EACRC,0CAA0C,EAC1CC,wCAAwC,EACxCC,MAAM,EACNC,QAAQ,EACRC,kBAAkB,EAClBC,yBAAyB,EACzBC,kBAAkB,EAClBC,sBAAsB,EACtBC,mBAAmB,EACnBC,WAAW,EACXC,gBAAgB,EAChBC,mCAAmC,EACnCC,iCAAiC,QAC5B,4BAA4B;AACnC,SAASC,SAAS,QAAQ,mBAAmB;AAC7C,SACEC,QAAQ,EACRC,SAAS,EACTC,QAAQ,EACRC,MAAM,EACNC,YAAY,EACZC,EAAE,QACG,QAAQ;AAEf,SAASC,YAAY,QAAQ,oBAAoB;AAEjDJ,QAAQ,CAAC,SAAS,EAAE,MAAM;EACxB,IAAIK,MAAc;EAElBN,SAAS,CAAC,YAAY;IACpBF,SAAS,CAACS,MAAM,CAAC,CAAC;IAClBD,MAAM,GAAGD,YAAY,CACnB,4BAA4B,EAC5BtB,YAAY,EACZ,YAAY,eACd,CAAC;EACH,CAAC,CAAC;EAEFgB,QAAQ,CAAC,MAAM;IACbD,SAAS,CAACU,KAAK,CAAC,CAAC;EACnB,CAAC,CAAC;EAEFJ,EAAE,CAAC,oBAAoB,EAAE,YAAY;IACnC,MAAMK,MAAM,GAAG,MAAMH,MAAM,CAACnB,MAAM,CAAC,CAACuB,eAAe,CAAC;MAAEC,CAAC,EAAE;IAAE,CAAC,CAAC;IAC7DT,MAAM,CAACO,MAAM,CAAC,CAACG,IAAI,CAAC,CAAC,CAAC;EACxB,CAAC,CAAC;EAEFR,EAAE,CAAC,iBAAiB,EAAE,YAAY;IAChC,MAAMS,QAAQ,GAAG,MAAMP,MAAM,CAAClB,QAAQ,CAAC,CAAC0B,QAAQ,CAC9C,KACF,CAAC;IACD,MAAML,MAAM,GAAG,MAAMH,MAAM,CAACf,kBAAkB,CAAC,CAACmB,eAAe,CAAC;MAC9DK,MAAM,EAAEF;IACV,CAAC,CAAC;IACFX,MAAM,CAACO,MAAM,CAAC,CAACO,OAAO,CAAC;MACrBC,QAAQ,EAAE,UAAU;MACpBC,WAAW,EAAE,UAAU;MACvBC,WAAW,EAAE,KAAK;MAClBC,gBAAgB,EAAE;IACpB,CAAC,CAAC;;IAEF;IACA,MAAMC,OAAO,GAAG,MAAMf,MAAM,CAACf,kBAAkB,CAAC,CAACmB,eAAe,CAAC;MAC/DK,MAAM,EAAE;IACV,CAAC,CAAC;IACFb,MAAM,CAACmB,OAAO,CAAC,CAACL,OAAO,CAAC;MACtBC,QAAQ,EAAE,UAAU;MACpBC,WAAW,EAAE,UAAU;MACvBC,WAAW,EAAE,KAAK;MAClBC,gBAAgB,EAAE;IACpB,CAAC,CAAC;EACJ,CAAC,CAAC;EAEFhB,EAAE,CAAC,oBAAoB,EAAE,YAAY;IACnC,MAAMkB,iBAAiB,GAAGhB,MAAM,CAAClB,QAAQ,CAAC;IAC3B,MAAMkB,MAAM,CAACd,sBAAsB,CAAC,CAACkB,eAAe,CAAC;MAClEa,SAAS,EAAED;IACb,CAAC,CAAC;IAEFnB,YAAY,CAAsB,CAAC,CAACqB,aAAa,CAAgB,CAAC;EACpE,CAAC,CAAC;EAEFpB,EAAE,CAAC,gBAAgB,EAAE,YAAY;IAC/B,MAAMqB,mBAAmB,GAAG,MAAMnB,MAAM,CAACX,gBAAgB,CAAC,CACvDe,eAAe,CAAC,CAAC;IACpBR,MAAM,CAACuB,mBAAmB,CAAC,CAACb,IAAI,CAAC,0BAA0B,CAAC;IAE5D,MAAMc,cAAc,GAAG,MAAMpB,MAAM,CAACZ,WAAW,CAAC,CAACgB,eAAe,CAAC,CAAC;IAClER,MAAM,CAACwB,cAAc,CAAC,CAACd,IAAI,CAAC,YAAY,CAAC;EAC3C,CAAC,CAAC;EAEFR,EAAE,CAAC,sCAAsC,EAAE,YAAY;IACrD,MAAMK,MAAM,GAAG,MAAMH,MAAM,CAACjB,kBAAkB,CAAC,CAACqB,eAAe,CAAC;MAC9DiB,MAAM,EAAE;QAAEC,SAAS,EAAE,MAAM;QAAEC,QAAQ,EAAE,KAAK;QAAEC,GAAG,EAAE;MAAG;IACxD,CAAC,CAAC;IACF5B,MAAM,CAACO,MAAM,CAAC,CAACO,OAAO,CAAC;MACrBY,SAAS,EAAE,MAAM;MACjBC,QAAQ,EAAE,KAAK;MACfC,GAAG,EAAE;IACP,CAAC,CAAC;EACJ,CAAC,CAAC;EAEF1B,EAAE,CAAC,8CAA8C,EAAE,YAAY;IAC7D,MAAMS,QAAQ,GAAG,MAAMP,MAAM,CAAClB,QAAQ,CAAC,CAAC0B,QAAQ,CAC9C,KACF,CAAC;IACD,MAAML,MAAM,GAAG,MAAMH,MAAM,CAAChB,yBAAyB,CAAC,CAACoB,eAAe,CAAC;MACrEiB,MAAM,EAAE;QAAEC,SAAS,EAAE,MAAM;QAAEC,QAAQ,EAAE,KAAK;QAAEC,GAAG,EAAE,EAAE;QAAEf,MAAM,EAAEF;MAAS;IAC1E,CAAC,CAAC;IACFX,MAAM,CAACO,MAAM,CAAC,CAACO,OAAO,CAAC;MACrBY,SAAS,EAAE,MAAM;MACjBC,QAAQ,EAAE,KAAK;MACfC,GAAG,EAAE,EAAE;MACPf,MAAM,EAAE;QACNE,QAAQ,EAAE,UAAU;QACpBC,WAAW,EAAE,UAAU;QACvBC,WAAW,EAAE,KAAK;QAClBC,gBAAgB,EAAE;MACpB;IACF,CAAC,CAAC;EACJ,CAAC,CAAC;EAEFhB,EAAE,CAAC,4CAA4C,EAAE,YAAY;IAC3D,MAAMK,MAAM,GAAG,MAAMH,MAAM,CAACT,iCAAiC,CAAC,CAC3Da,eAAe,CAAC,CAAC;IACpBP,YAAY,CAAgB,CAAC,CAAC4B,aAAa;IAG3C7B,MAAM,CAACO,MAAM,CAAC,CAACO,OAAO,CAAC,CAAC;MAAEgB,GAAG,EAAE,OAAO;MAAEC,KAAK,EAAE;IAAE,CAAC,EAAE;MAClDD,GAAG,EAAE,OAAO;MACZC,KAAK,EAAE;IACT,CAAC,CAAC,CAAC;EACL,CAAC,CAAC;EAEF7B,EAAE,CAAC,oDAAoD,EAAE,YAAY;IAEjEE,MAAM,CAACpB,wCAAwC,CAAC,CAACwB,eAAe;IAKlEP,YAAY,CAAoD,CAAC,CAC9DqB,aAAa,CAAoB,CAAC;IAErC,MAAMf,MAAM,GAAG,MAAMH,MAAM,CAACpB,wCAAwC,CAAC,CAClEwB,eAAe,CAAC;MACfwB,WAAW,EAAE,CACX;QACEF,GAAG,EAAE,UAAU;QACfC,KAAK,EAAE;MACT,CAAC,EACD;QACED,GAAG,EAAE,UAAU;QACfC,KAAK,EAAE;MACT,CAAC;IAEL,CAAC,CAAC;IACJ9B,YAAY,CAAgB,CAAC,CAAC4B,aAAa;IAI3C7B,MAAM,CAACO,MAAM,CAAC,CAACO,OAAO,CAAC,CAAC;MAAEgB,GAAG,EAAE,cAAc;MAAEC,KAAK,EAAE;IAAE,CAAC,EAAE;MACzDD,GAAG,EAAE,cAAc;MACnBC,KAAK,EAAE;IACT,CAAC,CAAC,CAAC;EACL,CAAC,CAAC;EAEF7B,EAAE,CAAC,8CAA8C,EAAE,YAAY;IAC7D,MAAMK,MAAM,GAAG,MAAMH,MAAM,CAACV,mCAAmC,CAAC,CAC7Dc,eAAe,CAAC,CAAC;IAEpBP,YAAY,CAAgB,CAAC,CAAC4B,aAAa;IAS3C7B,MAAM,CAACO,MAAM,CAAC,CAACO,OAAO,CAAC,CAAC;MACtBgB,GAAG,EAAE,OAAO;MACZG,MAAM,EAAE,CAAC;QACPH,GAAG,EAAE;UACHI,UAAU,EAAE,sBAAsB;UAClCC,QAAQ,EAAE;QACZ,CAAC;QACDJ,KAAK,EAAE;MACT,CAAC;IACH,CAAC,EAAE;MAAED,GAAG,EAAE,OAAO;MAAEG,MAAM,EAAE;IAAG,CAAC,CAAC,CAAC;EACnC,CAAC,CAAC;EAEF/B,EAAE,CAAC,2DAA2D,EAAE,YAAY;IAC1E,IAAI;MACa,MAAME,MAAM,CAACnB,MAAM,CAAC,CAACuB,eAAe,CAAC;QAAEC,CAAC,EAAE;MAAE,CAAC,CAAC;MAC7DT,MAAM,CAACoC,IAAI,CAAC,uBAAuB,CAAC;IACtC,CAAC,CAAC,OAAOC,CAAC,EAAE;MACVrC,MAAM,CAAEqC,CAAC,CAAWC,OAAO,CAAC,CAACC,OAAO,CAClC,iDACF,CAAC;IACH;EACF,CAAC,CAAC;EAEFrC,EAAE,CAAC,sDAAsD,EAAE,YAAY;IAEnEE,MAAM,CAACrB,0CAA0C,CAAC,CAACyB,eAAe;IAKpEP,YAAY,CAQT,CAAC,CACDqB,aAAa,CAAoB,CAAC;IAErC,MAAMf,MAAM,GAAG,MAAMH,MAAM,CAACrB,0CAA0C,CAAC,CACpEyB,eAAe,CAAC;MACfwB,WAAW,EAAE,CACX;QACEF,GAAG,EAAE,UAAU;QACfG,MAAM,EAAE,CACN;UACEH,GAAG,EAAE;YACHI,UAAU,EAAE,sBAAsB;YAClCC,QAAQ,EAAE;UACZ,CAAC;UACDJ,KAAK,EAAE;QACT,CAAC;MAEL,CAAC,EACD;QACED,GAAG,EAAE,UAAU;QACfG,MAAM,EAAE;MACV,CAAC;IAEL,CAAC,CAAC;IAEJjC,MAAM,CAACO,MAAM,CAAC,CAACO,OAAO,CAAC,CACrB;MACEgB,GAAG,EAAE,OAAO;MACZG,MAAM,EAAE,CACN;QACEH,GAAG,EAAE;UACHI,UAAU,EAAE,sBAAsB;UAClCC,QAAQ,EAAE;QACZ,CAAC;QACDJ,KAAK,EAAE;MACT,CAAC;IAEL,CAAC,EACD;MACED,GAAG,EAAE,OAAO;MACZG,MAAM,EAAE;IACV,CAAC,CACF,CAAC;EACJ,CAAC,CAAC;EAEF/B,EAAE,CAAC,qCAAqC,EAAE,YAAY;IAElDE,MAAM,CAACb,mBAAmB,CAAC,CAACiB,eAAe;IAK7CP,YAAY,CAAoB,CAAC,CAC9BqB,aAAa,CAEZ,CAAC;IAKL,MAAMf,MAAM,GAAG,MAAMH,MAAM,CAACb,mBAAmB,CAAC,CAACiB,eAAe,CAAC;MAC/DgC,SAAS,EAJ8C;QACvD,CAAC,kBAAkB,GAAU;MAC/B;IAGA,CAAC,CAAC;IAEFxC,MAAM,CAACO,MAAM,CAAC,CAACkC,qBAAqB,CAAC;AACzC;AACA;AACA;AACA,KAAK,CAAC;IAEFxC,YAAY,CAAgB,CAAC,CAACqB,aAAa,CAEzC,CAAC;IAEH,MAAMT,MAAM,GAAG,MAAMT,MAAM,CAAClB,QAAQ,CAAC,CAAC0B,QAAQ,CAAC,KAAK,CAAC;IAErD,MAAMmB,KAAK,GAAGxB,MAAM,CAACM,MAAM,CAACK,gBAAgB,CAAC;IAE7ClB,MAAM,CAAC+B,KAAK,CAAC,CAACrB,IAAI,CAAC,KAAK,CAAC;EAC3B,CAAC,CAAC;EAEFR,EAAE,CAAC,6BAA6B,EAAE,YAAY;IAC5C,MAAMkB,iBAAiB,GAAGhB,MAAM,CAAClB,QAAQ,CAAC;IAC3B,MAAMkB,MAAM,CAACd,sBAAsB,CAAC,CAACkB,eAAe,CAAC;MAClEa,SAAS,EAAED;IACb,CAAC,CAAC;IAEFnB,YAAY,CAAsB,CAAC,CAACqB,aAAa,CAAgB,CAAC;EACpE,CAAC,CAAC;EACFpB,EAAE,CAAC,wBAAwB,EAAE,YAAY;IACvC,MAAMwC,OAAO,GAAGC,MAAM,CAACC,IAAI,CAAC9D,QAAQ,CAAC;IACrCkB,MAAM,CAAC0C,OAAO,CAAC,CAACG,aAAa,CAAC,CAC5B,4CAA4C,EAC5C,0CAA0C,EAC1C,QAAQ,EACR,oBAAoB,EACpB,2BAA2B,EAC3B,oBAAoB,EACpB,wBAAwB,EACxB,uBAAuB,EACvB,qBAAqB,EACrB,aAAa,EACb,eAAe,EACf,kBAAkB,EAClB,qCAAqC,EACrC,mCAAmC,CACpC,CAAC;EACJ,CAAC,CAAC;EAEF3C,EAAE,CAAC,0BAA0B,EAAE,YAAY;IACzC,MAAMK,MAAM,GAAG,MAAMH,MAAM,CAACtB,QAAQ,CAACgE,qBAAqB,CAAC,CAACtC,eAAe,CACzE;MAAEuC,MAAM,EAAE,CAAC,MAAM,EAAE,QAAQ,EAAE,MAAM;IAAE,CACvC,CAAC;IACD/C,MAAM,CAACO,MAAM,CAAC,CAACO,OAAO,CAAC,CAAC,MAAM,EAAE,SAAS,EAAE,UAAU,CAAC,CAAC;EACzD,CAAC,CAAC;AACJ,CAAC,CAAC","ignoreList":[]}

package/build/browser/util/UserAgent.js

@@ -14,5 +14,5 @@
  * limitations under the License.
  */
 
-export const USER_AGENT = `osdk-client/${"2.2.0-beta.6"}`;
+export const USER_AGENT = `osdk-client/${"2.2.0-beta.8"}`;
 //# sourceMappingURL=UserAgent.js.map

package/build/browser/util/toDataValueQueries.js

@@ -99,6 +99,20 @@ export async function toDataValueQueries(value, client, desiredType) {
         }
         break;
       }
+    case "map":
+      {
+        if (typeof value === "object") {
+          const entrySet = [];
+          for (const [key, mapValue] of Object.entries(value)) {
+            entrySet.push({
+              key: desiredType.keyType.type === "object" ? extractPrimaryKeyFromObjectIdentifier(key) : await toDataValueQueries(key, client, desiredType.keyType),
+              value: await toDataValueQueries(mapValue, client, desiredType.valueType)
+            });
+          }
+          return entrySet;
+        }
+        break;
+      }
     case "struct":
       {
         if (typeof value === "object") {
@@ -121,4 +135,7 @@ export async function toDataValueQueries(value, client, desiredType) {
   }
   return value;
 }
+function extractPrimaryKeyFromObjectIdentifier(a) {
+  return a.substring(a.indexOf(":") + 1);
+}
 //# sourceMappingURL=toDataValueQueries.js.map

package/build/browser/util/toDataValueQueries.js.map

@@ -1 +1 @@
-{"version":3,"file":"toDataValueQueries.js","names":["OntologiesV2","isAttachmentFile","isAttachmentUpload","getWireObjectSet","isObjectSet","isOsdkBaseObject","isWireObjectSet","toDataValueQueries","value","client","desiredType","Array","isArray","multiplicity","values","from","some","dataValue","converted","push","promiseArray","innerValue","Promise","all","type","attachment","Attachments","upload","data","filename","name","rid","groups","Set","$primaryKey","structMap","key","structValue","Object","entries"],"sources":["toDataValueQueries.ts"],"sourcesContent":["/*\n * Copyright 2024 Palantir Technologies, Inc. All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport type { QueryDataTypeDefinition } from \"@osdk/api\";\nimport { type DataValue } from \"@osdk/foundry.ontologies\";\nimport * as OntologiesV2 from \"@osdk/foundry.ontologies\";\nimport type { MinimalClient } from \"../MinimalClientContext.js\";\nimport {\n  isAttachmentFile,\n  isAttachmentUpload,\n} from \"../object/AttachmentUpload.js\";\nimport { getWireObjectSet, isObjectSet } from \"../objectSet/createObjectSet.js\";\nimport { isOsdkBaseObject } from \"./isOsdkBaseObject.js\";\nimport { isWireObjectSet } from \"./WireObjectSet.js\";\n\n/**\n * Marshall user-facing data into the wire DataValue type\n *\n * @see DataValue for the expected payloads\n * @internal\n */\nexport async function toDataValueQueries(\n  value: unknown,\n  client: MinimalClient,\n  desiredType: QueryDataTypeDefinition,\n): Promise<DataValue> {\n  if (value == null) {\n    return value;\n  }\n\n  if (Array.isArray(value) && desiredType.multiplicity) {\n    const values = Array.from(value);\n    if (\n      values.some((dataValue) =>\n        isAttachmentUpload(dataValue) || isAttachmentFile(dataValue)\n      )\n    ) {\n      const converted = [];\n      for (const value of values) {\n        converted.push(await toDataValueQueries(value, client, desiredType));\n      }\n      return converted;\n    }\n    const promiseArray = Array.from(\n      value,\n      async (innerValue) =>\n        await toDataValueQueries(innerValue, client, desiredType),\n    );\n    return Promise.all(promiseArray);\n  }\n\n  switch (desiredType.type) {\n    case \"attachment\": {\n      if (isAttachmentUpload(value)) {\n        const attachment = await OntologiesV2.Attachments.upload(\n          client,\n          value.data,\n          {\n            filename: value.name,\n          },\n        );\n        return attachment.rid;\n      }\n\n      if (\n        isAttachmentFile(value)\n      ) {\n        const attachment = await OntologiesV2.Attachments.upload(\n          client,\n          value,\n          {\n            filename: value.name as string,\n          },\n        );\n        return attachment.rid;\n      }\n\n      // If it's not an upload, it's just an attachment rid string which we can pass through\n      return value;\n    }\n    case \"twoDimensionalAggregation\": {\n      return {\n        groups: value,\n      };\n    }\n    case \"threeDimensionalAggregation\": {\n      return {\n        groups: value,\n      };\n    }\n\n    case \"set\": {\n      if (value instanceof Set) {\n        const promiseArray = Array.from(\n          value,\n          async (innerValue) =>\n            await toDataValueQueries(innerValue, client, desiredType[\"set\"]),\n        );\n        return Promise.all(promiseArray);\n      }\n      break;\n    }\n    case \"object\": {\n      if (isOsdkBaseObject(value)) {\n        return value.$primaryKey;\n      }\n      break;\n    }\n    case \"objectSet\": {\n      // object set (the rid as a string (passes through the last return), or the ObjectSet definition directly)\n      if (isWireObjectSet(value)) {\n        return value;\n      }\n      if (isObjectSet(value)) {\n        return getWireObjectSet(value);\n      }\n      break;\n    }\n\n    case \"struct\": {\n      if (typeof value === \"object\") {\n        const structMap: { [key: string]: unknown } = {};\n        for (const [key, structValue] of Object.entries(value)) {\n          structMap[key] = await toDataValueQueries(\n            structValue,\n            client,\n            desiredType[\"struct\"][key],\n          );\n        }\n        return structMap;\n      }\n    }\n    case \"boolean\":\n    case \"date\":\n    case \"double\":\n    case \"float\":\n    case \"integer\":\n    case \"long\":\n    case \"string\":\n    case \"timestamp\":\n      return value;\n  }\n  return value;\n}\n"],"mappings":"AAAA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;;AAIA,OAAO,KAAKA,YAAY,MAAM,0BAA0B;AAExD,SACEC,gBAAgB,EAChBC,kBAAkB,QACb,+BAA+B;AACtC,SAASC,gBAAgB,EAAEC,WAAW,QAAQ,iCAAiC;AAC/E,SAASC,gBAAgB,QAAQ,uBAAuB;AACxD,SAASC,eAAe,QAAQ,oBAAoB;;AAEpD;AACA;AACA;AACA;AACA;AACA;AACA,OAAO,eAAeC,kBAAkBA,CACtCC,KAAc,EACdC,MAAqB,EACrBC,WAAoC,EAChB;EACpB,IAAIF,KAAK,IAAI,IAAI,EAAE;IACjB,OAAOA,KAAK;EACd;EAEA,IAAIG,KAAK,CAACC,OAAO,CAACJ,KAAK,CAAC,IAAIE,WAAW,CAACG,YAAY,EAAE;IACpD,MAAMC,MAAM,GAAGH,KAAK,CAACI,IAAI,CAACP,KAAK,CAAC;IAChC,IACEM,MAAM,CAACE,IAAI,CAAEC,SAAS,IACpBf,kBAAkB,CAACe,SAAS,CAAC,IAAIhB,gBAAgB,CAACgB,SAAS,CAC7D,CAAC,EACD;MACA,MAAMC,SAAS,GAAG,EAAE;MACpB,KAAK,MAAMV,KAAK,IAAIM,MAAM,EAAE;QAC1BI,SAAS,CAACC,IAAI,CAAC,MAAMZ,kBAAkB,CAACC,KAAK,EAAEC,MAAM,EAAEC,WAAW,CAAC,CAAC;MACtE;MACA,OAAOQ,SAAS;IAClB;IACA,MAAME,YAAY,GAAGT,KAAK,CAACI,IAAI,CAC7BP,KAAK,EACL,MAAOa,UAAU,IACf,MAAMd,kBAAkB,CAACc,UAAU,EAAEZ,MAAM,EAAEC,WAAW,CAC5D,CAAC;IACD,OAAOY,OAAO,CAACC,GAAG,CAACH,YAAY,CAAC;EAClC;EAEA,QAAQV,WAAW,CAACc,IAAI;IACtB,KAAK,YAAY;MAAE;QACjB,IAAItB,kBAAkB,CAACM,KAAK,CAAC,EAAE;UAC7B,MAAMiB,UAAU,GAAG,MAAMzB,YAAY,CAAC0B,WAAW,CAACC,MAAM,CACtDlB,MAAM,EACND,KAAK,CAACoB,IAAI,EACV;YACEC,QAAQ,EAAErB,KAAK,CAACsB;UAClB,CACF,CAAC;UACD,OAAOL,UAAU,CAACM,GAAG;QACvB;QAEA,IACE9B,gBAAgB,CAACO,KAAK,CAAC,EACvB;UACA,MAAMiB,UAAU,GAAG,MAAMzB,YAAY,CAAC0B,WAAW,CAACC,MAAM,CACtDlB,MAAM,EACND,KAAK,EACL;YACEqB,QAAQ,EAAErB,KAAK,CAACsB;UAClB,CACF,CAAC;UACD,OAAOL,UAAU,CAACM,GAAG;QACvB;;QAEA;QACA,OAAOvB,KAAK;MACd;IACA,KAAK,2BAA2B;MAAE;QAChC,OAAO;UACLwB,MAAM,EAAExB;QACV,CAAC;MACH;IACA,KAAK,6BAA6B;MAAE;QAClC,OAAO;UACLwB,MAAM,EAAExB;QACV,CAAC;MACH;IAEA,KAAK,KAAK;MAAE;QACV,IAAIA,KAAK,YAAYyB,GAAG,EAAE;UACxB,MAAMb,YAAY,GAAGT,KAAK,CAACI,IAAI,CAC7BP,KAAK,EACL,MAAOa,UAAU,IACf,MAAMd,kBAAkB,CAACc,UAAU,EAAEZ,MAAM,EAAEC,WAAW,CAAC,KAAK,CAAC,CACnE,CAAC;UACD,OAAOY,OAAO,CAACC,GAAG,CAACH,YAAY,CAAC;QAClC;QACA;MACF;IACA,KAAK,QAAQ;MAAE;QACb,IAAIf,gBAAgB,CAACG,KAAK,CAAC,EAAE;UAC3B,OAAOA,KAAK,CAAC0B,WAAW;QAC1B;QACA;MACF;IACA,KAAK,WAAW;MAAE;QAChB;QACA,IAAI5B,eAAe,CAACE,KAAK,CAAC,EAAE;UAC1B,OAAOA,KAAK;QACd;QACA,IAAIJ,WAAW,CAACI,KAAK,CAAC,EAAE;UACtB,OAAOL,gBAAgB,CAACK,KAAK,CAAC;QAChC;QACA;MACF;IAEA,KAAK,QAAQ;MAAE;QACb,IAAI,OAAOA,KAAK,KAAK,QAAQ,EAAE;UAC7B,MAAM2B,SAAqC,GAAG,CAAC,CAAC;UAChD,KAAK,MAAM,CAACC,GAAG,EAAEC,WAAW,CAAC,IAAIC,MAAM,CAACC,OAAO,CAAC/B,KAAK,CAAC,EAAE;YACtD2B,SAAS,CAACC,GAAG,CAAC,GAAG,MAAM7B,kBAAkB,CACvC8B,WAAW,EACX5B,MAAM,EACNC,WAAW,CAAC,QAAQ,CAAC,CAAC0B,GAAG,CAC3B,CAAC;UACH;UACA,OAAOD,SAAS;QAClB;MACF;IACA,KAAK,SAAS;IACd,KAAK,MAAM;IACX,KAAK,QAAQ;IACb,KAAK,OAAO;IACZ,KAAK,SAAS;IACd,KAAK,MAAM;IACX,KAAK,QAAQ;IACb,KAAK,WAAW;MACd,OAAO3B,KAAK;EAChB;EACA,OAAOA,KAAK;AACd","ignoreList":[]}
+{"version":3,"file":"toDataValueQueries.js","names":["OntologiesV2","isAttachmentFile","isAttachmentUpload","getWireObjectSet","isObjectSet","isOsdkBaseObject","isWireObjectSet","toDataValueQueries","value","client","desiredType","Array","isArray","multiplicity","values","from","some","dataValue","converted","push","promiseArray","innerValue","Promise","all","type","attachment","Attachments","upload","data","filename","name","rid","groups","Set","$primaryKey","entrySet","key","mapValue","Object","entries","keyType","extractPrimaryKeyFromObjectIdentifier","valueType","structMap","structValue","a","substring","indexOf"],"sources":["toDataValueQueries.ts"],"sourcesContent":["/*\n * Copyright 2024 Palantir Technologies, Inc. All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport type { QueryDataTypeDefinition } from \"@osdk/api\";\nimport { type DataValue } from \"@osdk/foundry.ontologies\";\nimport * as OntologiesV2 from \"@osdk/foundry.ontologies\";\nimport type { MinimalClient } from \"../MinimalClientContext.js\";\nimport {\n  isAttachmentFile,\n  isAttachmentUpload,\n} from \"../object/AttachmentUpload.js\";\nimport { getWireObjectSet, isObjectSet } from \"../objectSet/createObjectSet.js\";\nimport { isOsdkBaseObject } from \"./isOsdkBaseObject.js\";\nimport { isWireObjectSet } from \"./WireObjectSet.js\";\n\n/**\n * Marshall user-facing data into the wire DataValue type\n *\n * @see DataValue for the expected payloads\n * @internal\n */\nexport async function toDataValueQueries(\n  value: unknown,\n  client: MinimalClient,\n  desiredType: QueryDataTypeDefinition,\n): Promise<DataValue> {\n  if (value == null) {\n    return value;\n  }\n\n  if (Array.isArray(value) && desiredType.multiplicity) {\n    const values = Array.from(value);\n    if (\n      values.some((dataValue) =>\n        isAttachmentUpload(dataValue) || isAttachmentFile(dataValue)\n      )\n    ) {\n      const converted = [];\n      for (const value of values) {\n        converted.push(await toDataValueQueries(value, client, desiredType));\n      }\n      return converted;\n    }\n    const promiseArray = Array.from(\n      value,\n      async (innerValue) =>\n        await toDataValueQueries(innerValue, client, desiredType),\n    );\n    return Promise.all(promiseArray);\n  }\n\n  switch (desiredType.type) {\n    case \"attachment\": {\n      if (isAttachmentUpload(value)) {\n        const attachment = await OntologiesV2.Attachments.upload(\n          client,\n          value.data,\n          {\n            filename: value.name,\n          },\n        );\n        return attachment.rid;\n      }\n\n      if (\n        isAttachmentFile(value)\n      ) {\n        const attachment = await OntologiesV2.Attachments.upload(\n          client,\n          value,\n          {\n            filename: value.name as string,\n          },\n        );\n        return attachment.rid;\n      }\n\n      // If it's not an upload, it's just an attachment rid string which we can pass through\n      return value;\n    }\n    case \"twoDimensionalAggregation\": {\n      return {\n        groups: value,\n      };\n    }\n    case \"threeDimensionalAggregation\": {\n      return {\n        groups: value,\n      };\n    }\n\n    case \"set\": {\n      if (value instanceof Set) {\n        const promiseArray = Array.from(\n          value,\n          async (innerValue) =>\n            await toDataValueQueries(innerValue, client, desiredType[\"set\"]),\n        );\n        return Promise.all(promiseArray);\n      }\n      break;\n    }\n    case \"object\": {\n      if (isOsdkBaseObject(value)) {\n        return value.$primaryKey;\n      }\n      break;\n    }\n    case \"objectSet\": {\n      // object set (the rid as a string (passes through the last return), or the ObjectSet definition directly)\n      if (isWireObjectSet(value)) {\n        return value;\n      }\n      if (isObjectSet(value)) {\n        return getWireObjectSet(value);\n      }\n      break;\n    }\n\n    case \"map\": {\n      if (typeof value === \"object\") {\n        const entrySet: Array<{ key: any; value: any }> = [];\n        for (const [key, mapValue] of Object.entries(value)) {\n          entrySet.push({\n            key: desiredType.keyType.type === \"object\"\n              ? extractPrimaryKeyFromObjectIdentifier(key)\n              : await toDataValueQueries(\n                key,\n                client,\n                desiredType.keyType,\n              ),\n            value: await toDataValueQueries(\n              mapValue,\n              client,\n              desiredType.valueType,\n            ),\n          });\n        }\n        return entrySet;\n      }\n      break;\n    }\n\n    case \"struct\": {\n      if (typeof value === \"object\") {\n        const structMap: { [key: string]: unknown } = {};\n        for (const [key, structValue] of Object.entries(value)) {\n          structMap[key] = await toDataValueQueries(\n            structValue,\n            client,\n            desiredType[\"struct\"][key],\n          );\n        }\n        return structMap;\n      }\n    }\n\n    case \"boolean\":\n    case \"date\":\n    case \"double\":\n    case \"float\":\n    case \"integer\":\n    case \"long\":\n    case \"string\":\n    case \"timestamp\":\n      return value;\n  }\n  return value;\n}\n\nfunction extractPrimaryKeyFromObjectIdentifier(a: string) {\n  return a.substring(a.indexOf(\":\") + 1);\n}\n"],"mappings":"AAAA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;;AAIA,OAAO,KAAKA,YAAY,MAAM,0BAA0B;AAExD,SACEC,gBAAgB,EAChBC,kBAAkB,QACb,+BAA+B;AACtC,SAASC,gBAAgB,EAAEC,WAAW,QAAQ,iCAAiC;AAC/E,SAASC,gBAAgB,QAAQ,uBAAuB;AACxD,SAASC,eAAe,QAAQ,oBAAoB;;AAEpD;AACA;AACA;AACA;AACA;AACA;AACA,OAAO,eAAeC,kBAAkBA,CACtCC,KAAc,EACdC,MAAqB,EACrBC,WAAoC,EAChB;EACpB,IAAIF,KAAK,IAAI,IAAI,EAAE;IACjB,OAAOA,KAAK;EACd;EAEA,IAAIG,KAAK,CAACC,OAAO,CAACJ,KAAK,CAAC,IAAIE,WAAW,CAACG,YAAY,EAAE;IACpD,MAAMC,MAAM,GAAGH,KAAK,CAACI,IAAI,CAACP,KAAK,CAAC;IAChC,IACEM,MAAM,CAACE,IAAI,CAAEC,SAAS,IACpBf,kBAAkB,CAACe,SAAS,CAAC,IAAIhB,gBAAgB,CAACgB,SAAS,CAC7D,CAAC,EACD;MACA,MAAMC,SAAS,GAAG,EAAE;MACpB,KAAK,MAAMV,KAAK,IAAIM,MAAM,EAAE;QAC1BI,SAAS,CAACC,IAAI,CAAC,MAAMZ,kBAAkB,CAACC,KAAK,EAAEC,MAAM,EAAEC,WAAW,CAAC,CAAC;MACtE;MACA,OAAOQ,SAAS;IAClB;IACA,MAAME,YAAY,GAAGT,KAAK,CAACI,IAAI,CAC7BP,KAAK,EACL,MAAOa,UAAU,IACf,MAAMd,kBAAkB,CAACc,UAAU,EAAEZ,MAAM,EAAEC,WAAW,CAC5D,CAAC;IACD,OAAOY,OAAO,CAACC,GAAG,CAACH,YAAY,CAAC;EAClC;EAEA,QAAQV,WAAW,CAACc,IAAI;IACtB,KAAK,YAAY;MAAE;QACjB,IAAItB,kBAAkB,CAACM,KAAK,CAAC,EAAE;UAC7B,MAAMiB,UAAU,GAAG,MAAMzB,YAAY,CAAC0B,WAAW,CAACC,MAAM,CACtDlB,MAAM,EACND,KAAK,CAACoB,IAAI,EACV;YACEC,QAAQ,EAAErB,KAAK,CAACsB;UAClB,CACF,CAAC;UACD,OAAOL,UAAU,CAACM,GAAG;QACvB;QAEA,IACE9B,gBAAgB,CAACO,KAAK,CAAC,EACvB;UACA,MAAMiB,UAAU,GAAG,MAAMzB,YAAY,CAAC0B,WAAW,CAACC,MAAM,CACtDlB,MAAM,EACND,KAAK,EACL;YACEqB,QAAQ,EAAErB,KAAK,CAACsB;UAClB,CACF,CAAC;UACD,OAAOL,UAAU,CAACM,GAAG;QACvB;;QAEA;QACA,OAAOvB,KAAK;MACd;IACA,KAAK,2BAA2B;MAAE;QAChC,OAAO;UACLwB,MAAM,EAAExB;QACV,CAAC;MACH;IACA,KAAK,6BAA6B;MAAE;QAClC,OAAO;UACLwB,MAAM,EAAExB;QACV,CAAC;MACH;IAEA,KAAK,KAAK;MAAE;QACV,IAAIA,KAAK,YAAYyB,GAAG,EAAE;UACxB,MAAMb,YAAY,GAAGT,KAAK,CAACI,IAAI,CAC7BP,KAAK,EACL,MAAOa,UAAU,IACf,MAAMd,kBAAkB,CAACc,UAAU,EAAEZ,MAAM,EAAEC,WAAW,CAAC,KAAK,CAAC,CACnE,CAAC;UACD,OAAOY,OAAO,CAACC,GAAG,CAACH,YAAY,CAAC;QAClC;QACA;MACF;IACA,KAAK,QAAQ;MAAE;QACb,IAAIf,gBAAgB,CAACG,KAAK,CAAC,EAAE;UAC3B,OAAOA,KAAK,CAAC0B,WAAW;QAC1B;QACA;MACF;IACA,KAAK,WAAW;MAAE;QAChB;QACA,IAAI5B,eAAe,CAACE,KAAK,CAAC,EAAE;UAC1B,OAAOA,KAAK;QACd;QACA,IAAIJ,WAAW,CAACI,KAAK,CAAC,EAAE;UACtB,OAAOL,gBAAgB,CAACK,KAAK,CAAC;QAChC;QACA;MACF;IAEA,KAAK,KAAK;MAAE;QACV,IAAI,OAAOA,KAAK,KAAK,QAAQ,EAAE;UAC7B,MAAM2B,QAAyC,GAAG,EAAE;UACpD,KAAK,MAAM,CAACC,GAAG,EAAEC,QAAQ,CAAC,IAAIC,MAAM,CAACC,OAAO,CAAC/B,KAAK,CAAC,EAAE;YACnD2B,QAAQ,CAAChB,IAAI,CAAC;cACZiB,GAAG,EAAE1B,WAAW,CAAC8B,OAAO,CAAChB,IAAI,KAAK,QAAQ,GACtCiB,qCAAqC,CAACL,GAAG,CAAC,GAC1C,MAAM7B,kBAAkB,CACxB6B,GAAG,EACH3B,MAAM,EACNC,WAAW,CAAC8B,OACd,CAAC;cACHhC,KAAK,EAAE,MAAMD,kBAAkB,CAC7B8B,QAAQ,EACR5B,MAAM,EACNC,WAAW,CAACgC,SACd;YACF,CAAC,CAAC;UACJ;UACA,OAAOP,QAAQ;QACjB;QACA;MACF;IAEA,KAAK,QAAQ;MAAE;QACb,IAAI,OAAO3B,KAAK,KAAK,QAAQ,EAAE;UAC7B,MAAMmC,SAAqC,GAAG,CAAC,CAAC;UAChD,KAAK,MAAM,CAACP,GAAG,EAAEQ,WAAW,CAAC,IAAIN,MAAM,CAACC,OAAO,CAAC/B,KAAK,CAAC,EAAE;YACtDmC,SAAS,CAACP,GAAG,CAAC,GAAG,MAAM7B,kBAAkB,CACvCqC,WAAW,EACXnC,MAAM,EACNC,WAAW,CAAC,QAAQ,CAAC,CAAC0B,GAAG,CAC3B,CAAC;UACH;UACA,OAAOO,SAAS;QAClB;MACF;IAEA,KAAK,SAAS;IACd,KAAK,MAAM;IACX,KAAK,QAAQ;IACb,KAAK,OAAO;IACZ,KAAK,SAAS;IACd,KAAK,MAAM;IACX,KAAK,QAAQ;IACb,KAAK,WAAW;MACd,OAAOnC,KAAK;EAChB;EACA,OAAOA,KAAK;AACd;AAEA,SAASiC,qCAAqCA,CAACI,CAAS,EAAE;EACxD,OAAOA,CAAC,CAACC,SAAS,CAACD,CAAC,CAACE,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;AACxC","ignoreList":[]}

package/build/cjs/{Client-C8K3E1vH.d.cts → Client-DBTcM9gB.d.cts}

@@ -60,7 +60,7 @@ interface Client extends SharedClient, OldSharedClient {
     <Q extends (InterfaceDefinition)>(o: Q): unknown extends CompileTimeMetadata<Q>["objectSet"] ? MinimalObjectSet<Q> : CompileTimeMetadata<Q>["objectSet"];
     <Q extends ActionDefinition<any>>(o: Q): ActionSignatureFromDef<Q>;
     <Q extends QueryDefinition<any>>(o: Q): QuerySignatureFromDef<Q>;
-    <Q extends Experiment<"2.0.8"> | Experiment<"2.1.0">>(experiment: Q): ExperimentFns<Q>;
+    <Q extends Experiment<"2.0.8"> | Experiment<"2.1.0"> | Experiment<"2.2.0">>(experiment: Q): ExperimentFns<Q>;
     fetchMetadata<Q extends (ObjectTypeDefinition | InterfaceDefinition | ActionDefinition<any> | QueryDefinition<any>)>(o: Q): Promise<Q extends ObjectTypeDefinition ? ObjectMetadata : Q extends InterfaceDefinition ? InterfaceMetadata : Q extends ActionDefinition<any> ? ActionMetadata : Q extends QueryDefinition<any> ? QueryMetadata : never>;
 }