@@ -5,8 +5,7 @@ use crate::{
5
5
manage_neuron:: { SetDissolveTimestamp , StartDissolving } ,
6
6
VotingPowerEconomics ,
7
7
} ,
8
- temporarily_disable_private_neuron_enforcement, temporarily_disable_voting_power_adjustment,
9
- temporarily_enable_private_neuron_enforcement, temporarily_enable_voting_power_adjustment,
8
+ temporarily_disable_voting_power_adjustment, temporarily_enable_voting_power_adjustment,
10
9
} ;
11
10
use ic_cdk:: println;
12
11
@@ -437,63 +436,37 @@ fn test_visibility_when_converting_neuron_to_neuron_info_and_neuron_proto() {
437
436
) ;
438
437
439
438
// Case 1: visibility is explicitly set.
440
- for set_enforcement in [
441
- temporarily_enable_private_neuron_enforcement,
442
- temporarily_disable_private_neuron_enforcement,
443
- ] {
444
- let _restore_on_drop = set_enforcement ( ) ;
445
-
446
- for visibility in [ Visibility :: Public , Visibility :: Private ] {
447
- let neuron = builder. clone ( ) . with_visibility ( Some ( visibility) ) . build ( ) ;
439
+ for visibility in [ Visibility :: Public , Visibility :: Private ] {
440
+ let neuron = builder. clone ( ) . with_visibility ( Some ( visibility) ) . build ( ) ;
448
441
449
- assert_eq ! ( neuron. visibility( ) , Some ( visibility) , ) ;
450
-
451
- let neuron_info = neuron. get_neuron_info (
452
- & VotingPowerEconomics :: DEFAULT ,
453
- timestamp_seconds,
454
- principal_id,
455
- ) ;
456
- assert_eq ! ( neuron_info. visibility, Some ( visibility as i32 ) , ) ;
457
-
458
- let neuron_proto = neuron. into_proto ( & VotingPowerEconomics :: DEFAULT , timestamp_seconds) ;
459
- assert_eq ! ( neuron_proto. visibility, Some ( visibility as i32 ) , ) ;
460
- }
461
- }
462
-
463
- // Case 2: visibility is not set.
464
- let neuron = builder. clone ( ) . build ( ) ;
465
- {
466
- let _restore_on_drop = temporarily_disable_private_neuron_enforcement ( ) ;
467
-
468
- assert_eq ! ( neuron. visibility( ) , None , ) ;
442
+ assert_eq ! ( neuron. visibility( ) , Some ( visibility) , ) ;
469
443
470
444
let neuron_info = neuron. get_neuron_info (
471
445
& VotingPowerEconomics :: DEFAULT ,
472
446
timestamp_seconds,
473
447
principal_id,
474
448
) ;
475
- assert_eq ! ( neuron_info. visibility, None , ) ;
449
+ assert_eq ! ( neuron_info. visibility, Some ( visibility as i32 ) , ) ;
476
450
477
- let neuron_proto = neuron
478
- . clone ( )
479
- . into_proto ( & VotingPowerEconomics :: DEFAULT , timestamp_seconds) ;
480
- assert_eq ! ( neuron_proto. visibility, None , ) ;
451
+ let neuron_proto = neuron. into_proto ( & VotingPowerEconomics :: DEFAULT , timestamp_seconds) ;
452
+ assert_eq ! ( neuron_proto. visibility, Some ( visibility as i32 ) , ) ;
481
453
}
482
- {
483
- let _restore_on_drop = temporarily_enable_private_neuron_enforcement ( ) ;
484
454
485
- assert_eq ! ( neuron . visibility( ) , Some ( Visibility :: Private ) , ) ;
455
+ // Case 2: visibility is not set.
486
456
487
- let neuron_info = neuron. get_neuron_info (
488
- & VotingPowerEconomics :: DEFAULT ,
489
- timestamp_seconds,
490
- principal_id,
491
- ) ;
492
- assert_eq ! ( neuron_info. visibility, Some ( Visibility :: Private as i32 ) , ) ;
457
+ let neuron = builder. clone ( ) . build ( ) ;
493
458
494
- let neuron_proto = neuron. into_proto ( & VotingPowerEconomics :: DEFAULT , timestamp_seconds) ;
495
- assert_eq ! ( neuron_proto. visibility, Some ( Visibility :: Private as i32 ) , ) ;
496
- }
459
+ assert_eq ! ( neuron. visibility( ) , Some ( Visibility :: Private ) , ) ;
460
+
461
+ let neuron_info = neuron. get_neuron_info (
462
+ & VotingPowerEconomics :: DEFAULT ,
463
+ timestamp_seconds,
464
+ principal_id,
465
+ ) ;
466
+ assert_eq ! ( neuron_info. visibility, Some ( Visibility :: Private as i32 ) , ) ;
467
+
468
+ let neuron_proto = neuron. into_proto ( & VotingPowerEconomics :: DEFAULT , timestamp_seconds) ;
469
+ assert_eq ! ( neuron_proto. visibility, Some ( Visibility :: Private as i32 ) , ) ;
497
470
498
471
// Case 3: Known neurons are always public.
499
472
let neuron = builder
@@ -502,26 +475,20 @@ fn test_visibility_when_converting_neuron_to_neuron_info_and_neuron_proto() {
502
475
description : Some ( "neuron description" . to_string ( ) ) ,
503
476
} ) )
504
477
. build ( ) ;
505
- for set_enforcement in [
506
- temporarily_enable_private_neuron_enforcement,
507
- temporarily_disable_private_neuron_enforcement,
508
- ] {
509
- let _restore_on_drop = set_enforcement ( ) ;
510
478
511
- assert_eq ! ( neuron. visibility( ) , Some ( Visibility :: Public ) , ) ;
479
+ assert_eq ! ( neuron. visibility( ) , Some ( Visibility :: Public ) , ) ;
512
480
513
- let neuron_info = neuron. get_neuron_info (
514
- & VotingPowerEconomics :: DEFAULT ,
515
- timestamp_seconds,
516
- principal_id,
517
- ) ;
518
- assert_eq ! ( neuron_info. visibility, Some ( Visibility :: Public as i32 ) , ) ;
481
+ let neuron_info = neuron. get_neuron_info (
482
+ & VotingPowerEconomics :: DEFAULT ,
483
+ timestamp_seconds,
484
+ principal_id,
485
+ ) ;
486
+ assert_eq ! ( neuron_info. visibility, Some ( Visibility :: Public as i32 ) , ) ;
519
487
520
- let neuron_proto = neuron
521
- . clone ( )
522
- . into_proto ( & VotingPowerEconomics :: DEFAULT , timestamp_seconds) ;
523
- assert_eq ! ( neuron_proto. visibility, Some ( Visibility :: Public as i32 ) , ) ;
524
- }
488
+ let neuron_proto = neuron
489
+ . clone ( )
490
+ . into_proto ( & VotingPowerEconomics :: DEFAULT , timestamp_seconds) ;
491
+ assert_eq ! ( neuron_proto. visibility, Some ( Visibility :: Public as i32 ) , ) ;
525
492
}
526
493
527
494
#[ test]
0 commit comments