Skip to content

Commit

Permalink
Merge pull request openwsn-berkeley#250 from geonnave/explicit-creden…
Browse files Browse the repository at this point in the history
…tial-fetching-authz-test

Tests (authz): add mock function to fetch cred_i
  • Loading branch information
geonnave authored Mar 20, 2024
2 parents 6802356 + 28ae744 commit a9ed47e
Showing 1 changed file with 13 additions and 1 deletion.
14 changes: 13 additions & 1 deletion lib/src/lib.rs
Original file line number Diff line number Diff line change
Expand Up @@ -646,6 +646,14 @@ mod test_authz {
let cred_i = CredentialRPK::new(CRED_I.try_into().unwrap()).unwrap();
let cred_r = CredentialRPK::new(CRED_R.try_into().unwrap()).unwrap();

let mock_fetch_cred_i = |id_cred_i: CredentialRPK| -> Result<CredentialRPK, EDHOCError> {
if id_cred_i.kid == cred_i.kid {
Ok(cred_i.clone())
} else {
Err(EDHOCError::UnexpectedCredential)
}
};

// ==== initialize edhoc ====
let mut initiator = EdhocInitiator::new(default_crypto());
let responder = EdhocResponder::new(default_crypto(), R, cred_r);
Expand Down Expand Up @@ -708,7 +716,11 @@ mod test_authz {
.unwrap();

let (responder, id_cred_i, _ead_3) = responder.parse_message_3(&message_3).unwrap();
let valid_cred_i = credential_check_or_fetch(Some(cred_i), id_cred_i).unwrap();
let valid_cred_i = if id_cred_i.reference_only() {
mock_fetch_cred_i(id_cred_i).unwrap()
} else {
id_cred_i
};
let (mut _responder, r_prk_out) = responder.verify_message_3(valid_cred_i).unwrap();

// check that prk_out is equal at initiator and responder side
Expand Down

0 comments on commit a9ed47e

Please sign in to comment.