fixup! Implement query_exactly_one

biscuit-auth · Jan 17, 2025 · 790c7de · 790c7de
1 parent 7462884
commit 790c7de
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 7 deletions.
diff --git a/biscuit-auth/src/token/authorizer.rs b/biscuit-auth/src/token/authorizer.rs
@@ -161,8 +161,7 @@ impl Authorizer {
     /// # use biscuit_auth::KeyPair;
     /// # use biscuit_auth::Biscuit;
     /// let keypair = KeyPair::new();
-    /// let mut builder = Biscuit::builder();
-    /// builder.add_fact("user(\"John Doe\", 42)");
+    /// let builder = Biscuit::builder().fact("user(\"John Doe\", 42)").unwrap();
     ///
     /// let biscuit = builder.build(&keypair).unwrap();
     ///
@@ -1087,8 +1086,7 @@ mod tests {
         use crate::Biscuit;
         use crate::KeyPair;
         let keypair = KeyPair::new();
-        let mut builder = Biscuit::builder();
-        builder.add_fact("user(\"John Doe\")").unwrap();
+        let builder = Biscuit::builder().fact("user(\"John Doe\")").unwrap();
 
         let biscuit = builder.build(&keypair).unwrap();
 
@@ -1122,9 +1120,11 @@ mod tests {
         use crate::Biscuit;
         use crate::KeyPair;
         let keypair = KeyPair::new();
-        let mut builder = Biscuit::builder();
-        builder.add_fact("user(\"John Doe\")").unwrap();
-        builder.add_fact("user(\"Jane Doe\")").unwrap();
+        let builder = Biscuit::builder()
+            .fact("user(\"John Doe\")")
+            .unwrap()
+            .fact("user(\"Jane Doe\")")
+            .unwrap();
 
         let biscuit = builder.build(&keypair).unwrap();
 

diff --git a/biscuit-capi/src/lib.rs b/biscuit-capi/src/lib.rs
@@ -98,6 +98,7 @@ pub enum ErrorKind {
     FormatSignatureInvalidSignatureGeneration,
     AlreadySealed,
     Execution,
+    UnexpectedQueryResult,
 }
 
 #[no_mangle]
@@ -175,6 +176,9 @@ pub extern "C" fn error_kind() -> ErrorKind {
                     Token::RunLimit(RunLimit::TooManyFacts) => ErrorKind::TooManyFacts,
                     Token::RunLimit(RunLimit::TooManyIterations) => ErrorKind::TooManyIterations,
                     Token::RunLimit(RunLimit::Timeout) => ErrorKind::Timeout,
+                    Token::RunLimit(RunLimit::UnexpectedQueryResult(_, _)) => {
+                        ErrorKind::UnexpectedQueryResult
+                    }
                     Token::ConversionError(_) => ErrorKind::ConversionError,
                     Token::Base64(_) => ErrorKind::FormatDeserializationError,
                     Token::Execution(_) => ErrorKind::Execution,