slightly more succinct

ZcashFoundation · Feb 28, 2024 · 5984b87 · 5984b87
1 parent 6cdeaad
commit 5984b87
Showing 1 changed file with 26 additions and 27 deletions.
diff --git a/zebra-chain/src/test_utils/vectors.rs b/zebra-chain/src/test_utils/vectors.rs
@@ -243,35 +243,34 @@ mod vector_tests {
                 }
             }
         }
-
-
         #[test]
         fn get_block_sapling_roots_bytes(network in networks()) {
-            let mainnet = Network::Mainnet;
-            let testnet = Network::Testnet;
-            let result = mainnet.get_block_sapling_roots_bytes(0, 1116000);
-            matches!(
-                result,
-                Err(SerializationError::NotACachedMainNetSaplingRootBytes(0))
-            );
-            let result = mainnet.get_block_sapling_roots_bytes(1046400, 0).unwrap();
-            let _correct_main_result: (&[u8], [u8; 32]) = (
-                &BLOCK_MAINNET_1046400_BYTES[..],
-                *SAPLING_FINAL_ROOT_MAINNET_1046400_BYTES,
-            );
-            matches!(result, _correct_main_result);
-
-            let result = testnet.get_block_sapling_roots_bytes(1046400, 0);
-            matches!(
-                result,
-                Err(SerializationError::NotACachedTestNetSaplingRootBytes(0))
-            );
-            let result = testnet.get_block_sapling_roots_bytes(0, 1116000).unwrap();
-            let _correct_test_result: (&[u8], [u8; 32]) = (
-                &BLOCK_TESTNET_1116000_BYTES[..],
-                *SAPLING_FINAL_ROOT_TESTNET_1116000_BYTES,
-            );
-            matches!(result, _correct_test_result);
+            match network {
+                Network::Mainnet => {
+                    matches!(
+                        network.get_block_sapling_roots_bytes(0, 1116000),
+                        Err(SerializationError::NotACachedMainNetSaplingRootBytes(0))
+                    );
+                    let result = network.get_block_sapling_roots_bytes(1046400, 0).unwrap();
+                    let _correct_main_result: (&[u8], [u8; 32]) = (
+                        &BLOCK_MAINNET_1046400_BYTES[..],
+                        *SAPLING_FINAL_ROOT_MAINNET_1046400_BYTES,
+                    );
+                    matches!(result, _correct_main_result);
+                },
+                Network::Testnet => {
+                    matches!(
+                        network.get_block_sapling_roots_bytes(1046400, 0),
+                        Err(SerializationError::NotACachedTestNetSaplingRootBytes(0))
+                    );
+                    let result = network.get_block_sapling_roots_bytes(0, 1116000).unwrap();
+                    let _correct_test_result: (&[u8], [u8; 32]) = (
+                        &BLOCK_TESTNET_1116000_BYTES[..],
+                        *SAPLING_FINAL_ROOT_TESTNET_1116000_BYTES,
+                    );
+                    matches!(result, _correct_test_result);
+                }
+            }
         }
     }