From b2ca522e7357e4b3ea00a9b1a83ffe16f93ea7d3 Mon Sep 17 00:00:00 2001
From: Chirantan Ekbote <cekbote@jumptrading.com>
Date: Mon, 23 Jan 2023 19:07:54 +0900
Subject: [PATCH] sdk/rust: Add test with hardcoded digest

Add a test for digest calculation that uses a hardcoded digest.  The
digest for this test was calculated using the `worm` CLI to ensure
interoperability.
---
 sdk/rust/core/src/vaa.rs | 57 ++++++++++++++++++++++++++++++++++++++++
 1 file changed, 57 insertions(+)

diff --git a/sdk/rust/core/src/vaa.rs b/sdk/rust/core/src/vaa.rs
index ea9c9f530..6e22847be 100644
--- a/sdk/rust/core/src/vaa.rs
+++ b/sdk/rust/core/src/vaa.rs
@@ -272,6 +272,11 @@ impl<P: Serialize> Body<P> {
 mod test {
     use serde_wormhole::RawMessage;
 
+    use crate::{
+        token::{Action, GovernancePacket},
+        GOVERNANCE_EMITTER,
+    };
+
     use super::*;
 
     #[test]
@@ -348,4 +353,56 @@ mod test {
 
         assert_eq!(d1, d3);
     }
+
+    #[test]
+    fn stable_digest() {
+        let data = [
+            0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x01, 0x00, 0x01, 0x00, 0x00, 0x00, 0x00,
+            0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
+            0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x04,
+            0x00, 0x00, 0x00, 0x00, 0x03, 0xb4, 0x56, 0xb8, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
+            0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
+            0x00, 0x00, 0x54, 0x6f, 0x6b, 0x65, 0x6e, 0x42, 0x72, 0x69, 0x64, 0x67, 0x65, 0x01,
+            0x00, 0x00, 0x00, 0x02, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
+            0x00, 0x00, 0x02, 0x90, 0xfb, 0x16, 0x72, 0x08, 0xaf, 0x45, 0x5b, 0xb1, 0x37, 0x78,
+            0x01, 0x63, 0xb7, 0xb7, 0xa9, 0xa1, 0x0c, 0x16,
+        ];
+
+        let expected_digest = [
+            0x05, 0xd1, 0xfc, 0xc5, 0x31, 0x74, 0x6c, 0x7e, 0xfd, 0x7f, 0xee, 0xa2, 0x0a, 0x81,
+            0xd2, 0x79, 0x9f, 0x77, 0x7f, 0x30, 0x2b, 0x8a, 0x6a, 0x64, 0x24, 0xb8, 0x12, 0x09,
+            0xdc, 0x3f, 0x51, 0x1f,
+        ];
+
+        assert_eq!(expected_digest, digest(&data).unwrap().secp256k_hash);
+
+        let expected_body = Body {
+            timestamp: 1,
+            nonce: 1,
+            emitter_chain: Chain::Solana,
+            emitter_address: GOVERNANCE_EMITTER,
+            sequence: 62150328,
+            consistency_level: 0,
+            payload: GovernancePacket {
+                chain: Chain::Any,
+                action: Action::RegisterChain {
+                    chain: Chain::Ethereum,
+                    emitter_address: Address([
+                        0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
+                        0x02, 0x90, 0xfb, 0x16, 0x72, 0x08, 0xaf, 0x45, 0x5b, 0xb1, 0x37, 0x78,
+                        0x01, 0x63, 0xb7, 0xb7, 0xa9, 0xa1, 0x0c, 0x16,
+                    ]),
+                },
+            },
+        };
+
+        let body = serde_wormhole::from_slice(&data).unwrap();
+        assert_eq!(expected_body, body);
+        assert_eq!(expected_digest, body.digest().unwrap().secp256k_hash);
+
+        // Deferred parsing of the payload should still produce the same digest.
+        let body = serde_wormhole::from_slice::<Body<&RawMessage>>(&data).unwrap();
+        assert_eq!(&data[51..], body.payload.get());
+        assert_eq!(expected_digest, body.digest().unwrap().secp256k_hash);
+    }
 }