1 files changed, 5 insertions, 2 deletions
diff --git a/tests/support/factory_bot.rs b/tests/support/factory_bot.rs
index 969080a3..ba0d9c38 100644
--- a/tests/support/factory_bot.rs
+++ b/tests/support/factory_bot.rs
@@ -58,13 +58,16 @@ where
     f(build_channel(addr).await)
 }
 
-pub fn build_user(id: &str) -> cedar_policy::Entity {
+pub fn build_user(
+    id: &str,
+    attrs: std::collections::HashMap<String, cedar_policy::RestrictedExpression>,
+) -> cedar_policy::Entity {
     cedar_policy::Entity::new(
         cedar_policy::EntityUid::from_type_name_and_id(
             cedar_policy::EntityTypeName::from_str("User").unwrap(),
             cedar_policy::EntityId::from_str(id).unwrap(),
         ),
-        std::collections::HashMap::new(),
+        attrs,
         std::collections::HashSet::new(),
     )
     .unwrap()