@snowtop/ent 0.1.0-alpha16 → 0.1.0-alpha160-test2

package/testutils/fake_data/test_helpers.js

@@ -6,19 +6,23 @@ const viewer_1 = require("../../core/viewer");
 const ent_1 = require("../../core/ent");
 const snake_case_1 = require("snake-case");
 const write_1 = require("../write");
-const test_db_1 = require("../db/test_db");
+const temp_db_1 = require("../db/temp_db");
 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");
+const fake_tag_1 = require("./fake_tag");
+const db_1 = require("../../core/db");
 function getContactInput(user, input) {
     return {
         firstName: "Jon",
         lastName: "Snow",
         emailAddress: "foo@bar.com",
         userID: user.id,
+        createdAt: new Date(),
+        updatedAt: new Date(),
         ...input,
     };
 }
@@ -30,6 +34,8 @@ function getUserInput(input) {
         emailAddress: "foo@bar.com",
         phoneNumber: "415-212-1212",
         password: "pa$$w0rd",
+        createdAt: new Date(),
+        updatedAt: new Date(),
         ...input,
     };
 }
@@ -41,12 +47,14 @@ function getEventInput(user, input) {
         title: "title",
         description: "fun event",
         userID: user.id,
+        createdAt: new Date(),
+        updatedAt: new Date(),
         ...input,
     };
 }
 exports.getEventInput = getEventInput;
