@sentio/sdk-bundle 4.0.0-rc.2 → 4.0.0-rc.4

package/lib/testing/index.js

@@ -1,7 +1,7 @@
 import { createRequire as createRequireSdkShim } from 'module'; const require = createRequireSdkShim(import.meta.url);
 import {
   AptosNetwork
-} from "../chunk-TNDQS25D.js";
+} from "../chunk-ODKKD6SE.js";
 import "../chunk-EEZNVYCZ.js";
 import "../chunk-3CHN7BVE.js";
 import "../chunk-YNDNYJUQ.js";
@@ -13,21 +13,21 @@ import {
 import {
   getMoveCalls,
   getMoveCalls2
-} from "../chunk-3R7FPM4I.js";
+} from "../chunk-IQNHEJLA.js";
 import {
   SuiNetwork
-} from "../chunk-4NGWL7IU.js";
+} from "../chunk-TWK53OHY.js";
 import {
   IotaNetwork
-} from "../chunk-YSP7DBWI.js";
+} from "../chunk-ZB3ENFMP.js";
 import "../chunk-QUAXYDJ7.js";
 import "../chunk-ZJYIROIV.js";
 import "../chunk-742S42NO.js";
 import "../chunk-7IDGS56O.js";
 import {
   bytesToBigInt
-} from "../chunk-37EAJXVT.js";
-import "../chunk-XNXYRKKC.js";
+} from "../chunk-X6UU7BKN.js";
+import "../chunk-GEI3YMIY.js";
 import "../chunk-5TTNJF6G.js";
 import {
   SPLITTER,
@@ -51,10 +51,12 @@ import {
   ProcessBindingResponseSchema,
   ProcessBindingsRequestSchema,
   ProcessConfigRequestSchema,
+  ProcessStreamRequestSchema,
   RichValueListSchema,
   RichValueSchema,
   StartRequestSchema,
   Store,
+  TemplateInstanceSchema,
   UpdateTemplatesRequestSchema,
   buildSchema,
   create,
@@ -9753,11 +9755,12 @@ var require_cjs = __commonJS({
 
 // src/testing/test-processor-server.ts
 import {
+  DataBindingContext,
   Endpoints,
+  mergeProcessResults,
   PluginManager as PluginManager2,
-  ProcessorServiceImpl,
   State,
-  StoreContext
+  ProcessorServiceImplV3
 } from "@sentio/runtime";
 
 // src/testing/aptos-facet.ts
@@ -10731,7 +10734,7 @@ var MemoryDatabase = class {
           const entityName = entity[i];
           this.upsert(entityName, id, d);
         });
-        this.dbContext.result(create(DBResponseSchema, { opId: req.opId }));
+        this.sendResult(request, create(DBResponseSchema, { opId: req.opId }));
       }
       if (req.op.case === "delete") {
         const { id, entity } = req.op.value;
@@ -10739,12 +10742,13 @@ var MemoryDatabase = class {
           const entityName = entity[idx];
           this.delete(entityName, i);
         });
-        this.dbContext.result(create(DBResponseSchema, { opId: req.opId }));
+        this.sendResult(request, create(DBResponseSchema, { opId: req.opId }));
       }
       if (req.op.case === "get") {
         const { entity, id } = req.op.value;
         const data = this.getById(entity, id);
-        this.dbContext.result(
+        this.sendResult(
+          request,
           create(DBResponseSchema, {
             opId: req.opId,
             value: {
@@ -10761,7 +10765,8 @@ var MemoryDatabase = class {
         const list = this.listEntities(entity, filters);
         if (cursor) {
           const idx = parseInt(cursor);
-          this.dbContext.result(
+          this.sendResult(
+            request,
             create(DBResponseSchema, {
               opId: req.opId,
               value: { case: "entityList", value: { entities: list.slice(idx, idx + 1).map((d) => toEntity(d)) } },
@@ -10769,7 +10774,8 @@ var MemoryDatabase = class {
             })
           );
         } else {
-          this.dbContext.result(
+          this.sendResult(
+            request,
             create(DBResponseSchema, {
               opId: req.opId,
               value: { case: "entityList", value: { entities: list.length ? [toEntity(list[0])] : [] } },
@@ -10780,6 +10786,9 @@ var MemoryDatabase = class {
       }
     }
   }
+  sendResult(request, response) {
+    this.dbContext.result(response, request.processId);
+  }
   reset() {
     this.db.clear();
   }
@@ -11118,6 +11127,7 @@ var TestProcessorServer = class {
   contractConfigs;
   accountConfigs;
   storeContext;
+  nextProcessId = 1;
   aptos;
   eth;
   solana;
@@ -11128,7 +11138,7 @@ var TestProcessorServer = class {
   _db;
   constructor(loader, httpEndpoints = {}) {
     cleanTest();
-    this.service = new ProcessorServiceImpl(loader);
+    this.service = new ProcessorServiceImplV3(loader);
     this.aptos = new AptosFacet(this);
     this.solana = new SolanaFacet(this);
     this.eth = new EthFacet(this);
@@ -11141,7 +11151,7 @@ var TestProcessorServer = class {
       Endpoints.INSTANCE.chainServer.set(k, http);
     }
     const subject = new import_rxjs.Subject();
-    this.storeContext = new TestStoreContext(subject, 1);
+    this.storeContext = new TestStoreContext(subject, 1, this.service);
     this._db = new MemoryDatabase(this.storeContext);
   }
   async start(request = { templateInstances: [] }, context = TEST_CONTEXT) {
@@ -11155,7 +11165,8 @@ var TestProcessorServer = class {
     return res;
   }
   stop(request = create(EmptySchema), context = TEST_CONTEXT) {
-    return this.service.stop(request, context);
+    this._db.stop();
+    return request;
   }
   async getConfig(request = {}, context = TEST_CONTEXT) {
     const config = await this.service.getConfig(create(ProcessConfigRequestSchema, request), context);
@@ -11164,38 +11175,77 @@ var TestProcessorServer = class {
   }
   processBindings(request, context = TEST_CONTEXT) {
     const req = create(ProcessBindingsRequestSchema, request);
-    return PluginManager2.INSTANCE.dbContextLocalStorage.run(this.storeContext, async () => {
-      const ret = await this.service.processBindings(req, context);
-      if (ret.result?.states?.configUpdated) {
+    return this.processBindingList(req.bindings, context);
+  }
+  async processBinding(request, context = TEST_CONTEXT) {
+    return this.processBindingList([request], context);
+  }
+  async processBindingList(bindings, context = TEST_CONTEXT) {
+    const results = [];
+    for (const binding of bindings) {
+      const result = await this.processBindingV3(binding, context);
+      results.push(result);
+      if (result.states?.configUpdated) {
         await PluginManager2.INSTANCE.updateTemplates(
           create(UpdateTemplatesRequestSchema, {
-            chainId: req.bindings[0].chainId,
+            chainId: binding.chainId,
             templateInstances: this.storeContext.templateInstances
           })
         );
       }
-      return create(ProcessBindingResponseSchema, ret);
+    }
+    return create(ProcessBindingResponseSchema, {
+      result: mergeProcessResults(results)
     });
   }
-  async processBinding(request, context = TEST_CONTEXT) {
-    const ret = await PluginManager2.INSTANCE.dbContextLocalStorage.run(this.storeContext, () => {
-      return this.service.processBindings(create(ProcessBindingsRequestSchema, { bindings: [request] }), context);
+  processBindingV3(request, context) {
+    const processId = this.nextProcessId++;
+    const subject = this.storeContext.subject;
+    return new Promise((resolve, reject) => {
+      const subscription = subject.subscribe({
+        next: /* @__PURE__ */ __name((response) => {
+          if (response.processId !== processId) {
+            return;
+          }
+          if (response.value?.case === "tplRequest") {
+            this.storeContext.applyTemplateRequest(
+              (response.value.value.templates ?? []).map((template) => create(TemplateInstanceSchema, template)),
+              response.value.value.remove ?? false
+            );
+          }
+          if (response.value?.case === "result") {
+            subscription.unsubscribe();
+            const result = response.value.value;
+            if (result.states?.error) {
+              reject(new Error(result.states.error));
+            } else {
+              resolve(result);
+            }
+          }
+        }, "next"),
+        error: /* @__PURE__ */ __name((e) => {
+          subscription.unsubscribe();
+          reject(e);
+        }, "error")
+      });
+      this.service.handleRequest(
+        create(ProcessStreamRequestSchema, {
+          processId,
+          value: {
+            case: "binding",
+            value: request
+          }
+        }),
+        void 0,
+        subject
+      ).catch((e) => {
+        subscription.unsubscribe();
+        reject(e);
+      });
     });
-    if (ret.result?.states?.configUpdated) {
-      await PluginManager2.INSTANCE.updateTemplates(
-        create(UpdateTemplatesRequestSchema, {
-          chainId: request.chainId,
-          templateInstances: this.storeContext.templateInstances
-        })
-      );
-    }
-    return create(ProcessBindingResponseSchema, ret);
   }
   processBindingsStream(requests, context) {
-    throw new Error("Method not implemented.");
-  }
-  preprocessBindingsStream(requests, context) {
-    throw new Error("Method not implemented.");
+    return this.service.processBindingsStream(requests, context);
   }
   // processBindingsStream(request: AsyncIterable<ProcessStreamRequest>, context: HandlerContext) {
   //   return this.service.processBindingsStream(request, context)
@@ -11207,16 +11257,30 @@ var TestProcessorServer = class {
     return this._db.store;
   }
 };
-var TestStoreContext = class extends StoreContext {
-  constructor(subject, processId) {
-    super(subject, processId);
-    this.subject = subject;
+var TestStoreContext = class extends DataBindingContext {
+  constructor(subject, processId, service) {
+    super(processId, subject);
+    this.service = service;
   }
   static {
     __name(this, "TestStoreContext");
   }
   templateInstances = [];
-  sendTemplateRequest(templates, remove) {
+  result(dbResult, processId = this.processId) {
+    super.result(dbResult);
+    void this.service.handleRequest(
+      create(ProcessStreamRequestSchema, {
+        processId,
+        value: {
+          case: "dbResult",
+          value: dbResult
+        }
+      }),
+      void 0,
+      this.subject
+    );
+  }
+  applyTemplateRequest(templates, remove) {
     if (remove) {
       this.templateInstances = this.templateInstances.filter(
         (i) => !templates.find((t) => t.templateId === i.templateId && t.contract?.address === i.contract?.address)
@@ -11225,8 +11289,10 @@ var TestStoreContext = class extends StoreContext {
       this.templateInstances.push(...templates);
     }
   }
+  sendTemplateRequest(templates, remove) {
+    this.applyTemplateRequest(templates, remove);
+  }
   sendTimeseriesRequest(timeseries) {
-    throw new Error("Method not implemented.");
   }
 };
 import("node:process").then((p) => p.stdout.write(""));