Merge #36: Add test for sign_tx()

a1a1454 Add test for sign_tx (wszdexdrf)
43bd579 Add deserialize function for PSBT (wszdexdrf)
8d3c29b Change `automation.json` for signing tx. (wszdexdrf)

Pull request description:

  Also made necessary changes to automation.json (for ledger devices). Add
  deserializing method for PSBT.

ACKs for top commit:
  danielabrozzoni:
    utACK a1a1454

Tree-SHA512: 2d222006918c420d3d1cc21b7dc09494445791c10d067ba412551d419a3217e682ee73dce4fbc215cf8f1e20c32e828e1a3d76968e07999dfc0a8374ad93dd86

Author: danielabrozzoni

ci/automation.json

@@ -2,7 +2,7 @@
     "version": 1,
     "rules": [
         {
-            "regexp": "Address \\(\\d/\\d\\)|Message hash \\(\\d/\\d\\)",
+            "regexp": "Address \\(\\d/\\d\\)|Message hash \\(\\d/\\d\\)|Confirm|Fees|Review|Amount",
             "actions": [
                 [ "button", 2, true ],
                 [ "button", 2, false ]
@@ -20,7 +20,7 @@
             ]
         },
         {
-            "regexp": "Approve|Sign",
+            "regexp": "Approve|Sign|Accept",
             "actions": [
                 [ "button", 3, true ],
                 [ "button", 3, false ]

src/lib.rs

@@ -37,9 +37,13 @@ pub mod types;
 mod tests {
     use crate::types;
     use crate::HWIClient;
+    use std::collections::BTreeMap;
     use std::str::FromStr;
 
-    use bitcoin::util::bip32::DerivationPath;
+    use bitcoin::psbt::{Input, Output};
+    use bitcoin::util::bip32::{DerivationPath, KeySource};
+    use bitcoin::{secp256k1, Transaction};
+    use bitcoin::{TxIn, TxOut};
 
     #[test]
     #[serial]
@@ -145,10 +149,66 @@ mod tests {
     // #[serial]
     // fn test_display_address_with_path_taproot() {}
 
-    // TODO: Create PSBT with scratch using given Hardware Wallet
-    // #[test]
-    // #[serial]
-    // fn test_sign_tx() {}
+    #[test]
+    #[serial]
+    fn test_sign_tx() {
+        let devices = HWIClient::enumerate().unwrap();
+        let device = devices.first().unwrap();
+        let client = HWIClient::get_client(device, true, types::HWIChain::Test).unwrap();
+        let derivation_path = DerivationPath::from_str("m/44'/1'/0'/0/0").unwrap();
+
+        let address = client
+            .display_address_with_path(&derivation_path, types::HWIAddressType::Legacy)
+            .unwrap();
+
+        let pk = client.get_xpub(&derivation_path, true).unwrap();
+        let mut hd_keypaths: BTreeMap<secp256k1::PublicKey, KeySource> = Default::default();
+        // Here device fingerprint is same as master xpub fingerprint
+        hd_keypaths.insert(pk.public_key, (device.fingerprint, derivation_path));
+
+        let previous_tx = Transaction {
+            version: 1,
+            lock_time: 0,
+            input: vec![TxIn::default()],
+            output: vec![TxOut {
+                value: 100,
+                script_pubkey: address.address.script_pubkey(),
+            }],
+        };
+
+        let previous_txin = TxIn {
+            previous_output: bitcoin::OutPoint {
+                txid: previous_tx.txid(),
+                vout: Default::default(),
+            },
+            ..Default::default()
+        };
+        let psbt = bitcoin::psbt::PartiallySignedTransaction {
+            unsigned_tx: Transaction {
+                version: 1,
+                lock_time: 0,
+                input: vec![previous_txin],
+                output: vec![TxOut {
+                    value: 50,
+                    script_pubkey: address.address.script_pubkey(),
+                }],
+            },
+            xpub: Default::default(),
+            version: Default::default(),
+            proprietary: Default::default(),
+            unknown: Default::default(),
+
+            inputs: vec![Input {
+                non_witness_utxo: Some(previous_tx),
+                witness_utxo: None,
+                bip32_derivation: hd_keypaths,
+                ..Default::default()
+            }],
+            outputs: vec![Output::default()],
+        };
+        let client = get_first_device();
+        client.sign_tx(&psbt).unwrap();
+    }
 
     #[test]
     #[serial]

src/types.rs

@@ -49,9 +49,19 @@ pub struct HWIAddress {
 
 #[derive(Clone, Eq, PartialEq, Debug, Deserialize)]
 pub struct HWIPartiallySignedTransaction {
+    #[serde(deserialize_with = "deserialize_psbt")]
     pub psbt: PartiallySignedTransaction,
 }
 
+fn deserialize_psbt<'de, D: Deserializer<'de>>(
+    d: D,
+) -> Result<PartiallySignedTransaction, D::Error> {
+    let b64_string = String::deserialize(d)?;
+    let bytes = bitcoin::base64::decode(&b64_string).map_err(serde::de::Error::custom)?;
+    bitcoin::consensus::deserialize::<PartiallySignedTransaction>(&bytes[..])
+        .map_err(serde::de::Error::custom)
+}
+
 impl Deref for HWIPartiallySignedTransaction {
     type Target = PartiallySignedTransaction;