|
@@ -1,6 +1,6 @@
|
|
|
#[cfg(all(test, feature = "bridgeauth"))]
|
|
|
use super::proto::{
|
|
|
- check_blockage, issue_invite,
|
|
|
+ blockage_migration, check_blockage, issue_invite,
|
|
|
level_up::{self, LEVEL_INTERVAL},
|
|
|
migration, open_invite,
|
|
|
trust_promotion::{self, UNTRUSTED_INTERVAL},
|
|
@@ -303,6 +303,30 @@ impl TestHarness {
|
|
|
assert!(mig_cred.is_ok(), "Handle response should succeed");
|
|
|
mig_cred.unwrap()
|
|
|
}
|
|
|
+
|
|
|
+ pub fn blockage_migration(
|
|
|
+ &mut self,
|
|
|
+ rng: &mut (impl CryptoRng + RngCore),
|
|
|
+ cred: Lox,
|
|
|
+ mig_cred: Migration,
|
|
|
+ ) -> Lox {
|
|
|
+ let migration_request =
|
|
|
+ blockage_migration::request(rng, cred, mig_cred, self.ba.lox_pub.clone());
|
|
|
+ assert!(
|
|
|
+ migration_request.is_ok(),
|
|
|
+ "Migration request should succeed"
|
|
|
+ );
|
|
|
+ let (mig_request, mig_client_state) = migration_request.unwrap();
|
|
|
+ let migration_response = self.ba.handle_blockage_migration(mig_request);
|
|
|
+ assert!(
|
|
|
+ migration_response.is_ok(),
|
|
|
+ "Migration response from server should succeed"
|
|
|
+ );
|
|
|
+ let response = migration_response.unwrap();
|
|
|
+ let new_cred = blockage_migration::handle_response(mig_client_state, response);
|
|
|
+ assert!(new_cred.is_ok(), "Handle response should succeed");
|
|
|
+ new_cred.unwrap()
|
|
|
+ }
|
|
|
}
|
|
|
|
|
|
/// Create a random BridgeLine for testing
|