-async function createTestUser(input) {
-    const user = await (0, _1.createUser)(new viewer_1.LoggedOutViewer(), {
+async function createTestUser(input, ctx) {
+    const user = await (0, _1.createUser)(new viewer_1.LoggedOutViewer(ctx), {
         firstName: "Jon",
         lastName: "Snow",
         password: "12345678",
@@ -82,22 +90,29 @@ exports.inputs = [
         lastName: "Stark",
     },
 ];
-async function createAllContacts(input, slice) {
-    const user = await createTestUser(input);
+async function createAllContacts(opts) {
+    let { input, slice, user } = opts || {};
+    if (!user) {
+        user = await createTestUser(input, opts?.ctx);
+    }
+    const userr = user;
+    if (opts?.start) {
+        (0, jest_date_mock_1.advanceTo)(opts.start);
+    }
     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
         (0, jest_date_mock_1.advanceBy)(100);
-        const builder = (0, _1.getContactBuilder)(user.viewer, getContactInput(user, input));
+        const builder = (0, _1.getContactBuilder)(userr.viewer, getContactInput(userr, input));
         // add edge from user to contact
-        builder.orchestrator.addInboundEdge(user.id, _1.EdgeType.UserToContacts, const_1.NodeType.FakeUser, {
+        builder.orchestrator.addInboundEdge(userr.id, _1.EdgeType.UserToContacts, const_1.NodeType.FakeUser, {
             time: new Date(), // set time to advanceBy time
         });
         await builder.saveX();
         return await builder.editedEntX();
     }));
     expect(contacts.length).toBe(userInputs.length);
-    return [user, contacts];
+    return [userr, contacts];
 }
 exports.createAllContacts = createAllContacts;
 async function createUserPlusFriendRequests(input, slice) {
@@ -208,22 +223,23 @@ async function createTestEvent(user, input) {
     return await builder.editedEntX();
 }
 exports.createTestEvent = createTestEvent;
-async function setupTempDB() {
-    const tdb = new test_db_1.TempDB(tempDBTables());
+async function setupTempDB(global = false) {
+    const tdb = new temp_db_1.TempDB(db_1.Dialect.Postgres, tempDBTables(global));
     await tdb.beforeAll();
     // create once
     await createEdges();
     return tdb;
 }
 exports.setupTempDB = setupTempDB;
-function tempDBTables() {
+function tempDBTables(global = false) {
     const tables = [
         _1.FakeUser.getTestTable(),
         _1.FakeContact.getTestTable(),
         fake_event_1.FakeEvent.getTestTable(),
-        (0, test_db_1.assoc_edge_config_table)(),
+        fake_tag_1.FakeTag.getTestTable(),
+        (0, temp_db_1.assoc_edge_config_table)(),
     ];
-    edgeTableNames().forEach((tableName) => tables.push((0, test_db_1.assoc_edge_table)(tableName)));
+    edgeTableNames().forEach((tableName) => tables.push((0, temp_db_1.assoc_edge_table)(tableName, global)));
     return tables;
 }
 exports.tempDBTables = tempDBTables;
@@ -232,7 +248,7 @@ 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
-    (0, jest_date_mock_1.advanceTo)(mock_date_1.MockDate.getDate());
+    (0, jest_date_mock_1.advanceTo)(opts.startTime || 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) {
@@ -241,7 +257,7 @@ async function createAllEvents(opts) {
         const input = opts.eventInputs?.[idx];
         const builder = (0, fake_event_1.getEventBuilder)(user.viewer, getEventInput(user, input));
         await builder.saveX();
-        return await builder.editedEntX();
+        return builder.editedEntX();
     }));
     expect(events.length).toBe(opts.howMany);
     return [user, events];

package/testutils/fake_data/user_query.d.ts

@@ -1,11 +1,10 @@
-import { Ent, ID, Viewer } from "../../core/base";
+import { Data, Ent, ID, Viewer } from "../../core/base";
 import { CustomEdgeQueryBase } from "../../core/query/custom_query";
 import { AssocEdge } from "../../core/ent";
 import * as clause from "../../core/clause";
 import { AssocEdgeQueryBase, EdgeQuerySource } from "../../core/query/assoc_query";
 import { FakeUser, FakeEvent, FakeContact, EventToAttendeesQuery, EventToDeclinedQuery, EventToHostsQuery, EventToInvitedQuery, EventToMaybeQuery } from "./internal";
 import { RawCountLoaderFactory } from "../../core/loaders/raw_count_loader";
-import { IndexLoaderFactory } from "../../core/loaders/index_loader";
 import { QueryLoaderFactory } from "../../core/loaders/query_loader";
 export declare class UserToContactsQuery extends AssocEdgeQueryBase<FakeUser, FakeContact, AssocEdge> {
     constructor(viewer: Viewer, src: EdgeQuerySource<FakeUser>);
@@ -13,12 +12,22 @@ export declare class UserToContactsQuery extends AssocEdgeQueryBase<FakeUser, Fa
     sourceEnt(id: ID): Promise<FakeUser | null>;
 }
 export declare const userToContactsCountLoaderFactory: RawCountLoaderFactory;
-export declare const userToContactsDataLoaderFactory: IndexLoaderFactory;
+export declare const userToContactsDataLoaderFactory: QueryLoaderFactory<unknown>;
+export declare class UserToContactsFkeyQueryDeprecated extends CustomEdgeQueryBase<FakeUser, FakeContact> {
+    constructor(viewer: Viewer, src: ID | FakeUser);
+    static query(viewer: Viewer, src: FakeUser | ID): UserToContactsFkeyQueryDeprecated;
+    sourceEnt(id: ID): Promise<FakeUser | null>;
+}
 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 UserToContactsFkeyQueryAsc 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;
@@ -30,6 +39,8 @@ export declare class UserToFriendsQuery extends AssocEdgeQueryBase<FakeUser, Fak
     queryCustomEdge(): UserToCustomEdgeQuery;
 }
 export declare class CustomEdge extends AssocEdge {
+    deleted_at: Date | null;
+    constructor(data: Data);
     loadUser(viewer: Viewer): Promise<FakeUser | null>;
 }
 export declare class UserToCustomEdgeQuery extends AssocEdgeQueryBase<FakeUser, FakeUser, CustomEdge> {
@@ -51,7 +62,7 @@ export declare class UserToFriendRequestsQuery extends AssocEdgeQueryBase<FakeUs
     queryEventsAttending(): UserToEventsAttendingQuery;
     queryCustomEdge(): UserToCustomEdgeQuery;
 }
-export declare class UserToIncomingFriendRequestsQuery extends AssocEdgeQueryBase<FakeUser, FakeUser, AssocEdge> {
+export declare class UserToIncomingFriendRequestsQuery extends AssocEdgeQueryBase<FakeUser, FakeUser, CustomEdge> {
     constructor(viewer: Viewer, src: EdgeQuerySource<FakeUser, FakeUser>);
     getPrivacyPolicy(): import("../../core/base").PrivacyPolicy<Ent<Viewer<Ent<any> | null, ID | null>>, Viewer<Ent<any> | null, ID | null>>;
     sourceEnt(id: ID): Promise<FakeUser | null>;
@@ -61,6 +72,7 @@ export declare class UserToIncomingFriendRequestsQuery extends AssocEdgeQueryBas
     queryHostedEvents(): UserToHostedEventsQuery;
     queryEventsAttending(): UserToEventsAttendingQuery;
     queryCustomEdge(): UserToCustomEdgeQuery;
+    withoutTransformations(): this;
 }
 export declare class UserToEventsAttendingQuery extends AssocEdgeQueryBase<FakeUser, FakeEvent, AssocEdge> {
     constructor(viewer: Viewer, src: EdgeQuerySource<FakeUser, FakeEvent>);
@@ -84,8 +96,6 @@ export declare class UserToHostedEventsQuery extends AssocEdgeQueryBase<FakeUser
 }
 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<FakeUser, FakeEvent> {
     constructor(viewer: Viewer, src: ID | FakeUser);
     static query(viewer: Viewer, src: FakeUser | ID): UserToEventsInNextWeekQuery;

package/testutils/fake_data/user_query.js

@@ -1,7 +1,11 @@
 "use strict";
 var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
     if (k2 === undefined) k2 = k;
-    Object.defineProperty(o, k2, { enumerable: true, get: function() { return m[k]; } });
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
 }) : (function(o, m, k, k2) {
     if (k2 === undefined) k2 = k;
     o[k2] = m[k];
@@ -19,7 +23,7 @@ var __importStar = (this && this.__importStar) || function (mod) {
     return result;
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-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;
+exports.UserToFollowingQuery = exports.UserToEventsInNextWeekQuery = exports.getCompleteClause = exports.getNextWeekClause = exports.UserToHostedEventsQuery = exports.UserToEventsAttendingQuery = exports.UserToIncomingFriendRequestsQuery = exports.UserToFriendRequestsQuery = exports.UserToCustomEdgeQuery = exports.CustomEdge = exports.UserToFriendsQuery = exports.UserToContactsFkeyQueryAsc = exports.UserToContactsFkeyQuery = exports.UserToContactsFkeyQueryDeprecated = 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"));
@@ -28,7 +32,6 @@ const internal_1 = require("./internal");
 const raw_count_loader_1 = require("../../core/loaders/raw_count_loader");
 const assoc_count_loader_1 = require("../../core/loaders/assoc_count_loader");
 const assoc_edge_loader_1 = require("../../core/loaders/assoc_edge_loader");
-const index_loader_1 = require("../../core/loaders/index_loader");
 const fake_contact_1 = require("./fake_contact");
 const jest_date_mock_1 = require("jest-date-mock");
 const luxon_1 = require("luxon");
@@ -52,10 +55,13 @@ exports.userToContactsCountLoaderFactory = new raw_count_loader_1.RawCountLoader
     ...internal_1.FakeContact.loaderOptions(),
     groupCol: "user_id",
 });
-exports.userToContactsDataLoaderFactory = new index_loader_1.IndexLoaderFactory(internal_1.FakeContact.loaderOptions(), "user_id", {
+exports.userToContactsDataLoaderFactory = new query_loader_1.QueryLoaderFactory({
+    ...internal_1.FakeContact.loaderOptions(),
+    groupCol: "user_id",
     toPrime: [fake_contact_1.contactLoader],
 });
-class UserToContactsFkeyQuery extends custom_query_1.CustomEdgeQueryBase {
+// note: this is still used in graphql tests...
+class UserToContactsFkeyQueryDeprecated extends custom_query_1.CustomEdgeQueryBase {
     constructor(viewer, src) {
         super(viewer, {
             src,
@@ -63,6 +69,27 @@ class UserToContactsFkeyQuery extends custom_query_1.CustomEdgeQueryBase {
             countLoaderFactory: exports.userToContactsCountLoaderFactory,
             dataLoaderFactory: exports.userToContactsDataLoaderFactory,
             options: internal_1.FakeContact.loaderOptions(),
+            sortColumn: "created_at",
+        });
+    }
+    static query(viewer, src) {
+        return new UserToContactsFkeyQueryDeprecated(viewer, src);
+    }
+    sourceEnt(id) {
+        return internal_1.FakeUser.load(this.viewer, id);
+    }
+}
+exports.UserToContactsFkeyQueryDeprecated = UserToContactsFkeyQueryDeprecated;
+// this replaces UserToContactsFkeyQueryDeprecated
+class UserToContactsFkeyQuery extends custom_query_1.CustomEdgeQueryBase {
+    constructor(viewer, src) {
+        super(viewer, {
+            src,
+            loadEntOptions: internal_1.FakeContact.loaderOptions(),
+            groupCol: "user_id",
+            name: "user_to_contacts",
+            // instead of the id col...
+            sortColumn: "created_at",
         });
     }
     static query(viewer, src) {
@@ -73,6 +100,29 @@ class UserToContactsFkeyQuery extends custom_query_1.CustomEdgeQueryBase {
     }
 }
 exports.UserToContactsFkeyQuery = UserToContactsFkeyQuery;
+class UserToContactsFkeyQueryAsc extends custom_query_1.CustomEdgeQueryBase {
+    constructor(viewer, src) {
+        super(viewer, {
+            src,
+            loadEntOptions: internal_1.FakeContact.loaderOptions(),
+            groupCol: "user_id",
+            name: "user_to_contacts",
+            orderby: [
+                {
+                    column: "created_at",
+                    direction: "ASC",
+                },
+            ],
+        });
+    }
+    static query(viewer, src) {
+        return new UserToContactsFkeyQueryAsc(viewer, src);
+    }
+    sourceEnt(id) {
+        return internal_1.FakeUser.load(this.viewer, id);
+    }
+}
+exports.UserToContactsFkeyQueryAsc = UserToContactsFkeyQueryAsc;
 class UserToFriendsQuery extends assoc_query_1.AssocEdgeQueryBase {
     constructor(viewer, src) {
         super(viewer, src, new assoc_count_loader_1.AssocEdgeCountLoaderFactory(internal_1.EdgeType.UserToFriends), new assoc_edge_loader_1.AssocEdgeLoaderFactory(internal_1.EdgeType.UserToFriends, ent_1.AssocEdge), internal_1.FakeUser.loaderOptions());
@@ -102,6 +152,11 @@ class UserToFriendsQuery extends assoc_query_1.AssocEdgeQueryBase {
 exports.UserToFriendsQuery = UserToFriendsQuery;
 // example with custom method
 class CustomEdge extends ent_1.AssocEdge {
+    constructor(data) {
+        super(data);
+        this.deleted_at = null;
+        this.deleted_at = data.deleted_at;
+    }
     async loadUser(viewer) {
         return await internal_1.FakeUser.load(viewer, this.id2);
     }
@@ -160,7 +215,7 @@ class UserToFriendRequestsQuery extends assoc_query_1.AssocEdgeQueryBase {
 exports.UserToFriendRequestsQuery = UserToFriendRequestsQuery;
 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());
+        super(viewer, src, new assoc_count_loader_1.AssocEdgeCountLoaderFactory(internal_1.EdgeType.UserToIncomingFriendRequests), new assoc_edge_loader_1.AssocEdgeLoaderFactory(internal_1.EdgeType.UserToIncomingFriendRequests, CustomEdge), internal_1.FakeUser.loaderOptions());
     }
     getPrivacyPolicy() {
         return privacy_1.AllowIfViewerPrivacyPolicy;
@@ -186,6 +241,13 @@ class UserToIncomingFriendRequestsQuery extends assoc_query_1.AssocEdgeQueryBase
     queryCustomEdge() {
         return UserToCustomEdgeQuery.query(this.viewer, this);
     }
+    // this is generated in codegen. we'll just add it manually here
+    withoutTransformations() {
+        this.configureEdgeQueryableDataOptions({
+            disableTransformations: true,
+        });
+        return this;
+    }
 }
 exports.UserToIncomingFriendRequestsQuery = UserToIncomingFriendRequestsQuery;
 class UserToEventsAttendingQuery extends assoc_query_1.AssocEdgeQueryBase {
@@ -257,27 +319,14 @@ function getCompleteClause(id) {
     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: (0, exports.getNextWeekClause)(),
-});
-exports.userToEventsInNextWeekDataLoaderFactory = new query_loader_1.QueryLoaderFactory({
-    ...internal_1.FakeEvent.loaderOptions(),
-    groupCol: "user_id",
-    clause: (0, exports.getNextWeekClause)(),
-    toPrime: [fake_contact_1.contactLoader],
-    sortColumn: "start_time",
-});
 class UserToEventsInNextWeekQuery extends custom_query_1.CustomEdgeQueryBase {
     constructor(viewer, src) {
         super(viewer, {
             src,
-            // we want to reuse this and not create a new one every time...
-            countLoaderFactory: exports.userToEventsInNextWeekCountLoaderFactory,
-            dataLoaderFactory: exports.userToEventsInNextWeekDataLoaderFactory,
-            options: internal_1.FakeEvent.loaderOptions(),
-            // hmm TODO shouldn't need to write this twice...
+            groupCol: "user_id",
+            clause: (0, exports.getNextWeekClause)(),
+            loadEntOptions: internal_1.FakeEvent.loaderOptions(),
+            name: "events_in_next_week",
             sortColumn: "start_time",
         });
     }

package/testutils/fake_log.js

@@ -21,12 +21,12 @@ class FakeLogger {
         this.logs = [];
     }
 }
-exports.FakeLogger = FakeLogger;
 // todo this is quick and ideal.
 // more ideal is capturing all console.logs
 // we do it in golang for example
 // and will need this for production launch anyways
 FakeLogger.logs = [];
+exports.FakeLogger = FakeLogger;
 // TODO instead of needing to add this manually
 // we need to build a way to add global observers (and maybe triggers)
 // to be run on every action/mutation

package/testutils/parse_sql.d.ts

@@ -9,6 +9,12 @@ export declare class EqOp {
     constructor(col: string, value: any);
     apply(data: Data): boolean;
 }
+export declare class NotEqOp {
+    private col;
+    private value;
+    constructor(col: string, value: any);
+    apply(data: Data): boolean;
+}
 export declare class GreaterOp {
     private col;
     private value;

package/testutils/parse_sql.js

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.performQuery = exports.OrOp = exports.AndOp = exports.InOp = exports.LessEqOp = exports.GreaterEqOp = exports.LessOp = exports.GreaterOp = exports.EqOp = exports.getDataToReturn = void 0;
+exports.performQuery = exports.OrOp = exports.AndOp = exports.InOp = exports.LessEqOp = exports.GreaterEqOp = exports.LessOp = exports.GreaterOp = exports.NotEqOp = exports.EqOp = exports.getDataToReturn = void 0;
 const node_sql_parser_1 = require("node-sql-parser");
 function getTableName(table) {
     if (Array.isArray(table)) {
@@ -184,6 +184,16 @@ class EqOp {
     }
 }
 exports.EqOp = EqOp;
+class NotEqOp {
+    constructor(col, value) {
+        this.col = col;
+        this.value = value;
+    }
+    apply(data) {
+        return data[this.col] !== this.value;
+    }
+}
+exports.NotEqOp = NotEqOp;
 class GreaterOp {
     constructor(col, value) {
         this.col = col;
@@ -307,8 +317,12 @@ function getOp(where, values) {
             return new AndOp([getOp(where.left, values), getOp(where.right, values)]);
         case "OR":
             return new OrOp([getOp(where.left, values), getOp(where.right, values)]);
+        case "IS":
+            return new EqOp(getColumnFromRef(where.left), where.right?.value);
+        case "IS NOT":
+            return new NotEqOp(getColumnFromRef(where.left), where.right?.value);
         default:
-            console.log(where);
+            console.debug(where);
             throw new Error(`unsupported op ${where.operator}`);
     }
 }

package/testutils/test_edge_global_schema.d.ts

@@ -0,0 +1,15 @@
+import { EdgeUpdateOperation, TransformedEdgeUpdateOperation } from "../schema";
+import * as clause from "../core/clause";
+import { AssocEdge } from "../core/ent";
+import { Data } from "../core/base";
+export declare class EdgeWithDeletedAt extends AssocEdge {
+    deletedAt: Date | null;
+    constructor(data: Data);
+}
+export declare const testEdgeGlobalSchema: {
+    extraEdgeFields: {
+        deletedAt: import("../schema").TimestampField;
+    };
+    transformEdgeRead(): clause.Clause;
+    transformEdgeWrite(stmt: EdgeUpdateOperation): TransformedEdgeUpdateOperation | null;
+};

package/testutils/test_edge_global_schema.js

@@ -0,0 +1,62 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || function (mod) {
+    if (mod && mod.__esModule) return mod;
+    var result = {};
+    if (mod != null) for (var k in mod) if (k !== "default" && Object.prototype.hasOwnProperty.call(mod, k)) __createBinding(result, mod, k);
+    __setModuleDefault(result, mod);
+    return result;
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.testEdgeGlobalSchema = exports.EdgeWithDeletedAt = void 0;
+const schema_1 = require("../schema");
+const clause = __importStar(require("../core/clause"));
+const ent_1 = require("../core/ent");
+class EdgeWithDeletedAt extends ent_1.AssocEdge {
+    constructor(data) {
+        super(data);
+        this.deletedAt = data.deleted_at;
+    }
+}
+exports.EdgeWithDeletedAt = EdgeWithDeletedAt;
+exports.testEdgeGlobalSchema = {
+    extraEdgeFields: {
+        // need this to be lowerCamelCase because we do this based on field name
+        // #510
+        deletedAt: (0, schema_1.TimestampType)({
+            nullable: true,
+            index: true,
+            defaultValueOnCreate: () => null,
+        }),
+    },
+    transformEdgeRead() {
+        return clause.Eq("deleted_at", null);
+    },
+    transformEdgeWrite(stmt) {
+        switch (stmt.op) {
+            case schema_1.SQLStatementOperation.Delete:
+                return {
+                    op: schema_1.SQLStatementOperation.Update,
+                    data: {
+                        deleted_at: new Date(),
+                    },
+                };
+        }
+        return null;
+    },
+};

package/testutils/write.d.ts

@@ -1,5 +1,5 @@
-import { EditRowOptions, Data, ID, DataOptions, CreateRowOptions } from "../core/base";
+import { EditRowOptions, Data, DataOptions, CreateRowOptions } from "../core/base";
 import * as clause from "../core/clause";
 export declare function createRowForTest(options: CreateRowOptions, suffix?: string): Promise<Data | null>;
-export declare function editRowForTest(options: EditRowOptions, id: ID, suffix?: string): Promise<Data | null>;
+export declare function editRowForTest(options: EditRowOptions, suffix?: string): Promise<Data | null>;
 export declare function deleteRowsForTest(options: DataOptions, cls: clause.Clause): Promise<void>;

package/testutils/write.js

@@ -1,16 +1,42 @@
 "use strict";
-var __importDefault = (this && this.__importDefault) || function (mod) {
-    return (mod && mod.__esModule) ? mod : { "default": mod };
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || function (mod) {
+    if (mod && mod.__esModule) return mod;
+    var result = {};
+    if (mod != null) for (var k in mod) if (k !== "default" && Object.prototype.hasOwnProperty.call(mod, k)) __createBinding(result, mod, k);
+    __setModuleDefault(result, mod);
+    return result;
 };
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.deleteRowsForTest = exports.editRowForTest = exports.createRowForTest = void 0;
 const ent_1 = require("../core/ent");
-const db_1 = __importDefault(require("../core/db"));
+const db_1 = __importStar(require("../core/db"));
 function isSyncClient(client) {
     return client.execSync !== undefined;
 }
 async function createRowForTest(options, suffix) {
-    const client = await db_1.default.getInstance().getNewClient();
+    let client;
+    if (db_1.Dialect.SQLite === db_1.default.getDialect()) {
+        client = db_1.default.getInstance().getSQLiteClient();
+    }
+    else {
+        client = await db_1.default.getInstance().getNewClient();
+    }
     try {
         if (isSyncClient(client)) {
             return (0, ent_1.createRowSync)(client, options, suffix || "");
@@ -22,13 +48,13 @@ async function createRowForTest(options, suffix) {
     }
 }
 exports.createRowForTest = createRowForTest;
-async function editRowForTest(options, id, suffix) {
+async function editRowForTest(options, suffix) {
     const client = await db_1.default.getInstance().getNewClient();
     try {
         if (isSyncClient(client)) {
-            return (0, ent_1.editRowSync)(client, options, id, suffix || "");
+            return (0, ent_1.editRowSync)(client, options, suffix || "");
         }
-        return await (0, ent_1.editRow)(client, options, id, suffix);
+        return await (0, ent_1.editRow)(client, options, suffix);
     }
     finally {
         client.release();

package/tsc/ast.d.ts

@@ -10,23 +10,35 @@ export interface ClassInfo {
     wrapClassContents(inner: string): string;
 }
 export declare function getClassInfo(fileContents: string, sourceFile: ts.SourceFile, node: ts.ClassDeclaration): ClassInfo | undefined;
-declare type transformImportFn = (imp: string) => string;
+type transformImportFn = (imp: string) => string;
 interface transformOpts {
     removeImports?: string[];
     newImports?: string[];
     transform?: transformImportFn;
-    transformPath?: string;
+    transformPath: string;
 }
-export declare function transformImport(fileContents: string, importNode: ts.ImportDeclaration, sourceFile: ts.SourceFile, opts?: transformOpts): string | undefined;
+export declare function transformImport(fileContents: string, importNode: ts.ImportDeclaration, sourceFile: ts.SourceFile, opts: transformOpts): string | undefined;
 export declare function updateImportPath(fileContents: string, importNode: ts.ImportDeclaration, sourceFile: ts.SourceFile, newPath: string): string | undefined;
 export declare function isRelativeImport(node: ts.ImportDeclaration, sourceFile: ts.SourceFile): boolean;
 export declare function isRelativeGeneratedImport(node: ts.ImportDeclaration, sourceFile: ts.SourceFile): boolean;
+export declare function isSrcGeneratedImport(node: ts.ImportDeclaration, sourceFile: ts.SourceFile): boolean;
 interface importInfo {
     imports: string[];
     start: number;
     end: number;
     importText: string;
     importPath: string;
+    default?: string;
 }
 export declare function getImportInfo(imp: ts.ImportDeclaration, sourceFile: ts.SourceFile): importInfo | undefined;
+export declare function transformRelative(file: string, importPath: string, relative?: boolean): string;
+export interface customInfo {
+    viewerInfo: {
+        path: string;
+        name: string;
+    };
+    relativeImports?: boolean;
+    globalSchemaPath?: string;
+}
+export declare function getCustomInfo(): customInfo;
 export {};