Skip to content

Commit e115d9b

Browse files
committed
fmt
1 parent fd44659 commit e115d9b

File tree

1 file changed

+67
-42
lines changed
  • pallets/admin-utils/src/tests

1 file changed

+67
-42
lines changed

pallets/admin-utils/src/tests/mod.rs

Lines changed: 67 additions & 42 deletions
Original file line numberDiff line numberDiff line change
@@ -1469,58 +1469,83 @@ fn test_sudo_root_sets_subnet_moving_alpha() {
14691469

14701470
#[test]
14711471
fn test_sets_a_lower_value_clears_small_nominations() {
1472-
new_test_ext().execute_with(|| {
1473-
let hotkey: U256 = U256::from(3);
1474-
let owner_coldkey: U256 = U256::from(1);
1475-
let staker_coldkey: U256 = U256::from(2);
1476-
1477-
let initial_nominator_min_required_stake = 10u64;
1478-
let nominator_min_required_stake_0 = 5u64;
1479-
let nominator_min_required_stake_1 = 20u64;
1480-
1481-
assert!(nominator_min_required_stake_0 < nominator_min_required_stake_1);
1482-
assert!(nominator_min_required_stake_0 < initial_nominator_min_required_stake);
1472+
new_test_ext().execute_with(|| {
1473+
let hotkey: U256 = U256::from(3);
1474+
let owner_coldkey: U256 = U256::from(1);
1475+
let staker_coldkey: U256 = U256::from(2);
14831476

1484-
let to_stake = initial_nominator_min_required_stake + 1;
1477+
let initial_nominator_min_required_stake = 10u64;
1478+
let nominator_min_required_stake_0 = 5u64;
1479+
let nominator_min_required_stake_1 = 20u64;
14851480

1486-
assert!(to_stake > initial_nominator_min_required_stake);
1487-
assert!(to_stake > nominator_min_required_stake_0); // Should stay when set
1488-
assert!(to_stake < nominator_min_required_stake_1); // Should be removed when set
1481+
assert!(nominator_min_required_stake_0 < nominator_min_required_stake_1);
1482+
assert!(nominator_min_required_stake_0 < initial_nominator_min_required_stake);
14891483

1490-
// Create network
1491-
let netuid = 2;
1492-
add_network(netuid, 10);
1484+
let to_stake = initial_nominator_min_required_stake + 1;
14931485

1494-
// Register a neuron
1495-
register_ok_neuron(netuid, hotkey, owner_coldkey, 0);
1486+
assert!(to_stake > initial_nominator_min_required_stake);
1487+
assert!(to_stake > nominator_min_required_stake_0); // Should stay when set
1488+
assert!(to_stake < nominator_min_required_stake_1); // Should be removed when set
14961489

1497-
assert_ok!(AdminUtils::sudo_set_nominator_min_required_stake(
1498-
RuntimeOrigin::root(),
1499-
initial_nominator_min_required_stake
1500-
));
1501-
assert_eq!(SubtensorModule::get_nominator_min_required_stake(), initial_nominator_min_required_stake);
1490+
// Create network
1491+
let netuid = 2;
1492+
add_network(netuid, 10);
15021493

1494+
// Register a neuron
1495+
register_ok_neuron(netuid, hotkey, owner_coldkey, 0);
15031496

1504-
// Stake to the hotkey as staker_coldkey
1505-
SubtensorModule::increase_stake_for_hotkey_and_coldkey_on_subnet(&hotkey, &staker_coldkey, netuid, to_stake);
1497+
assert_ok!(AdminUtils::sudo_set_nominator_min_required_stake(
1498+
RuntimeOrigin::root(),
1499+
initial_nominator_min_required_stake
1500+
));
1501+
assert_eq!(
1502+
SubtensorModule::get_nominator_min_required_stake(),
1503+
initial_nominator_min_required_stake
1504+
);
15061505

1506+
// Stake to the hotkey as staker_coldkey
1507+
SubtensorModule::increase_stake_for_hotkey_and_coldkey_on_subnet(
1508+
&hotkey,
1509+
&staker_coldkey,
1510+
netuid,
1511+
to_stake,
1512+
);
15071513

1508-
assert_ok!(AdminUtils::sudo_set_nominator_min_required_stake(
1509-
RuntimeOrigin::root(),
1510-
nominator_min_required_stake_0
1511-
));
1512-
assert_eq!(SubtensorModule::get_nominator_min_required_stake(), nominator_min_required_stake_0);
1514+
assert_ok!(AdminUtils::sudo_set_nominator_min_required_stake(
1515+
RuntimeOrigin::root(),
1516+
nominator_min_required_stake_0
1517+
));
1518+
assert_eq!(
1519+
SubtensorModule::get_nominator_min_required_stake(),
1520+
nominator_min_required_stake_0
1521+
);
15131522

1514-
// Check this nomination is not cleared
1515-
assert!(SubtensorModule::get_stake_for_hotkey_and_coldkey_on_subnet(&hotkey, &staker_coldkey, netuid) > 0);
1523+
// Check this nomination is not cleared
1524+
assert!(
1525+
SubtensorModule::get_stake_for_hotkey_and_coldkey_on_subnet(
1526+
&hotkey,
1527+
&staker_coldkey,
1528+
netuid
1529+
) > 0
1530+
);
15161531

1517-
assert_ok!(AdminUtils::sudo_set_nominator_min_required_stake(
1518-
RuntimeOrigin::root(),
1519-
nominator_min_required_stake_1
1520-
));
1521-
assert_eq!(SubtensorModule::get_nominator_min_required_stake(), nominator_min_required_stake_1);
1532+
assert_ok!(AdminUtils::sudo_set_nominator_min_required_stake(
1533+
RuntimeOrigin::root(),
1534+
nominator_min_required_stake_1
1535+
));
1536+
assert_eq!(
1537+
SubtensorModule::get_nominator_min_required_stake(),
1538+
nominator_min_required_stake_1
1539+
);
15221540

1523-
// Check this nomination is cleared
1524-
assert_eq!(SubtensorModule::get_stake_for_hotkey_and_coldkey_on_subnet(&hotkey, &staker_coldkey, netuid), 0);
1525-
});
1541+
// Check this nomination is cleared
1542+
assert_eq!(
1543+
SubtensorModule::get_stake_for_hotkey_and_coldkey_on_subnet(
1544+
&hotkey,
1545+
&staker_coldkey,
1546+
netuid
1547+
),
1548+
0
1549+
);
1550+
});
15261551
}

0 commit comments

Comments
 (0)