Implement test function that calls action_generator_alice

xmr-btc/src/lib.rs

@@ -342,7 +342,7 @@ pub fn action_generator_alice<N, M, B>(
 ) -> GenBoxed<AliceAction, (), ()>
 where
     N: ReceiveBitcoinRedeemEncsig + Send + Sync,
-    M: monero::WatchForTransfer + Send + Sync,
+    M: Send + Sync,
     B: MedianTime + bitcoin::WatchForRawTransaction + Send + Sync,
 {
     enum SwapFailed {

xmr-btc/tests/on_chain.rs

@@ -2,27 +2,90 @@ pub mod harness;
 
 use anyhow::Result;
 use async_trait::async_trait;
+use futures::{
+    channel::mpsc::{Receiver, Sender},
+    SinkExt, StreamExt,
+};
 use genawaiter::GeneratorState;
 use harness::wallet::{bitcoin, monero};
 use xmr_btc::{
-    action_generator_bob,
-    bitcoin::{BroadcastSignedTransaction, SignTxLock},
+    action_generator_alice, action_generator_bob, alice,
+    bitcoin::{BroadcastSignedTransaction, EncryptedSignature, SignTxLock},
     bob,
-    monero::CreateWalletForOutput,
-    BobAction, ReceiveTransferProof,
+    monero::{CreateWalletForOutput, Transfer, TransferProof},
+    AliceAction, BobAction, ReceiveBitcoinRedeemEncsig, ReceiveTransferProof,
 };
 
-struct Network;
+type AliceNetwork = Network<EncryptedSignature>;
+type BobNetwork = Network<TransferProof>;
+
+#[derive(Debug)]
+struct Network<RecvMsg> {
+    // TODO: It is weird to use mpsc's in a situation where only one message is expected, but the
+    // ownership rules of Rust are making this painful
+    pub receiver: Receiver<RecvMsg>,
+}
 
 #[async_trait]
-impl ReceiveTransferProof for Network {
-    async fn receive_transfer_proof(&mut self) -> xmr_btc::monero::TransferProof {
-        todo!("use libp2p")
+impl ReceiveTransferProof for BobNetwork {
+    async fn receive_transfer_proof(&mut self) -> TransferProof {
+        self.receiver.next().await.unwrap()
+    }
+}
+
+#[async_trait]
+impl ReceiveBitcoinRedeemEncsig for AliceNetwork {
+    async fn receive_bitcoin_redeem_encsig(&mut self) -> EncryptedSignature {
+        self.receiver.next().await.unwrap()
+    }
+}
+
+async fn swap_as_alice(
+    network: &'static mut AliceNetwork,
+    // FIXME: It would be more intuitive to have a single network/transport struct instead of
+    // splitting into two, but Rust ownership rules make this tedious
+    mut sender: Sender<TransferProof>,
+    monero_wallet: &'static monero::AliceWallet<'static>,
+    bitcoin_wallet: &'static bitcoin::Wallet,
+    state: alice::State3,
+) -> Result<()> {
+    let mut action_generator =
+        action_generator_alice(network, monero_wallet, bitcoin_wallet, state);
+
+    loop {
+        match action_generator.async_resume().await {
+            GeneratorState::Yielded(AliceAction::LockXmr {
+                amount,
+                public_spend_key,
+                public_view_key,
+            }) => {
+                let (transfer_proof, _) = monero_wallet
+                    .transfer(public_spend_key, public_view_key, amount)
+                    .await?;
+
+                sender.send(transfer_proof).await.unwrap();
+            }
+            GeneratorState::Yielded(AliceAction::RedeemBtc(tx))
+            | GeneratorState::Yielded(AliceAction::CancelBtc(tx))
+            | GeneratorState::Yielded(AliceAction::PunishBtc(tx)) => {
+                let _ = bitcoin_wallet.broadcast_signed_transaction(tx).await?;
+            }
+            GeneratorState::Yielded(AliceAction::CreateMoneroWalletForOutput {
+                spend_key,
+                view_key,
+            }) => {
+                monero_wallet
+                    .create_and_load_wallet_for_output(spend_key, view_key)
+                    .await?;
+            }
+            GeneratorState::Complete(()) => return Ok(()),
+        }
     }
 }
 
 async fn swap_as_bob(
-    network: &'static mut Network,
+    network: &'static mut BobNetwork,
+    mut sender: Sender<EncryptedSignature>,
     monero_wallet: &'static monero::BobWallet<'static>,
     bitcoin_wallet: &'static bitcoin::Wallet,
     state: bob::State2,
@@ -37,8 +100,8 @@ async fn swap_as_bob(
                     .broadcast_signed_transaction(signed_tx_lock)
                     .await?;
             }
-            GeneratorState::Yielded(BobAction::SendBitcoinRedeemEncsig(_tx_redeem_encsig)) => {
-                todo!("use libp2p")
+            GeneratorState::Yielded(BobAction::SendBitcoinRedeemEncsig(tx_redeem_encsig)) => {
+                sender.send(tx_redeem_encsig).await.unwrap();
             }
             GeneratorState::Yielded(BobAction::CreateMoneroWalletForOutput {
                 spend_key,