test

TabbyML · Dec 1, 2023 · 23fde50 · 23fde50
1 parent 24cb9f9
commit 23fde50
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/ee/tabby-webserver/src/service/auth.rs b/ee/tabby-webserver/src/service/auth.rs
@@ -133,7 +133,7 @@ impl AuthenticationService for DbConn {
                 return err;
             };
 
-            let Some(invitation) = self.get_invitation_by_code(invitation_code).await? else {
+            let Some(invitation) = self.get_invitation_by_code(&invitation_code).await? else {
                 return err;
             };
 

diff --git a/ee/tabby-webserver/src/service/db.rs b/ee/tabby-webserver/src/service/db.rs
@@ -249,7 +249,8 @@ impl DbConn {
         Ok(invitations)
     }
 
-    pub async fn get_invitation_by_code(&self, code: String) -> Result<Option<Invitation>> {
+    pub async fn get_invitation_by_code(&self, code: &str) -> Result<Option<Invitation>> {
+        let code = code.to_owned();
         let token = self
             .conn
             .call(|conn| {
@@ -378,6 +379,8 @@ mod tests {
         assert_eq!(1, invitations.len());
 
         assert!(Uuid::parse_str(&invitations[0].code).is_ok());
+        let invitation = conn.get_invitation_by_code(&invitations[0].code).await.ok().flatten().unwrap();
+        assert_eq!(invitation.id, invitations[0].id);
 
         conn.delete_invitation(invitations[0].id).await.unwrap();