Add key assignment struct

cosmos · Nov 1, 2022 · c75a588 · c75a588
1 parent 42c2593
commit c75a588
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 0 deletions.
diff --git a/tests/difference/core/model/src/common.ts b/tests/difference/core/model/src/common.ts
@@ -109,6 +109,10 @@ type EndAndBeginBlock = {
   chain: Chain;
 };
 
+type KeyAssignment = {
+  kind: string;
+};
+
 type SystemSnapshot = {
   h: Record<Chain, number>;
   t: Record<Chain, number>;
@@ -204,6 +208,7 @@ export {
   UpdateClient,
   Deliver,
   EndAndBeginBlock,
+  KeyAssignment,
   SystemSnapshot,
   Status,
   Undelegation,

diff --git a/tests/difference/core/model/src/main.ts b/tests/difference/core/model/src/main.ts
@@ -23,6 +23,7 @@ import {
   UpdateClient,
   Deliver,
   EndAndBeginBlock,
+  KeyAssignment,
   TraceAction,
   Chain,
   Consequence,
@@ -109,6 +110,11 @@ class ActionGenerator {
         chain: _.sample([P, C]) as Chain,
       } as EndAndBeginBlock;
     }
+    if (kind == 'KeyAssignment') {
+      return {
+        kind,
+      } as KeyAssignment;
+    }
     throw `kind doesn't match`;
   };
 
@@ -142,6 +148,9 @@ class ActionGenerator {
         this.tLastTrustedHeader[chain] + TRUSTING_SECONDS
       );
     }
+    if (a.kind === 'KeyAssignment') {
+      return true;
+    }
     throw `kind doesn't match`;
   };