@snowtop/ent 0.1.0-alpha9 → 0.1.0-alpha90

package/testutils/db/value.js

@@ -0,0 +1,251 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.getDefaultValue = void 0;
+const uuid_1 = require("uuid");
+const schema_1 = require("../../schema");
+const schema_2 = require("../../schema");
+function random() {
+    return Math.random().toString(16).substring(2);
+}
+function randomEmail(domain) {
+    domain = domain || "email.com";
+    return `test+${random()}@${domain}`;
+}
+function randomPhoneNumber() {
+    return `+1${Math.random().toString(10).substring(2, 11)}`;
+}
+function coinFlip() {
+    return Math.floor(Math.random() * 10) >= 5;
+}
+function specialType(typ, col) {
+    let list = m.get(typ.dbType);
+    if (list?.length) {
+        for (const l of list) {
+            let regex = [];
+            if (Array.isArray(l.regex)) {
+                regex = l.regex;
+            }
+            else {
+                regex = [l.regex];
+            }
+            for (const r of regex) {
+                if (r.test(col)) {
+                    return l.newValue();
+                }
+            }
+        }
+    }
+    return undefined;
+}
+function getDefaultValue(f, col, infos) {
+    if (f.defaultValueOnCreate) {
+        // @ts-ignore
+        return f.defaultValueOnCreate();
+    }
+    // half the time, return null for nullable
+    if (f.nullable && coinFlip()) {
+        return null;
+    }
+    const specialVal = specialType(f.type, col);
+    if (specialVal !== undefined) {
+        return specialVal;
+    }
+    return getValueForType(f.type, f, infos);
+}
+exports.getDefaultValue = getDefaultValue;
+function getValueForType(typ, f, infos) {
+    switch (typ.dbType) {
+        case schema_1.DBType.UUID:
+            return (0, uuid_1.v4)();
+        case schema_1.DBType.Boolean:
+            return coinFlip();
+        case schema_1.DBType.Date:
+            return (0, schema_2.DateType)().format(new Date());
+        case schema_1.DBType.Time:
+            return (0, schema_2.TimeType)().format(new Date());
+        case schema_1.DBType.Timetz:
+            return (0, schema_2.TimetzType)().format(new Date());
+        case schema_1.DBType.Timestamp:
+            return (0, schema_2.TimestampType)().format(new Date());
+        case schema_1.DBType.Timestamptz:
+            return (0, schema_2.TimestamptzType)().format(new Date());
+        case schema_1.DBType.String:
+            return random();
+        case schema_1.DBType.Int:
+            return Math.floor(Math.random() * 100000000);
+        case schema_1.DBType.Float:
+            return Math.random() * 100000000;
+        case schema_1.DBType.Enum:
+        case schema_1.DBType.StringEnum:
+            if (typ.values) {
+                const idx = Math.floor(Math.random() * typ.values.length);
+                return typ.values[idx];
+            }
+            if (typ.enumMap) {
+                const vals = Object.values(typ.enumMap);
+                const idx = Math.floor(Math.random() * vals.length);
+                return vals[idx];
+            }
+            if (f.foreignKey) {
+                const schema = f.foreignKey.schema;
+                const col = f.foreignKey.column;
+                if (!infos) {
+                    throw new Error(`infos required for enum with foreignKey`);
+                }
+                const info = infos.get(schema);
+                if (!info) {
+                    throw new Error(`couldn't load data for schema ${schema}`);
+                }
+                if (!info.schema.dbRows) {
+                    throw new Error(`no dbRows for schema ${schema}`);
+                }
+                const idx = Math.floor(Math.random() * info.schema.dbRows.length);
+                return info.schema.dbRows[idx][col];
+            }
+            throw new Error("TODO: enum without values not currently supported");
+        case schema_1.DBType.IntEnum:
+            const vals = Object.values(typ.intEnumMap);
+            const idx = Math.floor(Math.random() * vals.length);
+            return vals[idx];
+        case schema_1.DBType.BigInt:
+            return BigInt(Math.floor(Math.random() * 100000000));
+        case schema_1.DBType.JSONB:
+            // type as list
+            if (typ.listElemType?.dbType === schema_1.DBType.JSONB) {
+                const values = [];
+                for (let i = 0; i < 10; i++) {
+                    values.push(getValueForType(typ.listElemType, f, infos));
+                }
+                if (!f.format) {
+                    throw new Error("invalid format");
+                }
+                return f.format(values);
+            }
+            return (0, schema_2.JSONBType)().format({});
+        case schema_1.DBType.JSON:
+            return (0, schema_2.JSONType)().format({});
+        case schema_1.DBType.List:
+            // just do 10
+            const values = [];
+            for (let i = 0; i < 10; i++) {
+                values.push(getValueForType(f.type.listElemType, f.__getElemField(), infos));
+            }
+            if (!f.format) {
+                throw new Error("invalid format");
+            }
+            return f.format(values);
+        default:
+            throw new Error(`unsupported type ${typ.dbType}`);
+    }
+}
+const emailType = {
+    dbType: schema_1.DBType.String,
+    newValue: () => {
+        return (0, schema_2.StringType)().format(randomEmail().toLowerCase());
+    },
+    regex: /^email(_address)|_email$/,
+};
+const pdt = (0, schema_2.StringType)();
+const phoneType = {
+    dbType: schema_1.DBType.String,
+    newValue: () => {
+        return randomPhoneNumber();
+    },
+    regex: /^phone(_number)?|_phone$|_phone_number$/,
+};
+const passwordType = {
+    dbType: schema_1.DBType.String,
+    newValue: () => {
+        // we don't use password type because when we're generating so many rows, it's too slow...
+        return random();
+    },
+    regex: /^password/,
+};
+const firstNames = [
+    "Daenerys",
+    "Jon",
+    "Arya",
+    "Sansa",
+    "Eddard",
+    "Khal",
+    "Robb",
+    "Joffrey",
+    "Ramsay",
+    "Cersei",
+    "Bolton",
+    "Oberyn",
+    "Jojen",
+    "Petyr",
+    "Brienne",
+    "Ygritte",
+    "Missandei",
+    "Shae",
+    "Sandor",
+    "Theon",
+    "Catelyn",
+    "Gilly",
+    "Samwell",
+    "Jaime",
+    "Stannis",
+    "Tyene",
+    "Obara",
+    "Nymeria",
+    "Elia",
+    "Ellaria",
+    "Myrcella",
+    "Hodor",
+    "Osha",
+    "Meera",
+    "Davos",
+    "Gendry",
+];
+const lastNames = [
+    "Stark",
+    "Targaryen",
+    "Lannister",
+    "Drogo",
+    "Baratheon",
+    "Reed",
+    "Martell",
+    "Tyrell",
+    "Clegane",
+    "Baelish",
+    "Greyjoy",
+    "Tarly",
+    "Sand",
+    "Snow",
+    "Bolton",
+    "Frey",
+    "Tarth",
+    "Payne",
+    "Seaworth",
+];
+const firstNameType = {
+    dbType: schema_1.DBType.String,
+    newValue: () => {
+        let idx = Math.floor(firstNames.length * Math.random());
+        return firstNames[idx];
+    },
+    regex: /^first_?(name)?/,
+};
+const lastNameType = {
+    dbType: schema_1.DBType.String,
+    newValue: () => {
+        let idx = Math.floor(lastNames.length * Math.random());
+        return lastNames[idx];
+    },
+    regex: /^last_?(name)?/,
+};
+let types = [
+    phoneType,
+    emailType,
+    passwordType,
+    firstNameType,
+    lastNameType,
+];
+let m = new Map();
+for (const type of types) {
+    let list = m.get(type.dbType) || [];
+    list.push(type);
+    m.set(type.dbType, list);
+}

