Update staking pallet tests

pallets/staking/src/mock.rs

@@ -400,11 +400,12 @@ impl entropy_shared::AttestationHandler<AccountId> for MockAttestationHandler {
         _x25519_public_key: entropy_shared::X25519PublicKey,
         quote: Vec<u8>,
         _context: QuoteContext,
-    ) -> Result<entropy_shared::BoundedVecEncodedVerifyingKey, sp_runtime::DispatchError> {
+    ) -> Result<entropy_shared::BoundedVecEncodedVerifyingKey, entropy_shared::VerifyQuoteError>
+    {
         let quote: Result<[u8; 32], _> = quote.try_into();
         match quote {
             Ok(q) if q == VALID_QUOTE => Ok([0; 33].to_vec().try_into().unwrap()),
-            Ok(q) if q == INVALID_QUOTE => Err(sp_runtime::DispatchError::Other("Invalid quote")),
+            Ok(q) if q == INVALID_QUOTE => Err(entropy_shared::VerifyQuoteError::BadQuote),
             _ => {
                 // We don't really want to verify quotes for tests in this pallet, so if we get
                 // something else we'll just accept it.

pallets/staking/src/tests.rs

@@ -218,7 +218,7 @@ fn it_doesnt_change_endpoint_with_invalid_quote() {
 
         assert_noop!(
             Staking::change_endpoint(RuntimeOrigin::signed(1), endpoint, INVALID_QUOTE.to_vec()),
-            Error::<Test>::FailedAttestationCheck
+            Error::<Test>::BadQuote
         );
     })
 }
@@ -338,7 +338,7 @@ fn it_doesnt_allow_changing_threshold_account_with_invalid_quote() {
                 NULL_ARR,
                 INVALID_QUOTE.to_vec()
             ),
-            Error::<Test>::FailedAttestationCheck
+            Error::<Test>::BadQuote
         );
     })
 }
@@ -711,7 +711,7 @@ fn it_requires_attestation_before_validate_is_succesful() {
                 joining_server_info.clone(),
                 INVALID_QUOTE.to_vec(),
             ),
-            Error::<Test>::FailedAttestationCheck
+            Error::<Test>::BadQuote
         );
 
         assert_eq!(Staking::threshold_server(bob), None);