@snowtop/ent 0.1.0-alpha88 → 0.1.0-alpha89-16dc3410-33c4-11ed-9fc4-5d4a927887be

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.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.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 +28,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 +51,12 @@ 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 {
+class UserToContactsFkeyQueryDeprecated extends custom_query_1.CustomEdgeQueryBase {
     constructor(viewer, src) {
         super(viewer, {
             src,
@@ -65,6 +66,24 @@ class UserToContactsFkeyQuery extends custom_query_1.CustomEdgeQueryBase {
             options: internal_1.FakeContact.loaderOptions(),
         });
     }
+    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",
+        });
+    }
     static query(viewer, src) {
         return new UserToContactsFkeyQuery(viewer, src);
     }
@@ -257,27 +276,15 @@ 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...
+            ...internal_1.FakeEvent.loaderOptions(),
+            groupCol: "user_id",
+            clause: (0, exports.getNextWeekClause)(),
+            loadEntOptions: internal_1.FakeEvent.loaderOptions(),
+            name: "events_in_next_week",
             sortColumn: "start_time",
         });
     }