@sentio/sdk 3.0.0-rc.8 → 3.0.0-rc.9

package/src/testing/test-processor-server.ts

@@ -14,10 +14,19 @@ import {
   ProcessStreamRequest,
   ProcessStreamResponse,
   ServerStreamingMethodResult,
-  StartRequest
+  StartRequest,
+  TemplateInstance,
+  TimeseriesResult
 } from '@sentio/protos'
 import { CallContext } from 'nice-grpc-common'
-import { Endpoints, PluginManager, ProcessorServiceImpl, State, StoreContext } from '@sentio/runtime'
+import {
+  Endpoints,
+  IDataBindingContext,
+  PluginManager,
+  ProcessorServiceImpl,
+  State,
+  StoreContext
+} from '@sentio/runtime'
 
 import { AptosFacet } from './aptos-facet.js'
 import { SolanaFacet } from './solana-facet.js'
@@ -45,7 +54,7 @@ export class TestProcessorServer implements ProcessorServiceImplementation {
   service: ProcessorServiceImpl
   contractConfigs: ContractConfig[]
   accountConfigs: AccountConfig[]
-  storeContext: StoreContext
+  storeContext: TestStoreContext
 
   aptos: AptosFacet
   eth: EthFacet
@@ -77,7 +86,7 @@ export class TestProcessorServer implements ProcessorServiceImplementation {
 
     // start a memory database for testing
     const subject = new Subject<DeepPartial<ProcessStreamResponse>>()
-    this.storeContext = new StoreContext(subject, 1)
+    this.storeContext = new TestStoreContext(subject, 1)
     this._db = new MemoryDatabase(this.storeContext)
   }
 
@@ -87,6 +96,7 @@ export class TestProcessorServer implements ProcessorServiceImplementation {
     this.contractConfigs = config.contractConfigs
     this.accountConfigs = config.accountConfigs
     this._db.start()
+    this.storeContext.templateInstances = request.templateInstances
     return res
   }
 
@@ -94,23 +104,43 @@ export class TestProcessorServer implements ProcessorServiceImplementation {
     return this.service.stop(request, context)
   }
 
-  getConfig(request: ProcessConfigRequest, context = TEST_CONTEXT): Promise<ProcessConfigResponse> {
-    return this.service.getConfig(request, context)
+  async getConfig(request: ProcessConfigRequest, context = TEST_CONTEXT): Promise<ProcessConfigResponse> {
+    const config = await this.service.getConfig(request, context)
+    return {
+      ...config,
+      templateInstances: this.storeContext.templateInstances
+    }
   }
 
   processBindings(
     request: ProcessBindingsRequest,
     context: CallContext = TEST_CONTEXT
   ): Promise<ProcessBindingResponse> {
-    return PluginManager.INSTANCE.dbContextLocalStorage.run(this.storeContext, () => {
-      return this.service.processBindings(request, context)
+    return PluginManager.INSTANCE.dbContextLocalStorage.run(this.storeContext, async () => {
+      const ret = await this.service.processBindings(request, context)
+      if (ret.result?.states?.configUpdated) {
+        // template may changed
+        await PluginManager.INSTANCE.updateTemplates({
+          chainId: request.bindings[0].chainId,
+          templateInstances: this.storeContext.templateInstances
+        })
+      }
+      return ret
     })
   }
 
-  processBinding(request: DataBinding, context: CallContext = TEST_CONTEXT): Promise<ProcessBindingResponse> {
-    return PluginManager.INSTANCE.dbContextLocalStorage.run(this.storeContext, () => {
+  async processBinding(request: DataBinding, context: CallContext = TEST_CONTEXT): Promise<ProcessBindingResponse> {
+    const ret = await PluginManager.INSTANCE.dbContextLocalStorage.run(this.storeContext, () => {
       return this.service.processBindings({ bindings: [request] }, context)
     })
+    if (ret.result?.states?.configUpdated) {
+      // template may changed
+      await PluginManager.INSTANCE.updateTemplates({
+        chainId: request.chainId,
+        templateInstances: this.storeContext.templateInstances
+      })
+    }
+    return ret
   }
 
   processBindingsStream(
@@ -138,3 +168,27 @@ export class TestProcessorServer implements ProcessorServiceImplementation {
     return this._db.store
   }
 }
+
+class TestStoreContext extends StoreContext implements IDataBindingContext {
+  constructor(
+    readonly subject: Subject<DeepPartial<ProcessStreamResponse>>,
+    processId: number
+  ) {
+    super(subject, processId)
+  }
+
+  templateInstances: TemplateInstance[] = []
+
+  sendTemplateRequest(templates: Array<TemplateInstance>, remove: boolean): void {
+    if (remove) {
+      this.templateInstances = this.templateInstances.filter(
+        (i) => !templates.find((t) => t.templateId === i.templateId && t.contract?.address === i.contract?.address)
+      )
+    } else {
+      this.templateInstances.push(...templates)
+    }
+  }
+  sendTimeseriesRequest(timeseries: Array<TimeseriesResult>): void {
+    throw new Error('Method not implemented.')
+  }
+}

package/lib/core/template.d.ts

@@ -1,2 +0,0 @@
-export { TemplateInstanceState } from '@sentio/runtime';
-//# sourceMappingURL=template.d.ts.map

package/lib/core/template.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"template.d.ts","sourceRoot":"","sources":["../../src/core/template.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,qBAAqB,EAAE,MAAM,iBAAiB,CAAA"}

package/lib/core/template.js

@@ -1,2 +0,0 @@
-export { TemplateInstanceState } from '@sentio/runtime';
-//# sourceMappingURL=template.js.map

package/lib/core/template.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"template.js","sourceRoot":"","sources":["../../src/core/template.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,qBAAqB,EAAE,MAAM,iBAAiB,CAAA"}

package/src/core/template.ts

@@ -1 +0,0 @@
-export { TemplateInstanceState } from '@sentio/runtime'