add tests for enum types against Trezor proto enums

mintlayer · Dec 13, 2024 · d8de7c1 · d8de7c1
1 parent b1a5be4
commit d8de7c1
Show file tree

Hide file tree

Showing 4 changed files with 313 additions and 48 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/trezor-common/Cargo.toml b/trezor-common/Cargo.toml
@@ -17,5 +17,6 @@ strum = { version = "0.26", default-features = false, features = ["derive"] }
 test-utils = { path = "../test-utils" }
 common = { path = "../common" }
 crypto = { path = "../crypto/" }
+trezor-client = { git = "https://github.com/mintlayer/mintlayer-trezor-firmware", branch = "feature/mintlayer-pk", features = ["bitcoin", "mintlayer"] }
 
 rstest.workspace = true
diff --git a/trezor-common/src/lib.rs b/trezor-common/src/lib.rs
@@ -77,7 +77,7 @@ pub enum OutputValue {
 }
 
 #[derive(Debug, Clone, PartialEq, Eq, Encode, Decode, EnumDiscriminants)]
-#[strum_discriminants(name(OutputTimeLockIndex), derive(EnumIter, FromPrimitive))]
+#[strum_discriminants(name(OutputTimeLockType), derive(EnumIter, FromPrimitive))]
 pub enum OutputTimeLock {
     #[codec(index = 0)]
     UntilHeight(#[codec(compact)] u64),
@@ -154,7 +154,7 @@ pub enum IsTokenFreezable {
 }
 
 #[derive(Debug, Clone, PartialEq, Eq, Encode, Decode, EnumDiscriminants)]
-#[strum_discriminants(name(TokenTotalSupplyIndex), derive(EnumIter, FromPrimitive))]
+#[strum_discriminants(name(TokenTotalSupplyType), derive(EnumIter, FromPrimitive))]
 pub enum TokenTotalSupply {
     #[codec(index = 0)]
     Fixed(Amount), // fixed to a certain amount
@@ -267,7 +267,7 @@ pub struct H256(pub [u8; 32]);
 pub struct HtlcSecretHash(pub [u8; 20]);
 
 #[derive(Debug, Clone, PartialEq, Eq, Encode, Decode, Ord, PartialOrd, EnumDiscriminants)]
-#[strum_discriminants(name(OutPointSourceIdIndex), derive(EnumIter, FromPrimitive))]
+#[strum_discriminants(name(OutPointSourceIdType), derive(EnumIter, FromPrimitive))]
 pub enum OutPointSourceId {
     #[codec(index = 0)]
     Transaction(H256),
@@ -323,7 +323,7 @@ type OrderId = H256;
 type TokenId = H256;
 
 #[derive(Encode, EnumDiscriminants)]
-#[strum_discriminants(name(AccountCommandIndex), derive(EnumIter, FromPrimitive))]
+#[strum_discriminants(name(AccountCommandType), derive(EnumIter, FromPrimitive))]
 pub enum AccountCommand {
     // Create certain amount of tokens and add them to circulating supply
     #[codec(index = 0)]