random token

input-output-hk · Jul 9, 2020 · d0adfdc · d0adfdc
1 parent 6c5674c
commit d0adfdc
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 0 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/vit-servicing-station-tests/Cargo.toml b/vit-servicing-station-tests/Cargo.toml
@@ -25,6 +25,7 @@ askama_shared = "0.8"
 vit-servicing-station-lib = {path= "../vit-servicing-station-lib"}
 lazy_static = "1.4"
 rand = "0.7"
+rand_core = "0.5"
 
 #diesel doesn't have an way to create empty db
 #rustqlite is used for that purpose

diff --git a/vit-servicing-station-tests/src/common/data/mod.rs b/vit-servicing-station-tests/src/common/data/mod.rs
@@ -1,4 +1,6 @@
 use chrono::offset::Utc;
+use rand_core::OsRng;
+use rand_core::RngCore;
 use vit_servicing_station_lib::{
     db::models::{
         api_tokens::APITokenData,
@@ -10,6 +12,17 @@ use vit_servicing_station_lib::{
     v0::api_token::APIToken,
 };
 
+fn bytes() -> [u8; 32] {
+    let mut random_bytes: [u8; 32] = [0; 32];
+    let mut rng = OsRng;
+    rng.fill_bytes(&mut random_bytes);
+    random_bytes
+}
+
+pub fn invalid_token_hash() -> String {
+    base64::encode_config(bytes().to_vec(), base64::URL_SAFE_NO_PAD)
+}
+
 pub fn token() -> (APITokenData, String) {
     let data = b"ffffffffffffffffffffffffffffffff".to_vec();
     let token_data = APITokenData {