@snowtop/ent 0.0.2 → 0.0.3-5.alpha.2

package/testutils/fake_data/fake_user.js

@@ -31,6 +31,8 @@ class FakeUser {
                 privacy_1.AllowIfViewerRule,
                 //can view user if friends
                 new privacy_1.AllowIfViewerInboundEdgeExistsRule(internal_1.EdgeType.UserToFriends),
+                //can view user if following
+                new privacy_1.AllowIfViewerInboundEdgeExistsRule(internal_1.EdgeType.UserToFollowing),
                 new privacy_1.AllowIfConditionAppliesRule((viewer, ent) => {
                     if (!(viewer instanceof ViewerWithAccessToken)) {
                         return false;
@@ -48,8 +50,8 @@ class FakeUser {
         };
         this.data = data;
         this.id = data.id;
-        this.createdAt = convert_1.convertDate(data.created_at);
-        this.updatedAt = convert_1.convertDate(data.updated_at);
+        this.createdAt = (0, convert_1.convertDate)(data.created_at);
+        this.updatedAt = (0, convert_1.convertDate)(data.updated_at);
         this.firstName = data.first_name;
         this.lastName = data.last_name;
         this.emailAddress = data.email_address;
@@ -69,7 +71,7 @@ class FakeUser {
         ];
     }
     static getTestTable() {
-        return test_db_1.table("fake_users", test_db_1.uuid("id", { primaryKey: true }), test_db_1.timestamptz("created_at"), test_db_1.timestamptz("updated_at"), test_db_1.text("first_name"), test_db_1.text("last_name"), test_db_1.text("email_address"), test_db_1.text("phone_number"), test_db_1.text("password"));
+        return (0, test_db_1.table)("fake_users", (0, test_db_1.uuid)("id", { primaryKey: true }), (0, test_db_1.timestamptz)("created_at"), (0, test_db_1.timestamptz)("updated_at"), (0, test_db_1.text)("first_name"), (0, test_db_1.text)("last_name"), (0, test_db_1.text)("email_address"), (0, test_db_1.text)("phone_number"), (0, test_db_1.text)("password"));
     }
     static loaderOptions() {
         return {
@@ -80,10 +82,10 @@ class FakeUser {
         };
     }
     static async load(v, id) {
-        return ent_1.loadEnt(v, id, FakeUser.loaderOptions());
+        return (0, ent_1.loadEnt)(v, id, FakeUser.loaderOptions());
     }
     static async loadX(v, id) {
-        return ent_1.loadEntX(v, id, FakeUser.loaderOptions());
+        return (0, ent_1.loadEntX)(v, id, FakeUser.loaderOptions());
     }
 }
 exports.FakeUser = FakeUser;
@@ -92,19 +94,19 @@ class FakeUserSchema extends schema_1.BaseEntSchema {
         super(...arguments);
         this.ent = FakeUser;
         this.fields = [
-            schema_1.StringType({
+            (0, schema_1.StringType)({
                 name: "firstName",
             }),
-            schema_1.StringType({
+            (0, schema_1.StringType)({
                 name: "lastName",
             }),
-            schema_1.StringType({
+            (0, schema_1.StringType)({
                 name: "emailAddress",
             }),
-            schema_1.StringType({
+            (0, schema_1.StringType)({
                 name: "phoneNumber",
             }),
-            schema_1.StringType({
+            (0, schema_1.StringType)({
                 name: "password",
                 nullable: true,
             }),

package/testutils/fake_data/test_helpers.d.ts

@@ -1,14 +1,18 @@
-import { Data } from "../../core/base";
+import { Data, Ent } from "../../core/base";
 import { AssocEdge } from "../../core/ent";
 import { TempDB } from "../db/test_db";
 import { FakeUser, UserCreateInput, ContactCreateInput, FakeContact } from ".";
 import { EventCreateInput, FakeEvent } from "./fake_event";
+import { BuilderSchema } from "../builder";
 export declare function getContactInput(user: FakeUser, input?: Partial<ContactCreateInput>): ContactCreateInput;
 export declare function getUserInput(input?: Partial<UserCreateInput>): UserCreateInput;
 export declare function getEventInput(user: FakeUser, input?: Partial<EventCreateInput>): EventCreateInput;
 export declare function createTestUser(input?: Partial<UserCreateInput>): Promise<FakeUser>;
 export declare const inputs: Partial<ContactCreateInput>[];
 export declare function createAllContacts(input?: Partial<UserCreateInput>, slice?: number): Promise<[FakeUser, FakeContact[]]>;
+export declare function createUserPlusFriendRequests(input?: Partial<UserCreateInput>, slice?: number): Promise<[FakeUser, FakeUser[]]>;
+export declare function addEdge<T extends Ent>(source: T, schema: BuilderSchema<T>, edgeType: string, inbound: boolean, // inbound or outbound
+...dest: Ent[]): Promise<void>;
 export declare function verifyUserToContactEdges(user: FakeUser, edges: AssocEdge[], contacts: FakeContact[]): void;
 export declare function verifyUserToContactRawData(user: FakeUser, edges: Data[], contacts: FakeContact[]): void;
 export declare function verifyUserToContacts(user: FakeUser, ents: FakeContact[], contacts: FakeContact[]): void;

package/testutils/fake_data/test_helpers.js

@@ -1,7 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.createAllEvents = exports.tempDBTables = exports.setupTempDB = exports.createTestEvent = exports.edgeTableNames = exports.createEdges = exports.verifyUserToContacts = exports.verifyUserToContactRawData = exports.verifyUserToContactEdges = exports.createAllContacts = exports.inputs = exports.createTestUser = exports.getEventInput = exports.getUserInput = exports.getContactInput = void 0;
-const assert_1 = require("assert");
+exports.createAllEvents = exports.tempDBTables = exports.setupTempDB = exports.createTestEvent = exports.edgeTableNames = exports.createEdges = exports.verifyUserToContacts = exports.verifyUserToContactRawData = exports.verifyUserToContactEdges = exports.addEdge = exports.createUserPlusFriendRequests = exports.createAllContacts = exports.inputs = exports.createTestUser = exports.getEventInput = exports.getUserInput = exports.getContactInput = void 0;
 const jest_date_mock_1 = require("jest-date-mock");
 const viewer_1 = require("../../core/viewer");
 const ent_1 = require("../../core/ent");
@@ -12,6 +11,8 @@ const _1 = require(".");
 const fake_event_1 = require("./fake_event");
 const const_1 = require("./const");
 const mock_date_1 = require("./../mock_date");
+const builder_1 = require("../builder");
+const action_1 = require("../../action");
 function getContactInput(user, input) {
     return {
         firstName: "Jon",
@@ -45,7 +46,7 @@ function getEventInput(user, input) {
 }
 exports.getEventInput = getEventInput;
 async function createTestUser(input) {
-    const user = await _1.createUser(new viewer_1.LoggedOutViewer(), {
+    const user = await (0, _1.createUser)(new viewer_1.LoggedOutViewer(), {
         firstName: "Jon",
         lastName: "Snow",
         password: "12345678",
@@ -54,7 +55,7 @@ async function createTestUser(input) {
         ...input,
     });
     if (!user) {
-        assert_1.fail("error creating user");
+        throw new Error("error creating user");
     }
     return user;
 }
@@ -86,8 +87,8 @@ async function createAllContacts(input, slice) {
     let userInputs = exports.inputs.slice(0, slice || exports.inputs.length);
     const contacts = await Promise.all(userInputs.map(async (input) => {
         // just to make times deterministic so that tests can consistently work
-        jest_date_mock_1.advanceBy(100);
-        const builder = _1.getContactBuilder(user.viewer, getContactInput(user, input));
+        (0, jest_date_mock_1.advanceBy)(100);
+        const builder = (0, _1.getContactBuilder)(user.viewer, getContactInput(user, input));
         // add edge from user to contact
         builder.orchestrator.addInboundEdge(user.id, _1.EdgeType.UserToContacts, const_1.NodeType.FakeUser, {
             time: new Date(), // set time to advanceBy time
@@ -99,6 +100,38 @@ async function createAllContacts(input, slice) {
     return [user, contacts];
 }
 exports.createAllContacts = createAllContacts;
+async function createUserPlusFriendRequests(input, slice) {
+    const user = await createTestUser(input);
+    let userInputs = exports.inputs.slice(0, slice || exports.inputs.length);
+    const friendRequests = await Promise.all(userInputs.map(async (input) => {
+        return createTestUser(input);
+    }));
+    expect(friendRequests.length).toBe(userInputs.length);
+    await addEdge(user, new _1.FakeUserSchema(), _1.EdgeType.UserToFriendRequests, true, ...friendRequests);
+    return [user, friendRequests];
+}
+exports.createUserPlusFriendRequests = createUserPlusFriendRequests;
+async function addEdge(source, schema, edgeType, inbound, // inbound or outbound
+...dest) {
+    const action = new builder_1.SimpleAction(source.viewer, schema, new Map(), action_1.WriteOperation.Edit, source);
+    dest.forEach(async (friendRequest) => {
+        // just to make times deterministic so that tests can consistently work
+        (0, jest_date_mock_1.advanceBy)(100);
+        // add edge
+        if (inbound) {
+            action.builder.orchestrator.addInboundEdge(friendRequest.id, edgeType, dest[0].nodeType, {
+                time: new Date(), // set time to advanceBy time
+            });
+        }
+        else {
+            action.builder.orchestrator.addOutboundEdge(friendRequest.id, edgeType, dest[0].nodeType, {
+                time: new Date(), // set time to advanceBy time
+            });
+        }
+    });
+    await action.saveX();
+}
+exports.addEdge = addEdge;
 function verifyUserToContactEdges(user, edges, contacts) {
     expect(edges.length).toBe(contacts.length);
     for (let i = 0; i < contacts.length; i++) {
@@ -138,10 +171,10 @@ async function createEdges() {
     const edges = Object.values(_1.EdgeType);
     for (let i = 0; i < edges.length; i++) {
         const edge = edges[i];
-        await write_1.createRowForTest({
+        await (0, write_1.createRowForTest)({
             tableName: "assoc_edge_config",
             fields: {
-                edge_table: snake_case_1.snakeCase(`${edge}_table`),
+                edge_table: (0, snake_case_1.snakeCase)(`${edge}_table`),
                 symmetric_edge: _1.SymmetricEdges.has(edge),
                 inverse_edge_type: _1.InverseEdges.get(edge) || null,
                 edge_type: edge,
@@ -150,19 +183,19 @@ async function createEdges() {
                 updated_at: new Date(),
             },
         });
-        const edgeData = await ent_1.loadEdgeData(edge);
+        const edgeData = await (0, ent_1.loadEdgeData)(edge);
         expect(edgeData).toBeDefined();
     }
 }
 exports.createEdges = createEdges;
 function edgeTableNames() {
     const edges = Object.values(_1.EdgeType);
-    return edges.map((edge) => snake_case_1.snakeCase(`${edge}_table`));
+    return edges.map((edge) => (0, snake_case_1.snakeCase)(`${edge}_table`));
 }
 exports.edgeTableNames = edgeTableNames;
 async function createTestEvent(user, input) {
     const vc = new viewer_1.IDViewer(user.id);
-    const builder = fake_event_1.getEventBuilder(vc, {
+    const builder = (0, fake_event_1.getEventBuilder)(vc, {
         startTime: new Date(),
         location: "fun house",
         description: "fun fun fun",
@@ -188,9 +221,9 @@ function tempDBTables() {
         _1.FakeUser.getTestTable(),
         _1.FakeContact.getTestTable(),
         fake_event_1.FakeEvent.getTestTable(),
-        test_db_1.assoc_edge_config_table(),
+        (0, test_db_1.assoc_edge_config_table)(),
     ];
-    edgeTableNames().forEach((tableName) => tables.push(test_db_1.assoc_edge_table(tableName)));
+    edgeTableNames().forEach((tableName) => tables.push((0, test_db_1.assoc_edge_table)(tableName)));
     return tables;
 }
 exports.tempDBTables = tempDBTables;
@@ -199,14 +232,14 @@ async function createAllEvents(opts) {
     let arr = new Array(opts.howMany);
     arr.fill(1);
     // start at date in case something else has used a date already
-    jest_date_mock_1.advanceTo(mock_date_1.MockDate.getDate());
+    (0, jest_date_mock_1.advanceTo)(mock_date_1.MockDate.getDate());
     const events = await Promise.all(arr.map(async (v, idx) => {
         // just to make times deterministic so that tests can consistently work
         if (opts.interval > 0) {
-            jest_date_mock_1.advanceBy(opts.interval);
+            (0, jest_date_mock_1.advanceBy)(opts.interval);
         }
         const input = opts.eventInputs?.[idx];
-        const builder = fake_event_1.getEventBuilder(user.viewer, getEventInput(user, input));
+        const builder = (0, fake_event_1.getEventBuilder)(user.viewer, getEventInput(user, input));
         await builder.saveX();
         return await builder.editedEntX();
     }));

package/testutils/fake_data/user_query.d.ts

@@ -1,4 +1,4 @@
-import { ID, Viewer } from "../../core/base";
+import { Ent, ID, Viewer } from "../../core/base";
 import { CustomEdgeQueryBase } from "../../core/query/custom_query";
 import { AssocEdge } from "../../core/ent";
 import * as clause from "../../core/clause";
@@ -10,16 +10,19 @@ import { QueryLoaderFactory } from "../../core/loaders/query_loader";
 export declare class UserToContactsQuery extends AssocEdgeQueryBase<FakeUser, FakeContact, AssocEdge> {
     constructor(viewer: Viewer, src: EdgeQuerySource<FakeUser>);
     static query(viewer: Viewer, src: EdgeQuerySource<FakeUser>): UserToContactsQuery;
+    sourceEnt(id: ID): Promise<FakeUser | null>;
 }
 export declare const userToContactsCountLoaderFactory: RawCountLoaderFactory;
 export declare const userToContactsDataLoaderFactory: IndexLoaderFactory;
-export declare class UserToContactsFkeyQuery extends CustomEdgeQueryBase<FakeContact> {
+export declare class UserToContactsFkeyQuery extends CustomEdgeQueryBase<FakeUser, FakeContact> {
     constructor(viewer: Viewer, src: ID | FakeUser);
     static query(viewer: Viewer, src: FakeUser | ID): UserToContactsFkeyQuery;
+    sourceEnt(id: ID): Promise<FakeUser | null>;
 }
 export declare class UserToFriendsQuery extends AssocEdgeQueryBase<FakeUser, FakeUser, AssocEdge> {
     constructor(viewer: Viewer, src: EdgeQuerySource<FakeUser>);
     static query(viewer: Viewer, src: EdgeQuerySource<FakeUser>): UserToFriendsQuery;
+    sourceEnt(id: ID): Promise<FakeUser | null>;
     queryContacts(): UserToContactsQuery;
     queryFriends(): UserToFriendsQuery;
     queryHostedEvents(): UserToHostedEventsQuery;
@@ -32,6 +35,7 @@ export declare class CustomEdge extends AssocEdge {
 export declare class UserToCustomEdgeQuery extends AssocEdgeQueryBase<FakeUser, FakeUser, CustomEdge> {
     constructor(viewer: Viewer, src: EdgeQuerySource<FakeUser>);
     static query(viewer: Viewer, src: EdgeQuerySource<FakeUser>): UserToCustomEdgeQuery;
+    sourceEnt(id: ID): Promise<FakeUser | null>;
     queryContacts(): UserToContactsQuery;
     queryFriends(): UserToFriendsQuery;
     queryHostedEvents(): UserToHostedEventsQuery;
@@ -40,6 +44,7 @@ export declare class UserToCustomEdgeQuery extends AssocEdgeQueryBase<FakeUser,
 export declare class UserToFriendRequestsQuery extends AssocEdgeQueryBase<FakeUser, FakeUser, AssocEdge> {
     constructor(viewer: Viewer, src: EdgeQuerySource<FakeUser>);
     static query(viewer: Viewer, src: EdgeQuerySource<FakeUser>): UserToFriendRequestsQuery;
+    sourceEnt(id: ID): Promise<FakeUser | null>;
     queryContacts(): UserToContactsQuery;
     queryFriends(): UserToFriendsQuery;
     queryHostedEvents(): UserToHostedEventsQuery;
@@ -47,8 +52,10 @@ export declare class UserToFriendRequestsQuery extends AssocEdgeQueryBase<FakeUs
     queryCustomEdge(): UserToCustomEdgeQuery;
 }
 export declare class UserToIncomingFriendRequestsQuery extends AssocEdgeQueryBase<FakeUser, FakeUser, AssocEdge> {
-    constructor(viewer: Viewer, src: EdgeQuerySource<FakeUser>);
-    static query(viewer: Viewer, src: EdgeQuerySource<FakeUser>): UserToIncomingFriendRequestsQuery;
+    constructor(viewer: Viewer, src: EdgeQuerySource<FakeUser, FakeUser>);
+    getPrivacyPolicy(): import("../../core/base").PrivacyPolicy;
+    sourceEnt(id: ID): Promise<FakeUser | null>;
+    static query(viewer: Viewer, src: EdgeQuerySource<FakeUser, FakeUser>): UserToIncomingFriendRequestsQuery;
     queryContacts(): UserToContactsQuery;
     queryFriends(): UserToFriendsQuery;
     queryHostedEvents(): UserToHostedEventsQuery;
@@ -56,8 +63,9 @@ export declare class UserToIncomingFriendRequestsQuery extends AssocEdgeQueryBas
     queryCustomEdge(): UserToCustomEdgeQuery;
 }
 export declare class UserToEventsAttendingQuery extends AssocEdgeQueryBase<FakeUser, FakeEvent, AssocEdge> {
-    constructor(viewer: Viewer, src: EdgeQuerySource<FakeUser>);
-    static query(viewer: Viewer, src: EdgeQuerySource<FakeUser>): UserToEventsAttendingQuery;
+    constructor(viewer: Viewer, src: EdgeQuerySource<FakeUser, FakeEvent>);
+    static query(viewer: Viewer, src: EdgeQuerySource<FakeUser, FakeEvent>): UserToEventsAttendingQuery;
+    sourceEnt(id: ID): Promise<FakeUser | null>;
     queryHosts(): EventToHostsQuery;
     queryAttendees(): EventToAttendeesQuery;
     queryInvited(): EventToInvitedQuery;
@@ -65,8 +73,9 @@ export declare class UserToEventsAttendingQuery extends AssocEdgeQueryBase<FakeU
     queryMaybe(): EventToMaybeQuery;
 }
 export declare class UserToHostedEventsQuery extends AssocEdgeQueryBase<FakeUser, FakeEvent, AssocEdge> {
-    constructor(viewer: Viewer, src: EdgeQuerySource<FakeUser>);
-    static query(viewer: Viewer, src: EdgeQuerySource<FakeUser>): UserToHostedEventsQuery;
+    constructor(viewer: Viewer, src: EdgeQuerySource<FakeUser, FakeEvent>);
+    static query(viewer: Viewer, src: EdgeQuerySource<FakeUser, FakeEvent>): UserToHostedEventsQuery;
+    sourceEnt(id: ID): Promise<FakeUser | null>;
     queryHosts(): EventToHostsQuery;
     queryAttendees(): EventToAttendeesQuery;
     queryInvited(): EventToInvitedQuery;
@@ -77,7 +86,14 @@ export declare const getNextWeekClause: () => clause.Clause;
 export declare function getCompleteClause(id: ID): clause.Clause;
 export declare const userToEventsInNextWeekCountLoaderFactory: RawCountLoaderFactory;
 export declare const userToEventsInNextWeekDataLoaderFactory: QueryLoaderFactory<unknown>;
-export declare class UserToEventsInNextWeekQuery extends CustomEdgeQueryBase<FakeEvent> {
+export declare class UserToEventsInNextWeekQuery extends CustomEdgeQueryBase<FakeUser, FakeEvent> {
     constructor(viewer: Viewer, src: ID | FakeUser);
     static query(viewer: Viewer, src: FakeUser | ID): UserToEventsInNextWeekQuery;
+    sourceEnt(id: ID): Promise<FakeUser | null>;
+    getPrivacyPolicy(): import("../../core/base").PrivacyPolicy;
+}
+export declare class UserToFollowingQuery extends AssocEdgeQueryBase<FakeUser, Ent, AssocEdge> {
+    constructor(viewer: Viewer, src: EdgeQuerySource<FakeUser, FakeUser>);
+    static query(viewer: Viewer, src: EdgeQuerySource<FakeUser, FakeUser>): UserToFollowingQuery;
+    sourceEnt(id: ID): Promise<FakeUser | null>;
 }

package/testutils/fake_data/user_query.js

@@ -19,7 +19,7 @@ var __importStar = (this && this.__importStar) || function (mod) {
     return result;
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.UserToEventsInNextWeekQuery = exports.userToEventsInNextWeekDataLoaderFactory = exports.userToEventsInNextWeekCountLoaderFactory = exports.getCompleteClause = exports.getNextWeekClause = exports.UserToHostedEventsQuery = exports.UserToEventsAttendingQuery = exports.UserToIncomingFriendRequestsQuery = exports.UserToFriendRequestsQuery = exports.UserToCustomEdgeQuery = exports.CustomEdge = exports.UserToFriendsQuery = exports.UserToContactsFkeyQuery = exports.userToContactsDataLoaderFactory = exports.userToContactsCountLoaderFactory = exports.UserToContactsQuery = void 0;
+exports.UserToFollowingQuery = exports.UserToEventsInNextWeekQuery = exports.userToEventsInNextWeekDataLoaderFactory = exports.userToEventsInNextWeekCountLoaderFactory = exports.getCompleteClause = exports.getNextWeekClause = exports.UserToHostedEventsQuery = exports.UserToEventsAttendingQuery = exports.UserToIncomingFriendRequestsQuery = exports.UserToFriendRequestsQuery = exports.UserToCustomEdgeQuery = exports.CustomEdge = exports.UserToFriendsQuery = exports.UserToContactsFkeyQuery = exports.userToContactsDataLoaderFactory = exports.userToContactsCountLoaderFactory = exports.UserToContactsQuery = void 0;
 const custom_query_1 = require("../../core/query/custom_query");
 const ent_1 = require("../../core/ent");
 const clause = __importStar(require("../../core/clause"));
@@ -34,6 +34,8 @@ const jest_date_mock_1 = require("jest-date-mock");
 const luxon_1 = require("luxon");
 const query_loader_1 = require("../../core/loaders/query_loader");
 const mock_date_1 = require("./../mock_date");
+const _1 = require(".");
+const privacy_1 = require("../../core/privacy");
 class UserToContactsQuery extends assoc_query_1.AssocEdgeQueryBase {
     constructor(viewer, src) {
         super(viewer, src, new assoc_count_loader_1.AssocEdgeCountLoaderFactory(internal_1.EdgeType.UserToContacts), new assoc_edge_loader_1.AssocEdgeLoaderFactory(internal_1.EdgeType.UserToContacts, ent_1.AssocEdge), internal_1.FakeContact.loaderOptions());
@@ -41,6 +43,9 @@ class UserToContactsQuery extends assoc_query_1.AssocEdgeQueryBase {
     static query(viewer, src) {
         return new UserToContactsQuery(viewer, src);
     }
+    sourceEnt(id) {
+        return internal_1.FakeUser.load(this.viewer, id);
+    }
 }
 exports.UserToContactsQuery = UserToContactsQuery;
 exports.userToContactsCountLoaderFactory = new raw_count_loader_1.RawCountLoaderFactory({
@@ -63,6 +68,9 @@ class UserToContactsFkeyQuery extends custom_query_1.CustomEdgeQueryBase {
     static query(viewer, src) {
         return new UserToContactsFkeyQuery(viewer, src);
     }
+    sourceEnt(id) {
+        return internal_1.FakeUser.load(this.viewer, id);
+    }
 }
 exports.UserToContactsFkeyQuery = UserToContactsFkeyQuery;
 class UserToFriendsQuery extends assoc_query_1.AssocEdgeQueryBase {
@@ -72,6 +80,9 @@ class UserToFriendsQuery extends assoc_query_1.AssocEdgeQueryBase {
     static query(viewer, src) {
         return new UserToFriendsQuery(viewer, src);
     }
+    sourceEnt(id) {
+        return internal_1.FakeUser.load(this.viewer, id);
+    }
     queryContacts() {
         return UserToContactsQuery.query(this.viewer, this);
     }
@@ -103,6 +114,9 @@ class UserToCustomEdgeQuery extends assoc_query_1.AssocEdgeQueryBase {
     static query(viewer, src) {
         return new UserToCustomEdgeQuery(viewer, src);
     }
+    sourceEnt(id) {
+        return internal_1.FakeUser.load(this.viewer, id);
+    }
     queryContacts() {
         return UserToContactsQuery.query(this.viewer, this);
     }
@@ -124,6 +138,9 @@ class UserToFriendRequestsQuery extends assoc_query_1.AssocEdgeQueryBase {
     static query(viewer, src) {
         return new UserToFriendRequestsQuery(viewer, src);
     }
+    sourceEnt(id) {
+        return internal_1.FakeUser.load(this.viewer, id);
+    }
     queryContacts() {
         return UserToContactsQuery.query(this.viewer, this);
     }
@@ -145,6 +162,12 @@ class UserToIncomingFriendRequestsQuery extends assoc_query_1.AssocEdgeQueryBase
     constructor(viewer, src) {
         super(viewer, src, new assoc_count_loader_1.AssocEdgeCountLoaderFactory(internal_1.EdgeType.UserToIncomingFriendRequests), new assoc_edge_loader_1.AssocEdgeLoaderFactory(internal_1.EdgeType.UserToIncomingFriendRequests, ent_1.AssocEdge), internal_1.FakeUser.loaderOptions());
     }
+    getPrivacyPolicy() {
+        return privacy_1.AllowIfViewerPrivacyPolicy;
+    }
+    sourceEnt(id) {
+        return internal_1.FakeUser.load(this.viewer, id);
+    }
     static query(viewer, src) {
         return new UserToIncomingFriendRequestsQuery(viewer, src);
     }
@@ -172,6 +195,9 @@ class UserToEventsAttendingQuery extends assoc_query_1.AssocEdgeQueryBase {
     static query(viewer, src) {
         return new UserToEventsAttendingQuery(viewer, src);
     }
+    sourceEnt(id) {
+        return internal_1.FakeUser.load(this.viewer, id);
+    }
     queryHosts() {
         return internal_1.EventToHostsQuery.query(this.viewer, this);
     }
@@ -196,6 +222,9 @@ class UserToHostedEventsQuery extends assoc_query_1.AssocEdgeQueryBase {
     static query(viewer, src) {
         return new UserToHostedEventsQuery(viewer, src);
     }
+    sourceEnt(id) {
+        return internal_1.FakeUser.load(this.viewer, id);
+    }
     queryHosts() {
         return internal_1.EventToHostsQuery.query(this.viewer, this);
     }
@@ -215,7 +244,7 @@ class UserToHostedEventsQuery extends assoc_query_1.AssocEdgeQueryBase {
 exports.UserToHostedEventsQuery = UserToHostedEventsQuery;
 const getNextWeekClause = () => {
     // get events starting within the next week
-    jest_date_mock_1.clear();
+    (0, jest_date_mock_1.clear)();
     const start = mock_date_1.MockDate.getDate();
     // 7 days
     const end = luxon_1.Interval.after(start, 86400 * 1000 * 7)
@@ -225,18 +254,18 @@ const getNextWeekClause = () => {
 };
 exports.getNextWeekClause = getNextWeekClause;
 function getCompleteClause(id) {
-    return clause.And(clause.Eq("user_id", id), exports.getNextWeekClause());
+    return clause.And(clause.Eq("user_id", id), (0, exports.getNextWeekClause)());
 }
 exports.getCompleteClause = getCompleteClause;
 exports.userToEventsInNextWeekCountLoaderFactory = new raw_count_loader_1.RawCountLoaderFactory({
     ...internal_1.FakeEvent.loaderOptions(),
     groupCol: "user_id",
-    clause: exports.getNextWeekClause(),
+    clause: (0, exports.getNextWeekClause)(),
 });
 exports.userToEventsInNextWeekDataLoaderFactory = new query_loader_1.QueryLoaderFactory({
     ...internal_1.FakeEvent.loaderOptions(),
     groupCol: "user_id",
-    clause: exports.getNextWeekClause(),
+    clause: (0, exports.getNextWeekClause)(),
     toPrime: [fake_contact_1.contactLoader],
     sortColumn: "start_time",
 });
@@ -255,5 +284,23 @@ class UserToEventsInNextWeekQuery extends custom_query_1.CustomEdgeQueryBase {
     static query(viewer, src) {
         return new UserToEventsInNextWeekQuery(viewer, src);
     }
+    sourceEnt(id) {
+        return internal_1.FakeUser.load(this.viewer, id);
+    }
+    getPrivacyPolicy() {
+        return privacy_1.AllowIfViewerPrivacyPolicy;
+    }
 }
 exports.UserToEventsInNextWeekQuery = UserToEventsInNextWeekQuery;
+class UserToFollowingQuery extends assoc_query_1.AssocEdgeQueryBase {
+    constructor(viewer, src) {
+        super(viewer, src, new assoc_count_loader_1.AssocEdgeCountLoaderFactory(internal_1.EdgeType.UserToFollowing), new assoc_edge_loader_1.AssocEdgeLoaderFactory(internal_1.EdgeType.UserToFollowing, ent_1.AssocEdge), _1.getLoaderOptions);
+    }
+    static query(viewer, src) {
+        return new UserToFollowingQuery(viewer, src);
+    }
+    sourceEnt(id) {
+        return internal_1.FakeUser.load(this.viewer, id);
+    }
+}
+exports.UserToFollowingQuery = UserToFollowingQuery;

package/testutils/parse_sql.js

@@ -74,7 +74,7 @@ function getColumns(cols) {
                 count = true;
             }
             else {
-                fail("unsupported expr type");
+                throw new Error("unsupported expr type");
             }
         }
     }
@@ -121,6 +121,22 @@ function getDataToReturn(data, colNames, returningAll) {
     return ret;
 }
 exports.getDataToReturn = getDataToReturn;
+function processBeforeStoring(val) {
+    if (typeof val !== "string") {
+        return val;
+    }
+    // convert postgres lists into lists before storing
+    if (val[0] === "{" && val[val.length - 1] === "}") {
+        try {
+            // valid json, don't convert
+            JSON.parse(val);
+        }
+        catch (e) {
+            return val.substring(1, val.length - 1).split(",");
+        }
+    }
+    return val;
+}
 function parseInsertStatement(ast, values, // values passed to query
 returningAll) {
     const tableName = getTableName(ast.table);
@@ -138,7 +154,7 @@ returningAll) {
     // INSERT INTO tableName (cols) VALUES (pos args)
     for (let i = 0; i < columns.length; i++) {
         let col = columns[i];
-        data[col] = values[i];
+        data[col] = processBeforeStoring(values[i]);
     }
     let returningData = null;
     if (returningAll) {
@@ -411,7 +427,7 @@ function parseUpdateStatement(ast, values, map, returningAll) {
     for (const set of ast.set) {
         let col = set.column;
         let value = getValueFromRegex(set.value, values);
-        overwrite[col] = value;
+        overwrite[col] = processBeforeStoring(value);
     }
     let columns = new Set();
     if (ast.returning) {

package/testutils/write.js

@@ -13,9 +13,9 @@ async function createRowForTest(options, suffix) {
     const client = await db_1.default.getInstance().getNewClient();
     try {
         if (isSyncClient(client)) {
-            return ent_1.createRowSync(client, options, suffix || "");
+            return (0, ent_1.createRowSync)(client, options, suffix || "");
         }
-        return await ent_1.createRow(client, options, suffix || "");
+        return await (0, ent_1.createRow)(client, options, suffix || "");
     }
     finally {
         client.release();
@@ -26,9 +26,9 @@ async function editRowForTest(options, id, suffix) {
     const client = await db_1.default.getInstance().getNewClient();
     try {
         if (isSyncClient(client)) {
-            return ent_1.editRowSync(client, options, id, suffix || "");
+            return (0, ent_1.editRowSync)(client, options, id, suffix || "");
         }
-        return await ent_1.editRow(client, options, id, suffix);
+        return await (0, ent_1.editRow)(client, options, id, suffix);
     }
     finally {
         client.release();
@@ -39,9 +39,9 @@ async function deleteRowsForTest(options, cls) {
     const client = await db_1.default.getInstance().getNewClient();
     try {
         if (isSyncClient(client)) {
-            return ent_1.deleteRowsSync(client, options, cls);
+            return (0, ent_1.deleteRowsSync)(client, options, cls);
         }
-        return await ent_1.deleteRows(client, options, cls);
+        return await (0, ent_1.deleteRows)(client, options, cls);
     }
     finally {
         client.release();