governance.move 23 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415
  1. module pyth::governance {
  2. use wormhole::vaa::{Self, VAA};
  3. use pyth::data_source::{Self};
  4. use wormhole::u16;
  5. use pyth::governance_instruction;
  6. use pyth::governance_action;
  7. use pyth::contract_upgrade;
  8. use pyth::set_governance_data_source;
  9. use pyth::set_data_sources;
  10. use pyth::set_stale_price_threshold;
  11. use pyth::error;
  12. use pyth::set_update_fee;
  13. use pyth::state;
  14. public entry fun execute_governance_instruction(vaa_bytes: vector<u8>) {
  15. let parsed_vaa = parse_and_verify_governance_vaa(vaa_bytes);
  16. let instruction = governance_instruction::from_byte_vec(vaa::destroy(parsed_vaa));
  17. // Dispatch the instruction to the appropiate handler
  18. let action = governance_instruction::get_action(&instruction);
  19. if (action == governance_action::new_contract_upgrade()) {
  20. assert!(governance_instruction::get_target_chain_id(&instruction) != 0,
  21. error::governance_contract_upgrade_chain_id_zero());
  22. contract_upgrade::execute(governance_instruction::destroy(instruction));
  23. } else if (action == governance_action::new_set_governance_data_source()) {
  24. set_governance_data_source::execute(governance_instruction::destroy(instruction));
  25. } else if (action == governance_action::new_set_data_sources()) {
  26. set_data_sources::execute(governance_instruction::destroy(instruction));
  27. } else if (action == governance_action::new_set_update_fee()) {
  28. set_update_fee::execute(governance_instruction::destroy(instruction));
  29. } else if (action == governance_action::new_set_stale_price_threshold()) {
  30. set_stale_price_threshold::execute(governance_instruction::destroy(instruction));
  31. } else {
  32. governance_instruction::destroy(instruction);
  33. assert!(false, error::invalid_governance_action());
  34. }
  35. }
  36. fun parse_and_verify_governance_vaa(bytes: vector<u8>): VAA {
  37. let parsed_vaa = vaa::parse_and_verify(bytes);
  38. // Check that the governance data source is valid
  39. assert!(
  40. state::is_valid_governance_data_source(
  41. data_source::new(
  42. u16::to_u64(vaa::get_emitter_chain(&parsed_vaa)),
  43. vaa::get_emitter_address(&parsed_vaa))),
  44. error::invalid_governance_data_source());
  45. // Check that the sequence number is greater than the last executed governance VAA
  46. let sequence = vaa::get_sequence(&parsed_vaa);
  47. assert!(sequence > state::get_last_executed_governance_sequence(), error::invalid_governance_sequence_number());
  48. state::set_last_executed_governance_sequence(sequence);
  49. parsed_vaa
  50. }
  51. }
  52. #[test_only]
  53. module pyth::governance_test {
  54. use pyth::data_source::{Self, DataSource};
  55. use pyth::pyth;
  56. use pyth::governance;
  57. use pyth::contract_upgrade_hash;
  58. use pyth::state;
  59. use wormhole::external_address;
  60. use std::account;
  61. use std::vector;
  62. #[test_only]
  63. fun setup_test(
  64. stale_price_threshold: u64,
  65. governance_emitter_chain_id: u64,
  66. governance_emitter_address: vector<u8>,
  67. update_fee: u64,
  68. ) {
  69. // Initialize wormhole with a large message collection fee
  70. wormhole::wormhole_test::setup(100000);
  71. // Deploy and initialize a test instance of the Pyth contract
  72. let deployer = account::create_signer_with_capability(&
  73. account::create_test_signer_cap(@0x277fa055b6a73c42c0662d5236c65c864ccbf2d4abd21f174a30c8b786eab84b));
  74. let (_pyth, signer_capability) = account::create_resource_account(&deployer, b"pyth");
  75. pyth::init_test(signer_capability, stale_price_threshold, governance_emitter_chain_id, governance_emitter_address, vector[], update_fee);
  76. }
  77. #[test]
  78. #[expected_failure(abort_code = 6)]
  79. fun test_execute_governance_instruction_invalid_vaa() {
  80. setup_test(50, 24, x"f06413c0148c78916554f134dcd17a7c8029a3a2bda475a4a1182305c53078bf", 100);
  81. let vaa_bytes = x"6c436741b108";
  82. governance::execute_governance_instruction(vaa_bytes);
  83. }
  84. #[test]
  85. #[expected_failure(abort_code = 65550)]
  86. fun test_execute_governance_instruction_invalid_data_source() {
  87. setup_test(100, 50, x"f06413c0148c78916554f134dcd17a7c8029a3a2bda475a4a1182305c53078bf", 100);
  88. // A VAA with:
  89. // - Emitter chain ID of 20
  90. // - Emitter address of 0xed67fcc21620d1bf9f69db61ea65ea36ae2df4f86c8e1b9503f0da287c24ed41
  91. let vaa_bytes = x"0100000000010066359039306c20c8e6d0047ca82aef1b3d1059a3196ab9b21ee9eb8d8438c4e06c3f181d86687cf52f8c4a167ce8af6a5dbadad22253a4016dc28a25f181a37301527e4f9b000000010014ed67fcc21620d1bf9f69db61ea65ea36ae2df4f86c8e1b9503f0da287c24ed410000000000000000005054474eb01087a85361f738f19454e66664d3c9";
  92. governance::execute_governance_instruction(vaa_bytes);
  93. }
  94. #[test]
  95. #[expected_failure(abort_code = 65551)]
  96. fun test_execute_governance_instruction_invalid_sequence_number_0() {
  97. setup_test(100, 50, x"f06413c0148c78916554f134dcd17a7c8029a3a2bda475a4a1182305c53078bf", 100);
  98. assert!(state::get_last_executed_governance_sequence() == 0, 1);
  99. // A VAA with:
  100. // - Emitter chain ID 50
  101. // - Emitter address 0xf06413c0148c78916554f134dcd17a7c8029a3a2bda475a4a1182305c53078bf
  102. // - Sequence number 0
  103. let vaa_bytes = x"010000000001004d7facf7151ada96a35a3f099843c5f13bd0e0a6cbf50722d4e456d370bbce8641ecc16450979d4c403888f9f08d5975503d810732dc95575880d2a4c64d40aa01527e4f9b000000010032f06413c0148c78916554f134dcd17a7c8029a3a2bda475a4a1182305c53078bf0000000000000000005054474eb01087a85361f738f19454e66664d3c9";
  104. governance::execute_governance_instruction(vaa_bytes);
  105. }
  106. #[test]
  107. #[expected_failure(abort_code = 65556)]
  108. fun test_execute_governance_instruction_invalid_instruction_magic() {
  109. setup_test(100, 50, x"f06413c0148c78916554f134dcd17a7c8029a3a2bda475a4a1182305c53078bf", 100);
  110. // A VAA with:
  111. // - Emitter chain ID 50
  112. // - Emitter address 0xf06413c0148c78916554f134dcd17a7c8029a3a2bda475a4a1182305c53078bf
  113. // - Sequence number 1
  114. // - A payload with the value x"5054474eb01087a85361f738f19454e66664d3c9", so the magic number will be 5054474e
  115. let vaa_bytes = x"01000000000100583334c65aff30780bf7f2ac783398a2a985e3e4873264e46c3cddfdfb2eaa484365e9f4a3ecc14d059ac1cf0a7b6a58075749ad17a3bfd4153d8f45b9084a3500527e4f9b000000010032f06413c0148c78916554f134dcd17a7c8029a3a2bda475a4a1182305c53078bf0000000000000001005054474eb01087a85361f738f19454e66664d3c9";
  116. governance::execute_governance_instruction(vaa_bytes);
  117. }
  118. #[test]
  119. #[expected_failure(abort_code = 65548)]
  120. fun test_execute_governance_instruction_invalid_module() {
  121. setup_test(100, 50, x"f06413c0148c78916554f134dcd17a7c8029a3a2bda475a4a1182305c53078bf", 100);
  122. // A VAA with:
  123. // - Emitter chain ID 50
  124. // - Emitter address 0xf06413c0148c78916554f134dcd17a7c8029a3a2bda475a4a1182305c53078bf
  125. // - Sequence number 1
  126. // - A payload representing a governance instruction with:
  127. // - Module number 2
  128. let vaa_bytes = x"010000000001001d9fd73b3fb0fc522eae5eb5bd40ddf68941894495d7cec8c8efdbf462e48715171b5c6d4bbca0c1e3843b3c28d0ca6f3f76874624b5595a3a2cbfdb3907b62501527e4f9b000000010032f06413c0148c78916554f134dcd17a7c8029a3a2bda475a4a1182305c53078bf0000000000000001005054474d0202001003001793a28e2e5b4cb88f69e96fb29a8287a88b23f0e99f5502f81744e904da8e3b4d000c9a4066ce1fa26da1c102a3e268abd3ca58e3b3c25f250e6ad9a3525066fbf8b00012f7778ca023d5cbe37449bab2faa2a133fe02b056c2c25605950320df08750f35";
  129. governance::execute_governance_instruction(vaa_bytes);
  130. }
  131. #[test]
  132. #[expected_failure(abort_code = 65549)]
  133. fun test_execute_governance_instruction_invalid_target_chain() {
  134. setup_test(100, 50, x"f06413c0148c78916554f134dcd17a7c8029a3a2bda475a4a1182305c53078bf", 100);
  135. // A VAA with:
  136. // - Emitter chain ID 50
  137. // - Emitter address 0xf06413c0148c78916554f134dcd17a7c8029a3a2bda475a4a1182305c53078bf
  138. // - Sequence number 1
  139. // - A payload representing a governance instruction with:
  140. // - Module number 1
  141. // - Target chain 17 != wormhole test chain ID 22
  142. let vaa_bytes = x"010000000001001ed81e10f8e52e6a7daeca12bf0859c14e8dabed737eaed9a1f8227190a9d11c48d58856713243c5d7de08ed49de4aa1efe7c5e6020c11056802e2d702aa4b2e00527e4f9b000000010032f06413c0148c78916554f134dcd17a7c8029a3a2bda475a4a1182305c53078bf0000000000000001005054474d0102001103001793a28e2e5b4cb88f69e96fb29a8287a88b23f0e99f5502f81744e904da8e3b4d000c9a4066ce1fa26da1c102a3e268abd3ca58e3b3c25f250e6ad9a3525066fbf8b00012f7778ca023d5cbe37449bab2faa2a133fe02b056c2c25605950320df08750f35";
  143. governance::execute_governance_instruction(vaa_bytes);
  144. }
  145. #[test]
  146. #[expected_failure(abort_code = 65552)]
  147. fun test_execute_governance_instruction_invalid_action() {
  148. setup_test(100, 50, x"f06413c0148c78916554f134dcd17a7c8029a3a2bda475a4a1182305c53078bf", 100);
  149. // A VAA with:
  150. // - Emitter chain ID 50
  151. // - Emitter address 0xf06413c0148c78916554f134dcd17a7c8029a3a2bda475a4a1182305c53078bf
  152. // - Sequence number 1
  153. // - A payload representing a governance instruction with:
  154. // - Module number 1
  155. // - Target chain 22
  156. // - Action 19 (invalid)
  157. let vaa_bytes = x"0100000000010049fdadd56a51e8bd30637dbf9fc79a154a80c96479ce223061ec1f5094f2908715d6c691e5f06068873daa79c87fc25deb62555db7c520468d05aa2437fda97201527e4f9b000000010032f06413c0148c78916554f134dcd17a7c8029a3a2bda475a4a1182305c53078bf0000000000000001005054474d0113001603001793a28e2e5b4cb88f69e96fb29a8287a88b23f0e99f5502f81744e904da8e3b4d000c9a4066ce1fa26da1c102a3e268abd3ca58e3b3c25f250e6ad9a3525066fbf8b00012f7778ca023d5cbe37449bab2faa2a133fe02b056c2c25605950320df08750f35";
  158. governance::execute_governance_instruction(vaa_bytes);
  159. }
  160. #[test]
  161. fun test_execute_governance_instruction_upgrade_contract() {
  162. setup_test(100, 50, x"f06413c0148c78916554f134dcd17a7c8029a3a2bda475a4a1182305c53078bf", 100);
  163. // A VAA with:
  164. // - Emitter chain ID 50
  165. // - Emitter address 0xf06413c0148c78916554f134dcd17a7c8029a3a2bda475a4a1182305c53078bf
  166. // - Sequence number 5
  167. // - A payload representing a governance instruction with:
  168. // - Module number 1
  169. // - Target chain 22
  170. // - AuthorizeContractUpgrade {
  171. // hash: 0xa381a47fd0e97f34c71ef491c82208f58cd0080e784c697e65966d2a25d20d56,
  172. // }
  173. let vaa_bytes = x"010000000001002242229aec7d320a437cb241672dacfbc34c9155c02f60cd806bbfcd69bb7ba667fc069e372ae0443a7f3e08eaad61930b00784faeb2b72ecf5d1b0f0fa486a101527e4f9b000000010032f06413c0148c78916554f134dcd17a7c8029a3a2bda475a4a1182305c53078bf0000000000000005005054474d01000016a381a47fd0e97f34c71ef491c82208f58cd0080e784c697e65966d2a25d20d56";
  174. governance::execute_governance_instruction(vaa_bytes);
  175. assert!(state::get_last_executed_governance_sequence() == 5, 1);
  176. assert!(state::get_contract_upgrade_authorized_hash() ==
  177. contract_upgrade_hash::from_byte_vec(x"a381a47fd0e97f34c71ef491c82208f58cd0080e784c697e65966d2a25d20d56"), 1);
  178. }
  179. #[test]
  180. #[expected_failure(abort_code = 65558)]
  181. fun test_execute_governance_instruction_upgrade_contract_chain_id_zero() {
  182. setup_test(100, 50, x"f06413c0148c78916554f134dcd17a7c8029a3a2bda475a4a1182305c53078bf", 100);
  183. // A VAA with:
  184. // - Emitter chain ID 50
  185. // - Emitter address 0xf06413c0148c78916554f134dcd17a7c8029a3a2bda475a4a1182305c53078bf
  186. // - Sequence number 5
  187. // - A payload representing a governance instruction with:
  188. // - Module number 1
  189. // - Target chain 0
  190. // - AuthorizeContractUpgrade {
  191. // hash: 0xa381a47fd0e97f34c71ef491c82208f58cd0080e784c697e65966d2a25d20d56,
  192. // }
  193. let vaa_bytes = x"01000000000100303c10020c537205ed0322b7ec9d9b296f4e3e12e39ebde985ed4ef4c8f5565256cfc6f90800c4683dba62b577cc994e2ca9135d32b955040b94718cdcb5527600527e4f9b000000010032f06413c0148c78916554f134dcd17a7c8029a3a2bda475a4a1182305c53078bf0000000000000005005054474d01000000a381a47fd0e97f34c71ef491c82208f58cd0080e784c697e65966d2a25d20d56";
  194. governance::execute_governance_instruction(vaa_bytes);
  195. assert!(state::get_last_executed_governance_sequence() == 5, 1);
  196. assert!(state::get_contract_upgrade_authorized_hash() ==
  197. contract_upgrade_hash::from_byte_vec(x"a381a47fd0e97f34c71ef491c82208f58cd0080e784c697e65966d2a25d20d56"), 1);
  198. }
  199. #[test]
  200. fun test_execute_governance_instruction_set_governance_data_source() {
  201. let initial_governance_emitter_chain_id = 50;
  202. let initial_governance_emitter_address = x"f06413c0148c78916554f134dcd17a7c8029a3a2bda475a4a1182305c53078bf";
  203. setup_test(100, initial_governance_emitter_chain_id, initial_governance_emitter_address, 100);
  204. state::set_last_executed_governance_sequence(25);
  205. let initial_governance_data_source = data_source::new(initial_governance_emitter_chain_id, external_address::from_bytes(initial_governance_emitter_address));
  206. assert!(state::is_valid_governance_data_source(initial_governance_data_source), 1);
  207. // A VAA with:
  208. // - Emitter chain ID 50
  209. // - Emitter address 0xf06413c0148c78916554f134dcd17a7c8029a3a2bda475a4a1182305c53078bf
  210. // - Sequence number 27
  211. // - A payload representing a governance instruction with:
  212. // - Module number 1
  213. // - Target chain 22
  214. // - SetGovernanceDataSource {
  215. // emitter_chain_id: 9,
  216. // emitter_address: 0x625bae57728a368652a0ab0a89808de5fffa61d3312f1a27c3e200e99b1f3058,
  217. // initial_sequence: 10,
  218. // }
  219. let vaa_bytes = x"01000000000100e8ce9e581b64ab7fbe168a0d9f86d1d2220e57947fb0c75174849838104d5fdf39ceb52ca44706bbe2817e6d33dd84ff92dc13ffe024578722178602ffd1775b01527e4f9b000000010032f06413c0148c78916554f134dcd17a7c8029a3a2bda475a4a1182305c53078bf000000000000001b005054474d010100160009625bae57728a368652a0ab0a89808de5fffa61d3312f1a27c3e200e99b1f3058000000000000000a";
  220. governance::execute_governance_instruction(vaa_bytes);
  221. // Check that the governance data source and sequence number has been updated correctly
  222. assert!(state::get_last_executed_governance_sequence() == 10, 1);
  223. assert!(!state::is_valid_governance_data_source(initial_governance_data_source), 1);
  224. assert!(state::is_valid_governance_data_source(
  225. data_source::new(9, external_address::from_bytes(x"625bae57728a368652a0ab0a89808de5fffa61d3312f1a27c3e200e99b1f3058")
  226. )), 1);
  227. // Check that we can successfully execute a governance VAA from the new governance data source
  228. // A VAA with:
  229. // - Emitter chain ID 9
  230. // - Emitter address 0x625bae57728a368652a0ab0a89808de5fffa61d3312f1a27c3e200e99b1f3058
  231. // - Sequence number 15
  232. // - A payload representing a governance instruction with:
  233. // - Module number 1
  234. // - Target chain 22
  235. // - SetStalePriceThreshold {
  236. // threshold: 900
  237. // }
  238. let second_vaa_bytes = x"010000000001008df31b9853fe9f49b1949b66e10795595c37dfc5dede5ea15c1d136cc104843e2048488dfffc3d791ac1c11c71cdb7b73f250b00eb6977cd80e943542142c3a500527e4f9b000000010009625bae57728a368652a0ab0a89808de5fffa61d3312f1a27c3e200e99b1f3058000000000000000f005054474d010400160000000000000384";
  239. governance::execute_governance_instruction(second_vaa_bytes);
  240. assert!(state::get_last_executed_governance_sequence() == 15, 1);
  241. assert!(state::get_stale_price_threshold_secs() == 900, 1);
  242. }
  243. #[test]
  244. #[expected_failure(abort_code = 65550)]
  245. fun test_execute_governance_instruction_set_governance_data_source_old_source_invalid() {
  246. let initial_governance_emitter_chain_id = 50;
  247. let initial_governance_emitter_address = x"f06413c0148c78916554f134dcd17a7c8029a3a2bda475a4a1182305c53078bf";
  248. setup_test(100, initial_governance_emitter_chain_id, initial_governance_emitter_address, 100);
  249. state::set_last_executed_governance_sequence(25);
  250. let initial_governance_data_source = data_source::new(initial_governance_emitter_chain_id, external_address::from_bytes(initial_governance_emitter_address));
  251. assert!(state::is_valid_governance_data_source(initial_governance_data_source), 1);
  252. // A VAA with:
  253. // - Emitter chain ID 50
  254. // - Emitter address x"f06413c0148c78916554f134dcd17a7c8029a3a2bda475a4a1182305c53078bf"
  255. // - Sequence number 27
  256. // - A payload representing a governance instruction with:
  257. // - Module number 1
  258. // - Target chain 22
  259. // - SetGovernanceDataSource {
  260. // emitter_chain_id: 9,
  261. // emitter_address: 0x625bae57728a368652a0ab0a89808de5fffa61d3312f1a27c3e200e99b1f3058,
  262. // initial_sequence: 10,
  263. // }
  264. let vaa_bytes = x"01000000000100e8ce9e581b64ab7fbe168a0d9f86d1d2220e57947fb0c75174849838104d5fdf39ceb52ca44706bbe2817e6d33dd84ff92dc13ffe024578722178602ffd1775b01527e4f9b000000010032f06413c0148c78916554f134dcd17a7c8029a3a2bda475a4a1182305c53078bf000000000000001b005054474d010100160009625bae57728a368652a0ab0a89808de5fffa61d3312f1a27c3e200e99b1f3058000000000000000a";
  265. governance::execute_governance_instruction(vaa_bytes);
  266. // Check that the governance data source and sequence number has been updated correctly
  267. assert!(state::get_last_executed_governance_sequence() == 10, 1);
  268. assert!(!state::is_valid_governance_data_source(initial_governance_data_source), 1);
  269. assert!(state::is_valid_governance_data_source(
  270. data_source::new(9, external_address::from_bytes(x"625bae57728a368652a0ab0a89808de5fffa61d3312f1a27c3e200e99b1f3058")
  271. )), 1);
  272. // Check that we can not longer execute governance VAA's from the old governance data source
  273. // A VAA with:
  274. // - Emitter chain ID 50
  275. // - Emitter address 0xf06413c0148c78916554f134dcd17a7c8029a3a2bda475a4a1182305c53078bf
  276. // - Sequence number 30
  277. let second_vaa_bytes = x"010000000001000e2670b14d716673d44f3766684a42a55c49feaf9a38acffb6971ec66fee2a211e7260413ccf4e3de608111dc0b92a131e8c9b8f5e83e6c36d5fc2228e46eb2d01527e4f9b000000010032f06413c0148c78916554f134dcd17a7c8029a3a2bda475a4a1182305c53078bf000000000000001e005054474d010400160000000000000384";
  278. governance::execute_governance_instruction(second_vaa_bytes);
  279. }
  280. #[test]
  281. fun test_execute_governance_instruction_set_update_fee() {
  282. let initial_update_fee = 325;
  283. setup_test(100, 50, x"f06413c0148c78916554f134dcd17a7c8029a3a2bda475a4a1182305c53078bf", initial_update_fee);
  284. assert!(state::get_base_update_fee() == initial_update_fee, 1);
  285. // A VAA with:
  286. // - Emitter chain ID 50
  287. // - Emitter address 0xf06413c0148c78916554f134dcd17a7c8029a3a2bda475a4a1182305c53078bf
  288. // - Sequence number 1
  289. // - A payload representing a governance instruction with:
  290. // - Module number 1
  291. // - Target chain 22
  292. // - SetUpdateFee {
  293. // mantissa: 17,
  294. // exponent: 3,
  295. // }
  296. let vaa_bytes = x"010000000001009f843a3359e75940cad00eaec50a1ac075aca3248634576437cfd53d95c2e29859a3a1902a3ef3e0529b434cf63ce96b21e4e6c05204ba62a446371aa132174000527e4f9b000000010032f06413c0148c78916554f134dcd17a7c8029a3a2bda475a4a1182305c53078bf0000000000000001005054474d0103001600000000000000110000000000000003";
  297. governance::execute_governance_instruction(vaa_bytes);
  298. assert!(state::get_last_executed_governance_sequence() == 1, 1);
  299. let expected = 17000;
  300. assert!(state::get_base_update_fee() == expected, 1);
  301. }
  302. #[test]
  303. fun test_execute_governance_instruction_set_stale_price_threshold() {
  304. let initial_stale_price_threshold = 125;
  305. setup_test(initial_stale_price_threshold, 50, x"f06413c0148c78916554f134dcd17a7c8029a3a2bda475a4a1182305c53078bf", 100);
  306. assert!(state::get_stale_price_threshold_secs() == initial_stale_price_threshold, 1);
  307. // A VAA with:
  308. // - Emitter chain ID 50
  309. // - Emitter address 0xf06413c0148c78916554f134dcd17a7c8029a3a2bda475a4a1182305c53078bf
  310. // - Sequence number 1
  311. // - A payload representing a governance instruction with:
  312. // - Module number 1
  313. // - Target chain 22
  314. // - SetStalePriceThreshold {
  315. // threshold: 756
  316. // }
  317. let vaa_bytes = x"01000000000100e863ad8824f2c2a1695c6b028fa36c5f654b5f3e8d33712032aa3a2197329f3e2c59fc86cc026e6c68608d9e13982f2a22098bbc877ae2b106f6659ea320850a00527e4f9b000000010032f06413c0148c78916554f134dcd17a7c8029a3a2bda475a4a1182305c53078bf0000000000000001005054474d0104001600000000000002f4";
  318. governance::execute_governance_instruction(vaa_bytes);
  319. assert!(state::get_last_executed_governance_sequence() == 1, 1);
  320. assert!(state::get_stale_price_threshold_secs() == 756, 1);
  321. }
  322. #[test]
  323. fun test_execute_governance_instruction_set_data_sources() {
  324. setup_test(100, 50, x"f06413c0148c78916554f134dcd17a7c8029a3a2bda475a4a1182305c53078bf", 100);
  325. // A VAA with:
  326. // - Emitter chain ID 50
  327. // - Emitter address 0xf06413c0148c78916554f134dcd17a7c8029a3a2bda475a4a1182305c53078bf
  328. // - Sequence number 1
  329. // - A payload representing a governance instruction with:
  330. // - Module number 1
  331. // - Target chain 22
  332. // - SetDataSources {
  333. // sources: [
  334. // (23, 0x93a28e2e5b4cb88f69e96fb29a8287a88b23f0e99f5502f81744e904da8e3b4d),
  335. // (12, 0x9a4066ce1fa26da1c102a3e268abd3ca58e3b3c25f250e6ad9a3525066fbf8b0),
  336. // (18, 0xf7778ca023d5cbe37449bab2faa2a133fe02b056c2c25605950320df08750f35)
  337. // ]
  338. // }
  339. let vaa_bytes = x"01000000000100d6c0b6dad041866337af989010c88e4230c77ea16aea579a6422aa44a4f0f57e5d0948e40606445bc0753554ffa0c2f9d5c45abf3d3b16a0158957f01cddb6d600527e4f9b000000010032f06413c0148c78916554f134dcd17a7c8029a3a2bda475a4a1182305c53078bf0000000000000001005054474d0102001603001793a28e2e5b4cb88f69e96fb29a8287a88b23f0e99f5502f81744e904da8e3b4d000c9a4066ce1fa26da1c102a3e268abd3ca58e3b3c25f250e6ad9a3525066fbf8b00012f7778ca023d5cbe37449bab2faa2a133fe02b056c2c25605950320df08750f35";
  340. // Set an initial data source
  341. let initial_data_source = data_source::new(11, external_address::from_bytes(x"4eeb85a8ee41cccd0becb6428cb8f12fd0790b3ad9e378f4dfd81014bc42db1d"));
  342. state::set_data_sources(vector<DataSource>[initial_data_source]);
  343. // Execute the VAA
  344. governance::execute_governance_instruction(vaa_bytes);
  345. assert!(state::get_last_executed_governance_sequence() == 1, 1);
  346. // Check that the data sources have been set correctly
  347. let expected = vector<DataSource>[
  348. data_source::new(23, external_address::from_bytes(x"93a28e2e5b4cb88f69e96fb29a8287a88b23f0e99f5502f81744e904da8e3b4d")),
  349. data_source::new(12, external_address::from_bytes(x"9a4066ce1fa26da1c102a3e268abd3ca58e3b3c25f250e6ad9a3525066fbf8b0")),
  350. data_source::new(18, external_address::from_bytes(x"f7778ca023d5cbe37449bab2faa2a133fe02b056c2c25605950320df08750f35")),
  351. ];
  352. assert!(!state::is_valid_data_source(initial_data_source), 1);
  353. while(vector::is_empty(&expected)) {
  354. assert!(state::is_valid_data_source(vector::pop_back(&mut expected)), 1);
  355. }
  356. }
  357. }