chore: add get_strategy_type (#525)

snapshot-labs · Aug 30, 2023 · aaa5b80 · aaa5b80
1 parent d75fe00
commit aaa5b80
Show file tree

Hide file tree

Showing 6 changed files with 33 additions and 0 deletions.
diff --git a/starknet/src/execution_strategies/vanilla.cairo b/starknet/src/execution_strategies/vanilla.cairo
@@ -33,6 +33,10 @@ mod VanillaExecutionStrategy {
             );
             self._num_executed.write(self._num_executed.read() + 1);
         }
+
+        fn get_strategy_type(self: @ContractState) -> felt252 {
+            'SimpleQuorumVanilla'
+        }
     }
 
     #[constructor]

diff --git a/starknet/src/interfaces/i_execution_strategy.cairo b/starknet/src/interfaces/i_execution_strategy.cairo
@@ -11,4 +11,6 @@ trait IExecutionStrategy<TContractState> {
         votes_abstain: u256,
         payload: Array<felt252>
     );
+
+    fn get_strategy_type(self: @TContractState) -> felt252;
 }
diff --git a/starknet/src/tests.cairo b/starknet/src/tests.cairo
@@ -4,6 +4,7 @@ mod test_space;
 mod test_upgrade;
 mod test_stark_tx_auth;
 
+mod execution_strategies;
 mod proposal_validation_strategies;
 mod voting_strategies;
 

diff --git a/starknet/src/tests/execution_strategies.cairo b/starknet/src/tests/execution_strategies.cairo
@@ -0,0 +1 @@
+mod vanilla;
diff --git a/starknet/src/tests/execution_strategies/vanilla.cairo b/starknet/src/tests/execution_strategies/vanilla.cairo
@@ -0,0 +1,17 @@
+#[cfg(test)]
+mod tests {
+    use sx::execution_strategies::vanilla::{VanillaExecutionStrategy};
+
+    #[test]
+    #[available_gas(10000000)]
+    fn get_strategy_type() {
+        let mut state: VanillaExecutionStrategy::ContractState =
+            VanillaExecutionStrategy::unsafe_new_contract_state();
+
+        let strategy_type = VanillaExecutionStrategy::VanillaExecutionStrategy::get_strategy_type(
+            @state
+        );
+
+        assert(strategy_type == 'SimpleQuorumVanilla', 'invalid strategy type');
+    }
+}
diff --git a/starknet/src/tests/mocks/executor.cairo b/starknet/src/tests/mocks/executor.cairo
@@ -33,6 +33,10 @@ mod ExecutorExecutionStrategy {
             let tx: Transaction = Serde::<Transaction>::deserialize(ref sp4n).unwrap();
             call_contract_syscall(tx.target, tx.selector, tx.data.span()).unwrap();
         }
+
+        fn get_strategy_type(self: @ContractState) -> felt252 {
+            'Executor'
+        }
     }
 
     #[constructor]
@@ -60,6 +64,10 @@ mod ExecutorWithoutTxExecutionStrategy {
             votes_abstain: u256,
             payload: Array<felt252>
         ) {}
+
+        fn get_strategy_type(self: @ContractState) -> felt252 {
+            'ExecutorWithoutTx'
+        }
     }
 
     #[constructor]