Skip to content

Commit fcf6a14

Browse files
committed
Small code clean-up.
1 parent ff5eafa commit fcf6a14

2 files changed

Lines changed: 6 additions & 8 deletions

File tree

src/callback.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -105,7 +105,7 @@ impl<'a> PolicyActionCbParam<'a> {
105105
// ==========================================================================
106106

107107
/// Helper trait that provides the [`as_any()`](AsAny::as_any) and [`as_mut_any()`](AsAny::as_mut_any) functions.
108-
pub trait AsAny {
108+
pub trait AsAny: Any {
109109
/// A helper function that returns the implementation as a [`&dyn Any`](std::any::Any) reference.
110110
fn as_any(&self) -> &dyn Any;
111111

tests/09_policy_test.rs

Lines changed: 5 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@ use rand_chacha::ChaChaRng;
2121
use serial_test::serial;
2222
use sha2::{Digest, Sha256};
2323
use std::{fs, path::Path};
24-
use tss2_fapi_rs::{FapiCallbacks, FapiContext, ImportData, KeyFlags, json::JsonValue};
24+
use tss2_fapi_rs::{FapiContext, ImportData, KeyFlags, json::JsonValue};
2525

2626
const KEY_FLAGS_SIGN: &[KeyFlags] = &[KeyFlags::NoDA, KeyFlags::Sign];
2727

@@ -188,9 +188,8 @@ fn test_policy_or() {
188188
};
189189

190190
// Check retrieved branches
191-
let callbacks: Box<dyn FapiCallbacks> = context.clear_callbacks().expect("Failed to clear callbacks!").expect("No callbacks!");
192-
let callbacks: &MyCallbacks = callbacks.as_any().downcast_ref().expect("Downcast has failed!");
193-
let branches = callbacks.get_branches();
191+
let callbacks = context.clear_callbacks().expect("Failed to clear callbacks!").expect("No callbacks!");
192+
let branches = callbacks.as_any().downcast_ref::<MyCallbacks>().expect("Downcast has failed!").get_branches();
194193
assert_eq!(branches.len(), 2);
195194
assert!(branches[0].eq_ignore_ascii_case("#0,PolicySignedRSA"));
196195
assert!(branches[1].eq_ignore_ascii_case("#1,PolicySignedECC"));
@@ -246,9 +245,8 @@ fn test_policy_action() {
246245
};
247246

248247
// Check retrieved actions
249-
let callbacks: Box<dyn FapiCallbacks> = context.clear_callbacks().expect("Failed to clear callbacks!").expect("No callbacks!");
250-
let callbacks: &MyCallbacks = callbacks.as_any().downcast_ref().expect("Downcast has failed!");
251-
let actions = callbacks.get_actions();
248+
let callbacks = context.clear_callbacks().expect("Failed to clear callbacks!").expect("No callbacks!");
249+
let actions = callbacks.as_any().downcast_ref::<MyCallbacks>().expect("Downcast has failed!").get_actions();
252250
assert!(!actions.is_empty());
253251
assert!(actions[0].eq_ignore_ascii_case("myaction"));
254252
});

0 commit comments

Comments
 (0)