wip

Author: Mr-Leshiy

rust/vote-tx-v2/src/decoding.rs

@@ -407,8 +407,7 @@ mod tests {
             voter_data: VoterData(voter_data),
         };
 
-        let signature = coset::CoseSign::default();
-        let generalized_tx = GeneralizedTx { tx_body, signature };
+        let generalized_tx = GeneralizedTx::new(tx_body);
 
         let bytes = generalized_tx.to_bytes().unwrap();
         let decoded = GeneralizedTx::from_bytes(&bytes).unwrap();

rust/vote-tx-v2/src/lib.rs

@@ -72,6 +72,24 @@ pub struct Proof(Vec<u8>);
 #[derive(Debug, Clone, PartialEq, Eq)]
 pub struct PropId(Vec<u8>);
 
+impl GeneralizedTx {
+    /// Creates a new `GeneralizedTx` struct.
+    #[must_use]
+    pub fn new(tx_body: TxBody) -> Self {
+        let signature = coset::CoseSignBuilder::new()
+            .protected(Self::cose_protected_header())
+            .build();
+        Self { tx_body, signature }
+    }
+
+    /// Returns the COSE protected header.
+    fn cose_protected_header() -> coset::Header {
+        coset::HeaderBuilder::new()
+            .content_format(coset::iana::CoapContentFormat::Cbor)
+            .build()
+    }
+}
+
 /// Cbor encodable and decodable type trait.
 pub trait Cbor<'a>: Encode<()> + Decode<'a, ()> {
     /// Encodes to CBOR encoded bytes.