Add comprehensive tests for mentor repository and authentication

- Implemented tests for creating, retrieving, updating, and deleting mentors in `mentor_repository_test.rs`.
- Added tests for user authentication, including successful login, invalid email formats, and inactive users in `auth_login_tests.rs`.
- Created a mock test environment setup in `mock_test.rs` to facilitate database operations during tests.
- Updated module structure to include new test files for mentors and authentication.
- Ensured cleanup of the database after tests to maintain isolation and prevent side effects.
This commit is contained in:
MythEclipse
2025-07-21 21:29:04 +07:00
parent e66f1f1634
commit 1a2e0c58b6
103 changed files with 7851 additions and 1509 deletions
+13
View File
@@ -64,3 +64,16 @@ pub struct RolesDetailQueryDto {
pub created_at: Option<String>,
pub updated_at: Option<String>,
}
impl Default for RolesDetailQueryDto {
fn default() -> Self {
Self {
id: Thing::from(("".to_string(), surrealdb::sql::Id::Number(0))),
name: String::new(),
permissions: Vec::new(),
is_deleted: false,
created_at: None,
updated_at: None,
}
}
}
+3 -1
View File
@@ -5,6 +5,7 @@ pub enum RolesEnum {
Admin,
User,
Staff,
Mentor, // Added Mentor role
}
impl fmt::Display for RolesEnum {
@@ -13,7 +14,8 @@ impl fmt::Display for RolesEnum {
RolesEnum::Admin => "Admin",
RolesEnum::User => "User",
RolesEnum::Staff => "Staff",
RolesEnum::Mentor => "Mentor", // Added Mentor role
};
write!(f, "{}", roles_str)
write!(f, "{roles_str}")
}
}
+67 -11
View File
@@ -7,8 +7,11 @@ use crate::{
};
use anyhow::{Result, bail};
use imphnen_utils::{DetailQueryBuilder, QueryListBuilder};
use serde_json;
use std::time::Instant;
use surrealdb::Uuid;
use surrealdb::sql::Thing;
use tracing::instrument;
pub struct RolesRepository<'a> {
state: &'a AppState,
@@ -19,10 +22,12 @@ impl<'a> RolesRepository<'a> {
Self { state }
}
#[instrument(skip(self, meta), err)]
pub async fn query_role_list(
&self,
meta: MetaRequestDto,
) -> Result<ResponseListSuccessDto<Vec<RolesListItemDto>>> {
let now = Instant::now();
let result: ResponseListSuccessDto<Vec<RolesSchema>> = QueryListBuilder::new(
&self.state.surrealdb_ws,
&ResourceEnum::Roles.to_string(),
@@ -31,6 +36,12 @@ impl<'a> RolesRepository<'a> {
.search_field("name")
.build()
.await?;
let elapsed = now.elapsed();
if std::env::var("RUST_ENV").unwrap_or_else(|_| "development".to_string())
== "development"
{
println!("Query 'query_role_list' took: {elapsed:.2?}");
}
let data = result
.data
.into_iter()
@@ -42,48 +53,71 @@ impl<'a> RolesRepository<'a> {
})
}
#[instrument(skip(self, name), err)]
pub async fn query_role_by_name(
&self,
name: String,
) -> Result<RolesDetailItemDto> {
let now = Instant::now();
let db = &self.state.surrealdb_ws;
let builder = DetailQueryBuilder::new(ResourceEnum::Roles.to_string())
.with_where("name")
.where_value(name.clone())
.with_where("name", Some(name.clone()))
.with_select_fields(vec!["*"])
.with_fetch("permissions");
let sql = builder.build();
let result: Option<RolesDetailQueryDto> = builder
.apply_bindings(db.query(sql).bind(("name", name)))
.await?
.take(0)?;
let role = match result {
Some(r) if !r.is_deleted => r,
_ => bail!("Role not found"),
};
let mut response = builder.apply_bindings(db.query(sql)).await?;
let result_vec: Vec<RolesDetailQueryDto> = response.take(0).map_err(|e| {
anyhow::anyhow!("Failed to take result from response: {:?}", e)
})?;
let role = result_vec
.into_iter()
.next()
.ok_or_else(|| anyhow::anyhow!("Role not found"))?;
let elapsed = now.elapsed();
if std::env::var("RUST_ENV").unwrap_or_else(|_| "development".to_string())
== "development"
{
println!("Query 'query_role_by_name' took: {elapsed:.2?}");
}
Ok(RolesDetailItemDto::from(&role))
}
#[instrument(skip(self, id), err)]
pub async fn query_role_by_id(&self, id: String) -> Result<RolesDetailItemDto> {
let now = Instant::now();
let db = &self.state.surrealdb_ws;
let thing_id = make_thing(&ResourceEnum::Roles.to_string(), &id);
let builder = DetailQueryBuilder::new(ResourceEnum::Roles.to_string())
.with_id(&id)
.with_select_fields(vec!["*"])
.with_fetch("permissions");
let sql = builder.build();
let sql_debug = sql.to_string(); // Move this line here
let result: Option<RolesDetailQueryDto> =
builder.apply_bindings(db.query(sql)).await?.take(0)?;
let elapsed = now.elapsed();
if std::env::var("RUST_ENV").unwrap_or_else(|_| "development".to_string())
== "development"
{
println!("Query 'query_role_by_id' took: {elapsed:.2?}");
}
let role = match result {
Some(r) if !r.is_deleted => r,
_ => bail!("Role not found"),
_ => bail!("Role not found sql: {} id: {}", sql_debug, thing_id),
};
Ok(RolesDetailItemDto::from(&role))
}
#[instrument(skip(self, payload), err)]
pub async fn query_create_role(
&self,
payload: RolesRequestCreateDto,
) -> Result<String> {
let now = Instant::now();
let db = &self.state.surrealdb_ws;
let role_id = Uuid::new_v4().to_string();
let permission_things: Vec<Thing> = payload
@@ -103,14 +137,22 @@ impl<'a> RolesRepository<'a> {
.create((&ResourceEnum::Roles.to_string(), role_id))
.content(role)
.await?;
let elapsed = now.elapsed();
if std::env::var("RUST_ENV").unwrap_or_else(|_| "development".to_string())
== "development"
{
println!("Query 'query_create_role' took: {elapsed:.2?}");
}
Ok("Role with permissions created successfully".into())
}
#[instrument(skip(self, id, data), err)]
pub async fn query_update_role(
&self,
id: String,
data: RolesRequestUpdateDto,
) -> Result<String> {
let now = Instant::now();
let db = &self.state.surrealdb_ws;
let existing = self.query_role_by_id(id.clone()).await?;
if existing.is_deleted {
@@ -119,13 +161,21 @@ impl<'a> RolesRepository<'a> {
let merged = RolesSchema::update(data, id.clone(), existing);
let record: Option<RolesSchema> =
db.update(get_id(&merged.id)?).content(merged).await?;
let elapsed = now.elapsed();
if std::env::var("RUST_ENV").unwrap_or_else(|_| "development".to_string())
== "development"
{
println!("Query 'query_update_role' took: {elapsed:.2?}");
}
match record {
Some(_) => Ok("Success update role".into()),
None => bail!("Failed to update role"),
}
}
#[instrument(skip(self, id), err)]
pub async fn query_delete_role(&self, id: String) -> Result<String> {
let now = Instant::now();
let db = &self.state.surrealdb_ws;
let role_id = make_thing(&ResourceEnum::Roles.to_string(), &id);
let role = self.query_role_by_id(role_id.id.to_raw()).await?;
@@ -137,6 +187,12 @@ impl<'a> RolesRepository<'a> {
.update(record_key)
.merge(serde_json::json!({ "is_deleted": true }))
.await?;
let elapsed = now.elapsed();
if std::env::var("RUST_ENV").unwrap_or_else(|_| "development".to_string())
== "development"
{
println!("Query 'query_delete_role' took: {elapsed:.2?}");
}
match record {
Some(_) => Ok("Success delete role".into()),
None => bail!("Failed to delete role"),