Governance: Fix MultiChoice VoteType serialization

solana-labs · Oct 26, 2022 · 392cbe2 · 392cbe2
1 parent 3bbb3cc
commit 392cbe2
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 5 deletions.
diff --git a/packages/governance-sdk/src/governance/accounts.ts b/packages/governance-sdk/src/governance/accounts.ts
@@ -201,11 +201,13 @@ export enum VoteTypeKind {
 
 export class VoteType {
   type: VoteTypeKind;
-  choiceCount: number | undefined;
+  max_voter_options: number | undefined;
+  max_winning_options: number | undefined;
 
   constructor(args: { type: VoteTypeKind; choiceCount: number | undefined }) {
     this.type = args.type;
-    this.choiceCount = args.choiceCount;
+    this.max_voter_options = args.choiceCount;
+    this.max_winning_options = args.choiceCount;
   }
 
   static SINGLE_CHOICE = new VoteType({

diff --git a/packages/governance-sdk/src/governance/serialisation.ts b/packages/governance-sdk/src/governance/serialisation.ts
@@ -100,7 +100,8 @@ import { deserializeBorsh } from '../tools/borsh';
     return VoteType.SINGLE_CHOICE;
   }
 
-  const choiceCount = reader.buf.readUInt16LE(reader.offset);
+  const choiceCount = reader.buf.readUInt8(reader.offset);
+  reader.offset += 2; // skip two bytes
   return VoteType.MULTI_CHOICE(choiceCount);
 };
 
@@ -111,8 +112,11 @@ import { deserializeBorsh } from '../tools/borsh';
   writer.length += 1;
 
   if (value.type === VoteTypeKind.MultiChoice) {
-    writer.buf.writeUInt16LE(value.choiceCount!, writer.length);
-    writer.length += 2;
+    // max_voter_options and max_winning_options are not implemented at backend
+    writer.buf.writeUInt8(value.max_voter_options || 0, writer.length);
+    writer.length += 1;
+    writer.buf.writeUInt8(value.max_winning_options || 0, writer.length);
+    writer.length += 1;
   }
 };