package/testutils/db_time_zone.d.ts

@@ -0,0 +1,4 @@
+export declare class DBTimeZone {
+    private static getVal;
+    static getDateOffset(d: Date): Promise<string>;
+}

package/testutils/db_time_zone.js

@@ -0,0 +1,41 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.DBTimeZone = void 0;
+const luxon_1 = require("luxon");
+const schema_1 = require("../schema");
+const db_1 = __importDefault(require("../core/db"));
+let dbCurrentZone = undefined;
+class DBTimeZone {
+    static async getVal() {
+        if (dbCurrentZone !== undefined) {
+            return dbCurrentZone;
+        }
+        const r = await db_1.default.getInstance()
+            .getPool()
+            .query("SELECT current_setting('TIMEZONE');");
+        if (r.rows.length) {
+            dbCurrentZone = r.rows[0].current_setting;
+        }
+        else {
+            dbCurrentZone = null;
+        }
+        return dbCurrentZone;
+    }
+    static async getDateOffset(d) {
+        let zone = await DBTimeZone.getVal();
+        let dt = luxon_1.DateTime.fromJSDate(d);
+        if (zone) {
+            dt = dt.setZone(zone);
+        }
+        // use
+        const val = (0, schema_1.leftPad)(dt.get("offset") / 60);
+        if (val == "00") {
+            return "+00";
+        }
+        return val;
+    }
+}
+exports.DBTimeZone = DBTimeZone;

