Add retrieve and redistribute tests

demox-labs · Aug 31, 2024 · 4a135b0 · 4a135b0
1 parent 8085635
commit 4a135b0
Show file tree

Hide file tree

Showing 4 changed files with 626 additions and 2 deletions.
diff --git a/pondo-bot/src/protocol/delegators.ts b/pondo-bot/src/protocol/delegators.ts
@@ -45,7 +45,12 @@ const unbondDelegator = async (delegatorProgramId: string) => {
   console.log(`Unbonding delegator ${delegatorProgramId}`);
   const delegatorProgram = await getProgram(delegatorProgramId);
   const delegatorProgramAddress = Aleo.Program.fromString(NETWORK!, delegatorProgram).toAddress();
-  const bondedState = JSON.parse(formatAleoString(await getMappingValue(delegatorProgramAddress, CREDITS_PROGRAM, 'bonded')));
+  const bondedStateString = await getMappingValue(delegatorProgramAddress, CREDITS_PROGRAM, 'bonded');
+  if (!bondedStateString) {
+    console.log(`Delegator ${delegatorProgramId} was forcibly unbonded`);
+    return finalizeToTerminalState(delegatorProgramId);
+  }
+  const bondedState = JSON.parse(formatAleoString(bondedStateString));
   const balance = BigInt(bondedState["microcredits"].slice(0, -3));
 
   console.log(`Delegator ${delegatorProgramId} has balance ${balance}`);

diff --git a/pondo-bot/src/protocol/runProtocol.ts b/pondo-bot/src/protocol/runProtocol.ts
@@ -189,7 +189,7 @@ const prepRebalance = async (pondoDelegatorStates: string[]): Promise<void> => {
   }
 };
 
-const getTopValidators = async (): Promise<string> => {
+export const getTopValidators = async (): Promise<string> => {
   let topValidators = await getMappingValue(
     '1u8',
     CORE_PROTOCOL_PROGRAM,