@@ -164,7 +164,6 @@ fn test_serving_set_metadata_update() {
164164}
165165
166166#[ test]
167- #[ cfg( not( tarpaulin) ) ]
168167fn test_axon_serving_rate_limit_exceeded ( ) {
169168 new_test_ext ( 1 ) . execute_with ( || {
170169 let hotkey_account_id = U256 :: from ( 1 ) ;
@@ -382,7 +381,6 @@ fn test_prometheus_serving_set_metadata_update() {
382381}
383382
384383#[ test]
385- #[ cfg( not( tarpaulin) ) ]
386384fn test_prometheus_serving_rate_limit_exceeded ( ) {
387385 new_test_ext ( 1 ) . execute_with ( || {
388386 let hotkey_account_id = U256 :: from ( 1 ) ;
@@ -586,7 +584,7 @@ fn test_do_set_identity() {
586584 ) ) ;
587585
588586 // Check if identity is set correctly
589- let stored_identity = Identities :: < Test > :: get ( coldkey) . unwrap ( ) ;
587+ let stored_identity = Identities :: < Test > :: get ( coldkey) . expect ( "Identity should be set" ) ;
590588 assert_eq ! ( stored_identity. name, name) ;
591589 assert_eq ! ( stored_identity. url, url) ;
592590 assert_eq ! ( stored_identity. image, image) ;
@@ -622,7 +620,8 @@ fn test_do_set_identity() {
622620 additional. clone( )
623621 ) ) ;
624622
625- let updated_identity = Identities :: < Test > :: get ( coldkey) . unwrap ( ) ;
623+ let updated_identity =
624+ Identities :: < Test > :: get ( coldkey) . expect ( "Updated identity should be set" ) ;
626625 assert_eq ! ( updated_identity. name, new_name) ;
627626 assert_eq ! ( updated_identity. url, new_url) ;
628627
@@ -760,7 +759,7 @@ fn test_set_and_get_identity() {
760759 ) ) ;
761760
762761 // Get and verify identity
763- let stored_identity = Identities :: < Test > :: get ( coldkey) . unwrap ( ) ;
762+ let stored_identity = Identities :: < Test > :: get ( coldkey) . expect ( "Identity should be set" ) ;
764763 assert_eq ! ( stored_identity. name, name) ;
765764 assert_eq ! ( stored_identity. url, url) ;
766765 assert_eq ! ( stored_identity. image, image) ;
@@ -782,7 +781,8 @@ fn test_set_and_get_identity() {
782781 ) ) ;
783782
784783 // Get and verify updated identity
785- let updated_identity = Identities :: < Test > :: get ( coldkey) . unwrap ( ) ;
784+ let updated_identity =
785+ Identities :: < Test > :: get ( coldkey) . expect ( "Updated identity should be set" ) ;
786786 assert_eq ! ( updated_identity. name, new_name) ;
787787 assert_eq ! ( updated_identity. url, new_url) ;
788788 assert_eq ! ( updated_identity. image, image) ;
@@ -807,9 +807,7 @@ fn test_migrate_set_hotkey_identities() {
807807 > ( ) ;
808808
809809 // Assert that the migration has run
810- assert ! ( HasMigrationRun :: <Test >:: get(
811- b"fix_total_coldkey_stake_v7" . to_vec( )
812- ) ) ;
810+ assert ! ( HasMigrationRun :: <Test >:: get( b"migrate_identities" . to_vec( ) ) ) ;
813811
814812 // Verify that some identities were set
815813 // Note: This assumes that at least one valid identity was in the JSON file
0 commit comments