package/testutils/fake_data/fake_contact.d.ts

@@ -1,4 +1,4 @@
-import { ID, Ent, Viewer, Data, LoadEntOptions, PrivacyPolicy } from "../../core/base";
+import { ID, Ent, Viewer, Data, LoadEntOptions, PrivacyPolicy, Context } from "../../core/base";
 import { SimpleBuilder } from "../builder";
 import { NodeType } from "./const";
 import { ObjectLoaderFactory } from "../../core/loaders";
@@ -13,13 +13,14 @@ export declare class FakeContact implements Ent {
     readonly lastName: string;
     readonly emailAddress: string;
     readonly userID: ID;
-    privacyPolicy: PrivacyPolicy;
+    getPrivacyPolicy(): PrivacyPolicy<this>;
     constructor(viewer: Viewer, data: Data);
     static getFields(): string[];
-    static getTestTable(): import("../db/test_db").Table;
+    static getTestTable(): import("../db/temp_db").Table;
     static loaderOptions(): LoadEntOptions<FakeContact>;
     static load(v: Viewer, id: ID): Promise<FakeContact | null>;
     static loadX(v: Viewer, id: ID): Promise<FakeContact>;
+    static loadRawData(id: ID, context?: Context): Promise<Data | null>;
 }
 export declare const FakeContactSchema: import("../builder").BuilderSchema<FakeContact>;
 export interface ContactCreateInput {
@@ -28,6 +29,6 @@ export interface ContactCreateInput {
     emailAddress: string;
     userID: ID;
 }
-export declare function getContactBuilder(viewer: Viewer, input: ContactCreateInput): SimpleBuilder<FakeContact>;
+export declare function getContactBuilder(viewer: Viewer, input: ContactCreateInput): SimpleBuilder<FakeContact, null>;
 export declare function createContact(viewer: Viewer, input: ContactCreateInput): Promise<void>;
 export declare const contactLoader: ObjectLoaderFactory<unknown>;

package/testutils/fake_data/fake_contact.js

@@ -6,16 +6,14 @@ const privacy_1 = require("../../core/privacy");
 const builder_1 = require("../builder");
 const schema_1 = require("../../schema");
 const const_1 = require("./const");
-const test_db_1 = require("../db/test_db");
+const temp_db_1 = require("../db/temp_db");
 const loaders_1 = require("../../core/loaders");
 const convert_1 = require("../../core/convert");
+const action_1 = require("../../action");
 class FakeContact {
     constructor(viewer, data) {
         this.viewer = viewer;
         this.nodeType = const_1.NodeType.FakeContact;
-        this.privacyPolicy = {
-            rules: [new privacy_1.AllowIfViewerIsRule("userID"), privacy_1.AlwaysDenyRule],
-        };
         this.data = data;
         this.id = data.id;
         this.createdAt = (0, convert_1.convertDate)(data.created_at);
@@ -25,6 +23,11 @@ class FakeContact {
         this.emailAddress = data.email_address;
         this.userID = data.user_id;
     }
+    getPrivacyPolicy() {
+        return {
+            rules: [new privacy_1.AllowIfViewerIsRule("userID"), privacy_1.AlwaysDenyRule],
+        };
+    }
     static getFields() {
         return [
             "id",
@@ -37,7 +40,7 @@ class FakeContact {
         ];
     }
     static getTestTable() {
-        return (0, test_db_1.table)("fake_contacts", (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.uuid)("user_id"));
+        return (0, temp_db_1.table)("fake_contacts", (0, temp_db_1.uuid)("id", { primaryKey: true }), (0, temp_db_1.timestamptz)("created_at"), (0, temp_db_1.timestamptz)("updated_at"), (0, temp_db_1.text)("first_name"), (0, temp_db_1.text)("last_name"), (0, temp_db_1.text)("email_address"), (0, temp_db_1.uuid)("user_id"));
     }
     static loaderOptions() {
         return {
@@ -57,6 +60,11 @@ class FakeContact {
     static async loadX(v, id) {
         return (0, ent_1.loadEntX)(v, id, FakeContact.loaderOptions());
     }
+    static async loadRawData(id, context) {
+        return FakeContact.loaderOptions()
+            .loaderFactory.createLoader(context)
+            .load(id);
+    }
 }
 exports.FakeContact = FakeContact;
 exports.FakeContactSchema = (0, builder_1.getBuilderSchemaFromFields)({
@@ -75,7 +83,7 @@ function getContactBuilder(viewer, input) {
     //To lock in the value of Date now incase of advanceTo/advanceBy
     m.set("createdAt", new Date());
     m.set("updatedAt", new Date());
-    return new builder_1.SimpleBuilder(viewer, exports.FakeContactSchema, m);
+    return new builder_1.SimpleBuilder(viewer, exports.FakeContactSchema, m, action_1.WriteOperation.Insert, null);
 }
 exports.getContactBuilder = getContactBuilder;
 async function createContact(viewer, input) {

package/testutils/fake_data/fake_event.d.ts

@@ -14,16 +14,18 @@ export declare class FakeEvent implements Ent {
     readonly title: string;
     readonly description: string | null;
     readonly userID: ID;
-    privacyPolicy: PrivacyPolicy;
+    getPrivacyPolicy(): PrivacyPolicy<this>;
     constructor(viewer: Viewer, data: Data);
     private static getFields;
-    static getTestTable(): import("../db/test_db").Table;
+    static getTestTable(): import("../db/temp_db").Table;
     static loaderOptions(): LoadEntOptions<FakeEvent>;
     static load(v: Viewer, id: ID): Promise<FakeEvent | null>;
     static loadX(v: Viewer, id: ID): Promise<FakeEvent>;
 }
 export declare const FakeEventSchema: import("../builder").BuilderSchema<FakeEvent>;
 export interface EventCreateInput {
+    createdAt?: Date;
+    updatedAt?: Date;
     startTime: Date;
     endTime?: Date | null;
     location: string;
@@ -31,5 +33,5 @@ export interface EventCreateInput {
     description?: string | null;
     userID: ID;
 }
-export declare function getEventBuilder(viewer: Viewer, input: EventCreateInput): SimpleBuilder<FakeEvent>;
+export declare function getEventBuilder(viewer: Viewer, input: EventCreateInput): SimpleBuilder<FakeEvent, null>;
 export declare function createEvent(viewer: Viewer, input: EventCreateInput): Promise<void>;

package/testutils/fake_data/fake_event.js

@@ -6,14 +6,14 @@ const privacy_1 = require("../../core/privacy");
 const builder_1 = require("../builder");
 const schema_1 = require("../../schema");
 const const_1 = require("./const");
-const test_db_1 = require("../db/test_db");
+const temp_db_1 = require("../db/temp_db");
 const loaders_1 = require("../../core/loaders");
 const convert_1 = require("../../core/convert");
+const action_1 = require("../../action");
 class FakeEvent {
     constructor(viewer, data) {
         this.viewer = viewer;
         this.nodeType = const_1.NodeType.FakeEvent;
-        this.privacyPolicy = privacy_1.AlwaysAllowPrivacyPolicy;
         this.data = data;
         this.id = data.id;
         this.createdAt = (0, convert_1.convertDate)(data.created_at);
@@ -25,6 +25,9 @@ class FakeEvent {
         this.description = data.description;
         this.userID = data.user_id;
     }
+    getPrivacyPolicy() {
+        return privacy_1.AlwaysAllowPrivacyPolicy;
+    }
     static getFields() {
         return [
             "id",
@@ -39,9 +42,9 @@ class FakeEvent {
         ];
     }
     static getTestTable() {
-        return (0, test_db_1.table)("fake_events", (0, test_db_1.uuid)("id", { primaryKey: true }), (0, test_db_1.timestamptz)("created_at"), (0, test_db_1.timestamptz)("updated_at"), 
+        return (0, temp_db_1.table)("fake_events", (0, temp_db_1.uuid)("id", { primaryKey: true }), (0, temp_db_1.timestamptz)("created_at"), (0, temp_db_1.timestamptz)("updated_at"), 
         // TODO index:true
-        (0, test_db_1.timestamptz)("start_time"), (0, test_db_1.timestamptz)("end_time", { nullable: true }), (0, test_db_1.text)("location"), (0, test_db_1.text)("title"), (0, test_db_1.text)("description", { nullable: true }), (0, test_db_1.uuid)("user_id"));
+        (0, temp_db_1.timestamptz)("start_time"), (0, temp_db_1.timestamptz)("end_time", { nullable: true }), (0, temp_db_1.text)("location"), (0, temp_db_1.text)("title"), (0, temp_db_1.text)("description", { nullable: true }), (0, temp_db_1.uuid)("user_id"));
     }
     static loaderOptions() {
         return {
@@ -84,7 +87,7 @@ function getEventBuilder(viewer, input) {
     for (const key in input) {
         m.set(key, input[key]);
     }
-    return new builder_1.SimpleBuilder(viewer, exports.FakeEventSchema, m);
+    return new builder_1.SimpleBuilder(viewer, exports.FakeEventSchema, m, action_1.WriteOperation.Insert, null);
 }
 exports.getEventBuilder = getEventBuilder;
 async function createEvent(viewer, input) {

package/testutils/fake_data/fake_user.d.ts

@@ -23,10 +23,10 @@ export declare class FakeUser implements Ent {
     readonly emailAddress: string;
     readonly phoneNumber: string | null;
     protected readonly password: string | null;
-    privacyPolicy: PrivacyPolicy;
+    getPrivacyPolicy(): PrivacyPolicy<this>;
     constructor(viewer: Viewer, data: Data);
     static getFields(): string[];
-    static getTestTable(): import("../db/test_db").Table;
+    static getTestTable(): import("../db/temp_db").Table;
     static loaderOptions(): LoadEntOptions<FakeUser>;
     static load(v: Viewer, id: ID): Promise<FakeUser | null>;
     static loadX(v: Viewer, id: ID): Promise<FakeUser>;
@@ -40,8 +40,8 @@ export interface UserCreateInput {
     password: string | null;
 }
 export declare type UserEditInput = Partial<UserCreateInput>;
-export declare function getUserBuilder(viewer: Viewer, input: UserCreateInput): import("../builder").SimpleBuilder<FakeUser>;
-export declare function getUserAction(viewer: Viewer, input: UserCreateInput): SimpleAction<FakeUser>;
+export declare function getUserBuilder(viewer: Viewer, input: UserCreateInput): import("../builder").SimpleBuilder<FakeUser, null>;
+export declare function getUserAction(viewer: Viewer, input: UserCreateInput): SimpleAction<FakeUser, null>;
 export declare function createUser(viewer: Viewer, input: UserCreateInput): Promise<FakeUser>;
 export declare const userLoader: ObjectLoaderFactory<unknown>;
 export declare const userEmailLoader: ObjectLoaderFactory<unknown>;

package/testutils/fake_data/fake_user.js

@@ -8,9 +8,10 @@ const schema_1 = require("../../schema");
 const internal_1 = require("./internal");
 const const_1 = require("./const");
 const viewer_1 = require("../../core/viewer");
-const test_db_1 = require("../db/test_db");
+const temp_db_1 = require("../db/temp_db");
 const loaders_1 = require("../../core/loaders");
 const convert_1 = require("../../core/convert");
+const action_1 = require("../../action");
 class ViewerWithAccessToken extends viewer_1.IDViewer {
     constructor(viewerID, opts) {
         super(viewerID, opts);
@@ -26,7 +27,18 @@ class FakeUser {
     constructor(viewer, data) {
         this.viewer = viewer;
         this.nodeType = const_1.NodeType.FakeUser;
-        this.privacyPolicy = {
+        this.data = data;
+        this.id = data.id;
+        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;
+        this.phoneNumber = data.phone_number;
+        this.password = data.password;
+    }
+    getPrivacyPolicy() {
+        return {
             rules: [
                 privacy_1.AllowIfViewerRule,
                 //can view user if friends
@@ -48,15 +60,6 @@ class FakeUser {
                 privacy_1.AlwaysDenyRule,
             ],
         };
-        this.data = data;
-        this.id = data.id;
-        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;
-        this.phoneNumber = data.phone_number;
-        this.password = data.password;
     }
     static getFields() {
         return [
@@ -71,7 +74,7 @@ class FakeUser {
         ];
     }
     static getTestTable() {
-        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"));
+        return (0, temp_db_1.table)("fake_users", (0, temp_db_1.uuid)("id", { primaryKey: true }), (0, temp_db_1.timestamptz)("created_at"), (0, temp_db_1.timestamptz)("updated_at"), (0, temp_db_1.text)("first_name"), (0, temp_db_1.text)("last_name"), (0, temp_db_1.text)("email_address"), (0, temp_db_1.text)("phone_number"), (0, temp_db_1.text)("password"));
     }
     static loaderOptions() {
         return {
@@ -108,7 +111,7 @@ function getUserAction(viewer, input) {
     for (const key in input) {
         m.set(key, input[key]);
     }
-    const action = new builder_1.SimpleAction(viewer, exports.FakeUserSchema, m);
+    const action = new builder_1.SimpleAction(viewer, exports.FakeUserSchema, m, action_1.WriteOperation.Insert, null);
     action.viewerForEntLoad = (data) => {
         // load the created ent using a VC of the newly created user.
         return new viewer_1.IDViewer(data.id);

package/testutils/fake_data/test_helpers.d.ts

@@ -1,6 +1,6 @@
 import { Data, Ent } from "../../core/base";
 import { AssocEdge } from "../../core/ent";
-import { TempDB } from "../db/test_db";
+import { TempDB } from "../db/temp_db";
 import { FakeUser, UserCreateInput, ContactCreateInput, FakeContact } from ".";
 import { EventCreateInput, FakeEvent } from "./fake_event";
 import { BuilderSchema } from "../builder";
@@ -20,12 +20,13 @@ export declare function createEdges(): Promise<void>;
 export declare function edgeTableNames(): string[];
 export declare function createTestEvent(user: FakeUser, input?: Partial<EventCreateInput>): Promise<FakeEvent>;
 export declare function setupTempDB(): Promise<TempDB>;
-export declare function tempDBTables(): import("../db/test_db").Table[];
+export declare function tempDBTables(global?: boolean): import("../db/temp_db").Table[];
 interface options {
     howMany: number;
     interval: number;
     userInput?: Partial<UserCreateInput>;
     eventInputs?: Partial<EventCreateInput>[];
+    startTime?: number | Date;
 }
 export declare function createAllEvents(opts: options): Promise<[FakeUser, FakeEvent[]]>;
 export {};

package/testutils/fake_data/test_helpers.js

@@ -6,7 +6,7 @@ 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");
@@ -41,6 +41,8 @@ function getEventInput(user, input) {
         title: "title",
         description: "fun event",
         userID: user.id,
+        createdAt: new Date(),
+        updatedAt: new Date(),
         ...input,
     };
 }
@@ -209,21 +211,21 @@ async function createTestEvent(user, input) {
 }
 exports.createTestEvent = createTestEvent;
 async function setupTempDB() {
-    const tdb = new test_db_1.TempDB(tempDBTables());
+    const tdb = new temp_db_1.TempDB(tempDBTables());
     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)(),
+        (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 +234,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) {