Get tests passing.

Author: ejMina226

packages/protocol/src/protocol/Protocol.ts

@@ -132,15 +132,6 @@ export class Protocol<
   public create(childContainerProvider: ChildContainerProvider) {
     super.create(childContainerProvider);
 
-    const runtime: ModuleContainer<any> = this.container.resolve("Runtime");
-    runtime.moduleNames.forEach((runtimeModuleName) => {
-      this.container.register(runtimeModuleName, {
-        useFactory: (dependencyContainer) => {
-          return runtime.resolve(runtimeModuleName);
-        },
-      });
-    });
-
     // Register the BlockModules seperately since we need to
     // inject them differently later
     const ABSTRACT_MODULE_TYPES = [
@@ -189,6 +180,15 @@ export class Protocol<
         );
       }
     });
+
+    const runtime: ModuleContainer<any> = this.container.resolve("Runtime");
+    runtime.moduleNames.forEach((runtimeModuleName) => {
+      this.container.register(runtimeModuleName, {
+        useFactory: (dependencyContainer) => {
+          return runtime.resolve(runtimeModuleName);
+        },
+      });
+    });
   }
 
   public async start() {

packages/protocol/test/TestingProtocol.ts

@@ -1,21 +1,17 @@
-import { WithZkProgrammable, ZkProgrammable } from "@proto-kit/common";
 import { container } from "tsyringe";
+import { Runtime } from "@proto-kit/module";
+import { Balance } from "@proto-kit/sequencer/test/integration/mocks/Balance";
+import { NoopRuntime } from "@proto-kit/sequencer/test/integration/mocks/NoopRuntime";
 
 import {
   AccountStateHook,
   BlockHeightHook,
   BlockProver,
   LastStateRootBlockHook,
-  MethodPublicOutput,
   Protocol,
   StateTransitionProver,
 } from "../src";
 
-class RuntimeMock implements WithZkProgrammable<undefined, MethodPublicOutput> {
-  zkProgrammable: ZkProgrammable<undefined, MethodPublicOutput> =
-    undefined as unknown as ZkProgrammable<undefined, MethodPublicOutput>;
-}
-
 export function createAndInitTestingProtocol() {
   const ProtocolClass = Protocol.from({
     modules: {
@@ -35,11 +31,22 @@ export function createAndInitTestingProtocol() {
     StateTransitionProver: {},
     LastStateRoot: {},
   });
-  protocol.create(() => container.createChildContainer());
 
-  protocol.registerValue({
-    Runtime: new RuntimeMock(),
+  const appChain = container.createChildContainer();
+
+  appChain.register("Runtime", {
+    useClass: Runtime.from({
+      modules: {
+        Balance,
+        NoopRuntime,
+      },
+      config: {
+        Balance: {},
+        NoopRuntime: {},
+      },
+    }),
   });
+  protocol.create(() => appChain.createChildContainer());
 
   return protocol;
 }