test: start DEX actions

penumbra-zone · Oct 30, 2024 · 3661cf0 · 3661cf0
1 parent f7acc3d
commit 3661cf0
Showing 1 changed file with 43 additions and 1 deletion.
diff --git a/crates/core/transaction/tests/generate_transaction_signing_test_vectors.rs b/crates/core/transaction/tests/generate_transaction_signing_test_vectors.rs
@@ -11,9 +11,13 @@ use ibc_types::core::{
 use ibc_types::timestamp::Timestamp;
 use penumbra_asset::asset::Id;
 use penumbra_dex::{
+    lp::{
+        position::{Position, State as PositionState},
+        Reserves, TradingFunction,
+    },
     swap::{SwapPlaintext, SwapPlan},
     swap_claim::SwapClaimPlan,
-    BatchSwapOutputData, TradingPair,
+    BatchSwapOutputData, PositionOpen, TradingPair,
 };
 use penumbra_fee::Fee;
 use penumbra_governance::{
@@ -441,6 +445,43 @@ fn proposal_deposit_claim_strategy() -> impl Strategy<Value = ProposalDepositCla
         })
 }
 
+fn position_state_strategy() -> impl Strategy<Value = PositionState> {
+    prop_oneof![Just(PositionState::Opened), Just(PositionState::Closed)]
+}
+
+fn trading_function_strategy() -> impl Strategy<Value = TradingFunction> {
+    (
+        amount_strategy(),
+        amount_strategy(),
+        asset_id_strategy(),
+        asset_id_strategy(),
+    )
+        .prop_map(|(p, q, asset_1, asset_2)| {
+            let trading_pair = TradingPair::new(asset_1, asset_2);
+            TradingFunction::new(trading_pair, 0u32, p, q)
+        })
+}
+
+fn position_strategy() -> impl Strategy<Value = Position> {
+    (
+        position_state_strategy(),
+        amount_strategy(),
+        amount_strategy(),
+        trading_function_strategy(),
+    )
+        .prop_map(|(state, r1, r2, phi)| Position {
+            state,
+            reserves: Reserves { r1, r2 },
+            phi,
+            nonce: [0u8; 32],
+            close_on_fill: true,
+        })
+}
+
+fn position_open_strategy() -> impl Strategy<Value = PositionOpen> {
+    (position_strategy()).prop_map(|position| PositionOpen { position })
+}
+
 fn action_plan_strategy(fvk: &FullViewingKey) -> impl Strategy<Value = ActionPlan> {
     prop_oneof![
         spend_plan_strategy(fvk).prop_map(ActionPlan::Spend),
@@ -457,6 +498,7 @@ fn action_plan_strategy(fvk: &FullViewingKey) -> impl Strategy<Value = ActionPla
         delegator_vote_strategy().prop_map(ActionPlan::DelegatorVote),
         validator_vote_strategy().prop_map(ActionPlan::ValidatorVote),
         proposal_deposit_claim_strategy().prop_map(ActionPlan::ProposalDepositClaim),
+        position_open_strategy().prop_map(ActionPlan::PositionOpen),
     ]
 }