plain token tests

FuelLabs · May 3, 2024 · d9eb6b1 · d9eb6b1
1 parent 869f07e
commit d9eb6b1
Showing 1 changed file with 18 additions and 0 deletions.
diff --git a/src/db/api_token.rs b/src/db/api_token.rs
@@ -118,3 +118,21 @@ impl DbConn {
             .map_err(|_| DatabaseError::NotFound("API Token".to_string()))
     }
 }
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    #[test]
+    fn test_plain_token_new() {
+        let token = PlainToken::new();
+        assert!(token.0.starts_with(TOKEN_PREFIX));
+        assert_eq!(token.hash(), Sha256::digest(token.0.as_bytes()).as_slice());
+    }
+
+    #[test]
+    fn test_plain_token_from() {
+        let token = PlainToken::from("123456".to_string());
+        assert_eq!(token.hash(), Sha256::digest(token.0.as_bytes()).as_slice());
+    }
+}