@osdk/client 2.2.0-beta.9 → 2.2.0-rc.25

package/build/browser/objectSet/ObjectSet.test.js

@@ -16,18 +16,20 @@
 
 import { isOk } from "@osdk/api";
 import { __EXPERIMENTAL__NOT_SUPPORTED_YET__fetchOneByRid, __EXPERIMENTAL__NOT_SUPPORTED_YET__fetchPageByRid } from "@osdk/api/unstable";
-import { $ontologyRid, BarInterface, BgaoNflPlayer, Employee, FooInterface, Office } from "@osdk/client.test.ontology";
-import { apiServer, stubData } from "@osdk/shared.test";
-import { afterAll, beforeAll, describe, expect, expectTypeOf, it } from "vitest";
+import { BarInterface, BgaoNflPlayer, Employee, FooInterface, objectTypeWithAllPropertyTypes, Office } from "@osdk/client.test.ontology";
+import { LegacyFauxFoundry, startNodeApiServer, stubData } from "@osdk/shared.test";
+import { beforeAll, describe, expect, expectTypeOf, it } from "vitest";
 import { createClient } from "../createClient.js";
 describe("ObjectSet", () => {
   let client;
-  beforeAll(async () => {
-    apiServer.listen();
-    client = createClient("https://stack.palantir.com", $ontologyRid, async () => "myAccessToken");
-  });
-  afterAll(() => {
-    apiServer.close();
+  beforeAll(() => {
+    const testSetup = startNodeApiServer(new LegacyFauxFoundry(), createClient);
+    ({
+      client
+    } = testSetup);
+    return () => {
+      testSetup.apiServer.close();
+    };
   });
   it("does not allow intersect/union/subtract with different object types", () => {
     const employeeObjectSet = client(Employee);
@@ -58,7 +60,7 @@ describe("ObjectSet", () => {
     expect(pks.has(stubData.employee2.employeeId)).toBe(true);
     expect(pks.has(stubData.employee3.employeeId)).toBe(true);
     expect(pks.has(stubData.employee4withDerived.employeeId)).toBe(true);
-    expect(pks.has(stubData.employeeFailsStrict.__primaryKey)).toBe(true);
+    expect(pks.has(stubData.employeePassesStrict.__primaryKey)).toBe(true);
     expect(pks.has(stubData.employee50050.employeeId)).toBe(true);
   });
   it("objects set subtract", async () => {
@@ -89,7 +91,7 @@ describe("ObjectSet", () => {
     expect(pks.has(stubData.employee2.employeeId)).toBe(true);
     expect(pks.has(stubData.employee3.employeeId)).toBe(true);
     expect(pks.has(stubData.employee4withDerived.employeeId)).toBe(true);
-    expect(pks.has(stubData.employeeFailsStrict.__primaryKey)).toBe(true);
+    expect(pks.has(stubData.employeePassesStrict.__primaryKey)).toBe(true);
     expect(pks.has(stubData.employee50050.employeeId)).toBe(true);
   });
   it("orders objects in ascending order without a filter, and returns all results", async () => {
@@ -104,7 +106,7 @@ describe("ObjectSet", () => {
         "employeeId": "asc"
       }
     });
-    expect(employees.map(e => e.$primaryKey)).toEqual([50030, 50031, 50032, 50035, stubData.employee50050.employeeId]);
+    expect(employees.map(e => e.$primaryKey)).toEqual([50030, 50031, 50032, 50033, 50035, stubData.employee50050.employeeId]);
   });
   it("allows fetching by PK from a base object set - fetchOne", async () => {
     const employee = await client(Employee).fetchOne(stubData.employee1.employeeId);
@@ -220,7 +222,7 @@ describe("ObjectSet", () => {
     expect(pks.has(stubData.employee2.employeeId)).toBe(true);
     expect(pks.has(stubData.employee3.employeeId)).toBe(true);
     expect(pks.has(stubData.employee4withDerived.employeeId)).toBe(true);
-    expect(pks.has(stubData.employeeFailsStrict.__primaryKey)).toBe(true);
+    expect(pks.has(stubData.employeePassesStrict.__primaryKey)).toBe(true);
     expect(pks.has(stubData.employee50050.employeeId)).toBe(true);
   });
   it("allows $in filter with ReadonlyArrays", () => {
@@ -400,30 +402,28 @@ describe("ObjectSet", () => {
     });
   });
   describe("Derived Properties Object Set", () => {
-    it("does not allow aggregate or selectProperty before a link type is selected", () => {
+    it("does not allow aggregate before a link type is selected", () => {
       client(Employee).withProperties({
         "derivedPropertyName": base =>
         // @ts-expect-error
         base.aggregate("employeeId:exactDistinct")
       });
-      client(Employee).withProperties({
-        "derivedPropertyName": base =>
-        // @ts-expect-error
-        base.selectProperty("employeeId")
-      });
     });
-    it("does not allow selectProperty when a many link was selected at any point", () => {
-      client(Employee).withProperties({
-        "derivedPropertyName": base => {
-          // @ts-expect-error
-          base.pivotTo("peeps").selectProperty("employeeId");
 
-          // @ts-expect-error
-          base.pivotTo("lead").pivotTo("peeps").selectProperty("employeeId");
-          return base.pivotTo("lead").selectProperty("employeeId");
-        }
-      });
-    });
+    // it("does not allow selectProperty when a many link was selected at any point", () => {
+    //   client(Employee).withProperties({
+    //     "derivedPropertyName": (base) => {
+    //       // @ts-expect-error
+    //       base.pivotTo("peeps").selectProperty("employeeId");
+
+    //       // @ts-expect-error
+    //       base.pivotTo("lead").pivotTo("peeps").selectProperty("employeeId");
+
+    //       return base.pivotTo("lead").selectProperty("employeeId");
+    //     },
+    //   });
+    // });
+
     it("enforces a return only of correct type", () => {
       client(Employee).withProperties({
         // @ts-expect-error
@@ -485,15 +485,23 @@ describe("ObjectSet", () => {
       });
       expectTypeOf(objectSet).branded.toEqualTypeOf();
     });
-    it("correctly narrows types of selectProperty function", () => {
-      client(Employee).withProperties({
-        "derivedPropertyName": base => {
-          // @ts-expect-error
-          base.pivotTo("lead").selectProperty("notAProperty");
-          return base.pivotTo("lead").selectProperty("employeeStatus");
-        }
-      });
-    });
+
+    // it("correctly narrows types of selectProperty function", () => {
+    //   client(Employee).withProperties({
+    //     "derivedPropertyName": (base) => {
+    //       // @ts-expect-error
+    //       base.pivotTo("lead").selectProperty("notAProperty");
+
+    //       return base.pivotTo("lead").selectProperty("employeeStatus");
+    //     },
+    //   }) satisfies ObjectSet<
+    //     Employee,
+    //     {
+    //       "derivedPropertyName": "stringTimeseries" | undefined;
+    //     }
+    //   >;
+    // });
+
     it("propagates derived property type to future object set operations with correct types", () => {
       client(Employee).withProperties({
         "derivedPropertyName": base => base.pivotTo("lead").aggregate("employeeId:sum")
@@ -539,47 +547,48 @@ describe("ObjectSet", () => {
         }
       });
       expectTypeOf(setAggregationObjectSet).toEqualTypeOf();
-      const selectPropertyObjectSet = client(Employee).withProperties({
-        "derivedPropertyName": base => base.pivotTo("lead").selectProperty("employeeId")
-      }).where({
-        "derivedPropertyName": {
-          "$eq": 3
-        }
-      });
-      expectTypeOf(selectPropertyObjectSet).toEqualTypeOf();
-      client(Employee).withProperties({
-        "derivedPropertyName": base => base.pivotTo("lead").selectProperty("startDate")
-      }).where({
-        "derivedPropertyName": {
-          "$eq": "datetimeFilter"
-        }
-      });
+
+      // const selectPropertyObjectSet = client(Employee).withProperties({
+      //   "derivedPropertyName": (base) =>
+      //     base.pivotTo("lead").selectProperty("employeeId"),
+      // }).where({ "derivedPropertyName": { "$eq": 3 } });
+
+      // expectTypeOf(selectPropertyObjectSet).toEqualTypeOf<
+      //   ObjectSet<Employee, {
+      //     derivedPropertyName: "integer";
+      //   }>
+      // >();
+
+      // client(Employee).withProperties({
+      //   "derivedPropertyName": (base) =>
+      //     base.pivotTo("lead").selectProperty("startDate"),
+      // }).where({ "derivedPropertyName": { "$eq": "datetimeFilter" } });
     });
     it("correctly types multiple property definitions in one clause", () => {
       const objectSet = client(Employee).withProperties({
         "derivedPropertyName": base => base.pivotTo("lead").aggregate("employeeId:sum"),
-        "derivedPropertyName2": base => base.pivotTo("lead").selectProperty("fullName")
+        "derivedPropertyName2": base => base.pivotTo("lead").aggregate("fullName:approximateDistinct")
       }).where({
         "derivedPropertyName": {
           "$eq": 3
         }
       }).where({
         "derivedPropertyName2": {
-          "$eq": "name"
+          "$eq": 5
         }
       });
       expectTypeOf(objectSet).toEqualTypeOf();
     });
     it("ensures other properties are consistently typed", () => {
       client(Employee).withProperties({
-        "derivedPropertyName": base => base.pivotTo("lead").selectProperty("employeeId")
+        "derivedPropertyName": base => base.pivotTo("lead").aggregate("employeeId:collectList")
       }).where({
         "fullName": {
           "$eq": "A"
         }
       });
       client(Employee).withProperties({
-        "derivedPropertyName": base => base.pivotTo("lead").selectProperty("employeeId")
+        "derivedPropertyName": base => base.pivotTo("lead").aggregate("employeeId:collectList")
       }).where({
         "employeeId": {
           "$eq": 2
@@ -588,16 +597,129 @@ describe("ObjectSet", () => {
     });
     it("allows fetching derived properties with correctly typed Osdk.Instance types", async () => {
       const objectWithRdp = await client(Employee).withProperties({
-        "derivedPropertyName": base => base.pivotTo("lead").selectProperty("employeeId")
+        "derivedPropertyName": base => base.pivotTo("lead").aggregate("employeeId:collectList")
       }).fetchOne(stubData.employee1.employeeId);
       expectTypeOf(objectWithRdp.derivedPropertyName).toEqualTypeOf();
-      expect(objectWithRdp.derivedPropertyName).toBe(stubData.employee2.__primaryKey);
-      const objectWithUndefinedRdp = await client(Employee).withProperties({
-        "derivedPropertyName": base => base.pivotTo("lead").selectProperty("employeeId")
-      }).fetchOne(stubData.employee2.employeeId, {
-        $select: ["derivedPropertyName"]
-      });
-      expect(objectWithUndefinedRdp.derivedPropertyName).toBeUndefined();
+      expect(objectWithRdp.derivedPropertyName).toEqual([stubData.employee2.__primaryKey]);
+
+      // const objectWithUndefinedRdp = await client(Employee).withProperties({
+      //   "derivedPropertyName": (base) =>
+      //     base.pivotTo("lead").aggregate("employeeId:collectList"),
+      // }).fetchOne(stubData.employee2.employeeId, {
+      //   $select: ["derivedPropertyName"],
+      // });
+
+      // expect(objectWithUndefinedRdp.derivedPropertyName).toBeUndefined();
+    });
+    it("correctly deserializes attachments and geo properties", async () => {
+      const objectWithRdp = await client(objectTypeWithAllPropertyTypes).withProperties({
+        "attachmentSelectDp": base => base.pivotTo("linkedObjectType").aggregate("attachment:collectSet"),
+        "geoSelectDp": base => base.pivotTo("linkedObjectType").aggregate("geoShape:collectSet"),
+        "geoCollectListDp": base => base.pivotTo("linkedObjectType").aggregate("geoShapeArray:collectList")
+      }).fetchOne(5);
+      expectTypeOf(objectWithRdp.attachmentSelectDp).toEqualTypeOf();
+      expect(objectWithRdp.attachmentSelectDp).toMatchInlineSnapshot(`
+        [
+          {
+            "fetchContents": [Function],
+            "fetchMetadata": [Function],
+            "rid": "ri.attachments.main.attachment.86016861-707f-4292-b258-6a7108915a75",
+          },
+        ]
+      `);
+      expectTypeOf(objectWithRdp.geoSelectDp).toEqualTypeOf();
+      expect(objectWithRdp.geoSelectDp).toMatchInlineSnapshot(`
+        [
+          {
+            "coordinates": [
+              [
+                [
+                  1,
+                  1,
+                ],
+                [
+                  1,
+                  2,
+                ],
+                [
+                  2,
+                  2,
+                ],
+                [
+                  2,
+                  1,
+                ],
+                [
+                  1,
+                  1,
+                ],
+              ],
+            ],
+            "type": "Polygon",
+          },
+        ]
+      `);
+      expectTypeOf(objectWithRdp.geoCollectListDp).toEqualTypeOf();
+      expect(objectWithRdp.geoCollectListDp).toMatchInlineSnapshot(`
+        [
+          [
+            {
+              "coordinates": [
+                [
+                  [
+                    1,
+                    1,
+                  ],
+                  [
+                    1,
+                    2,
+                  ],
+                  [
+                    2,
+                    2,
+                  ],
+                  [
+                    2,
+                    1,
+                  ],
+                  [
+                    1,
+                    1,
+                  ],
+                ],
+              ],
+              "type": "Polygon",
+            },
+          ],
+        ]
+      `);
+
+      // Tests that we deserialize properly for fetchPage as well
+      const fetchPageTest = await client(objectTypeWithAllPropertyTypes).withProperties({
+        "attachmentSelectDp": base => base.pivotTo("linkedObjectType").aggregate("attachment:collectList"),
+        "geoSelectDp": base => base.pivotTo("linkedObjectType").aggregate("geoShape:collectList"),
+        "geoCollectListDp": base => base.pivotTo("linkedObjectType").aggregate("geoShapeArray:collectList")
+      }).where({
+        "id": {
+          "$eq": 5
+        }
+      }).fetchPage();
+      expect(fetchPageTest.data[0].attachmentSelectDp).toMatchInlineSnapshot(`
+        [
+          {
+            "fetchContents": [Function],
+            "fetchMetadata": [Function],
+            "rid": "ri.attachments.main.attachment.86016861-707f-4292-b258-6a7108915a75",
+          },
+        ]
+      `);
+    });
+    it("correctly deserializes count", async () => {
+      const objectWithRdp = await client(Employee).withProperties({
+        "derivedPropertyName": base => base.pivotTo("lead").aggregate("$count")
+      }).fetchOne(stubData.employee1.employeeId);
+      expectTypeOf(objectWithRdp.derivedPropertyName).toEqualTypeOf();
+      expect(objectWithRdp.derivedPropertyName).toEqual(1);
     });
   });