From bfc43b9481bc0698f14816b4e94ee0a6d1e9963c Mon Sep 17 00:00:00 2001 From: John Reed <87283488+JohnReedV@users.noreply.github.com> Date: Wed, 15 May 2024 12:09:49 -0700 Subject: [PATCH 1/2] test dissolve_network --- pallets/subtensor/tests/root.rs | 69 +++++++++++++++++++++++++++++++++ 1 file changed, 69 insertions(+) diff --git a/pallets/subtensor/tests/root.rs b/pallets/subtensor/tests/root.rs index 02b045892..ac6bd724b 100644 --- a/pallets/subtensor/tests/root.rs +++ b/pallets/subtensor/tests/root.rs @@ -882,3 +882,72 @@ fn test_get_emission_across_entire_issuance_range() { } }); } + +#[test] +fn test_dissolve_network_ok() { + new_test_ext(1).execute_with(|| { + let netuid: u16 = 30; + let hotkey = U256::from(1); + + add_network(netuid, 0, 0); + let owner_coldkey = SubtensorModule::get_subnet_owner(netuid); + register_ok_neuron(netuid, hotkey, owner_coldkey, 3); + + assert!(SubtensorModule::if_subnet_exist(netuid)); + assert_ok!(SubtensorModule::dissolve_network(RuntimeOrigin::signed(owner_coldkey), netuid)); + assert!(!SubtensorModule::if_subnet_exist(netuid)) + + }); +} + +#[test] +fn test_dissolve_network_refund_coldkey_ok() { + new_test_ext(1).execute_with(|| { + let netuid: u16 = 30; + let hotkey = U256::from(1); + let subnet_locked_balance = 1000; + + add_network(netuid, 0, 0); + let owner_coldkey = SubtensorModule::get_subnet_owner(netuid); + register_ok_neuron(netuid, hotkey, owner_coldkey, 3); + + SubtensorModule::set_subnet_locked_balance(netuid, subnet_locked_balance); + let coldkey_balance = SubtensorModule::get_coldkey_balance(&owner_coldkey); + + assert!(SubtensorModule::if_subnet_exist(netuid)); + assert_ok!(SubtensorModule::dissolve_network(RuntimeOrigin::signed(owner_coldkey), netuid)); + assert!(!SubtensorModule::if_subnet_exist(netuid)); + + let coldkey_new_balance = SubtensorModule::get_coldkey_balance(&owner_coldkey); + + assert!(coldkey_new_balance > coldkey_balance); + assert_eq!(coldkey_new_balance, coldkey_balance + subnet_locked_balance); + }); +} + +#[test] +fn test_dissolve_network_not_owner_err() { + new_test_ext(1).execute_with(|| { + let netuid: u16 = 30; + let hotkey = U256::from(1); + let owner_coldkey = U256::from(2); + let random_coldkey = U256::from(3); + + add_network(netuid, 0, 0); + register_ok_neuron(netuid, hotkey, owner_coldkey, 3); + + assert_err!(SubtensorModule::dissolve_network(RuntimeOrigin::signed(random_coldkey), netuid), Error::::NotSubnetOwner); + + }); +} + +#[test] +fn test_dissolve_network_does_not_exist_err() { + new_test_ext(1).execute_with(|| { + let netuid: u16 = 30; + let coldkey = U256::from(2); + + assert_err!(SubtensorModule::dissolve_network(RuntimeOrigin::signed(coldkey), netuid), Error::::NetworkDoesNotExist); + + }); +} \ No newline at end of file From 71488789dfe18ce5f97993154e871d3a6c449a10 Mon Sep 17 00:00:00 2001 From: John Reed <87283488+JohnReedV@users.noreply.github.com> Date: Wed, 15 May 2024 12:11:26 -0700 Subject: [PATCH 2/2] cargo fmt --- pallets/subtensor/tests/root.rs | 25 +++++++++++++++++-------- 1 file changed, 17 insertions(+), 8 deletions(-) diff --git a/pallets/subtensor/tests/root.rs b/pallets/subtensor/tests/root.rs index ac6bd724b..31c689836 100644 --- a/pallets/subtensor/tests/root.rs +++ b/pallets/subtensor/tests/root.rs @@ -894,9 +894,11 @@ fn test_dissolve_network_ok() { register_ok_neuron(netuid, hotkey, owner_coldkey, 3); assert!(SubtensorModule::if_subnet_exist(netuid)); - assert_ok!(SubtensorModule::dissolve_network(RuntimeOrigin::signed(owner_coldkey), netuid)); + assert_ok!(SubtensorModule::dissolve_network( + RuntimeOrigin::signed(owner_coldkey), + netuid + )); assert!(!SubtensorModule::if_subnet_exist(netuid)) - }); } @@ -915,7 +917,10 @@ fn test_dissolve_network_refund_coldkey_ok() { let coldkey_balance = SubtensorModule::get_coldkey_balance(&owner_coldkey); assert!(SubtensorModule::if_subnet_exist(netuid)); - assert_ok!(SubtensorModule::dissolve_network(RuntimeOrigin::signed(owner_coldkey), netuid)); + assert_ok!(SubtensorModule::dissolve_network( + RuntimeOrigin::signed(owner_coldkey), + netuid + )); assert!(!SubtensorModule::if_subnet_exist(netuid)); let coldkey_new_balance = SubtensorModule::get_coldkey_balance(&owner_coldkey); @@ -936,8 +941,10 @@ fn test_dissolve_network_not_owner_err() { add_network(netuid, 0, 0); register_ok_neuron(netuid, hotkey, owner_coldkey, 3); - assert_err!(SubtensorModule::dissolve_network(RuntimeOrigin::signed(random_coldkey), netuid), Error::::NotSubnetOwner); - + assert_err!( + SubtensorModule::dissolve_network(RuntimeOrigin::signed(random_coldkey), netuid), + Error::::NotSubnetOwner + ); }); } @@ -947,7 +954,9 @@ fn test_dissolve_network_does_not_exist_err() { let netuid: u16 = 30; let coldkey = U256::from(2); - assert_err!(SubtensorModule::dissolve_network(RuntimeOrigin::signed(coldkey), netuid), Error::::NetworkDoesNotExist); - + assert_err!( + SubtensorModule::dissolve_network(RuntimeOrigin::signed(coldkey), netuid), + Error::::NetworkDoesNotExist + ); }); -} \ No newline at end of file +}