feat: utility to generate random strings
This commit is contained in:
parent
5d56cf132c
commit
5d636892a2
2 changed files with 73 additions and 0 deletions
5
src/utils/mod.rs
Normal file
5
src/utils/mod.rs
Normal file
|
@ -0,0 +1,5 @@
|
|||
// SPDX-FileCopyrightText: 2024 Aravinth Manivannan <realaravinth@batsense.net>
|
||||
//
|
||||
// SPDX-License-Identifier: AGPL-3.0-or-later
|
||||
|
||||
pub mod random_string;
|
68
src/utils/random_string.rs
Normal file
68
src/utils/random_string.rs
Normal file
|
@ -0,0 +1,68 @@
|
|||
// SPDX-FileCopyrightText: 2024 Aravinth Manivannan <realaravinth@batsense.net>
|
||||
//
|
||||
// SPDX-License-Identifier: AGPL-3.0-or-later
|
||||
|
||||
use std::sync::Arc;
|
||||
|
||||
use actix_web::web;
|
||||
use mockall::predicate::*;
|
||||
use mockall::*;
|
||||
|
||||
pub type WebGenerateRandomStringInterface = web::Data<Arc<dyn GenerateRandomStringInterface>>;
|
||||
|
||||
#[automock]
|
||||
pub trait GenerateRandomStringInterface: Send + Sync {
|
||||
fn get_random(&self, len: usize) -> String;
|
||||
}
|
||||
|
||||
pub struct GenerateRandomString;
|
||||
impl GenerateRandomStringInterface for GenerateRandomString {
|
||||
fn get_random(&self, len: usize) -> String {
|
||||
use std::iter;
|
||||
|
||||
use rand::{distributions::Alphanumeric, rngs::ThreadRng, thread_rng, Rng};
|
||||
|
||||
let mut rng: ThreadRng = thread_rng();
|
||||
|
||||
iter::repeat(())
|
||||
.map(|()| rng.sample(Alphanumeric))
|
||||
.map(char::from)
|
||||
.take(len)
|
||||
.collect::<String>()
|
||||
}
|
||||
}
|
||||
|
||||
impl GenerateRandomString {
|
||||
pub fn inject() -> impl FnOnce(&mut web::ServiceConfig) {
|
||||
let g = WebGenerateRandomStringInterface::new(Arc::new(GenerateRandomString));
|
||||
let f = move |cfg: &mut web::ServiceConfig| {
|
||||
cfg.app_data(g);
|
||||
};
|
||||
|
||||
Box::new(f)
|
||||
}
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
pub mod tests {
|
||||
use super::*;
|
||||
|
||||
use std::sync::Arc;
|
||||
|
||||
pub fn mock_generate_random_string(
|
||||
times: Option<usize>,
|
||||
returning: String,
|
||||
) -> Arc<dyn GenerateRandomStringInterface> {
|
||||
let mut m = MockGenerateRandomStringInterface::new();
|
||||
|
||||
if let Some(times) = times {
|
||||
m.expect_get_random()
|
||||
.times(times)
|
||||
.return_const(returning.to_string());
|
||||
} else {
|
||||
m.expect_get_random().return_const(returning.to_string());
|
||||
}
|
||||
|
||||
Arc::new(m)
|
||||
}
|
||||
}
|
Loading…
Reference in a new issue