fmt
Some checks failed
Deploy / deploy (push) Has been skipped
CI / lint (push) Failing after 7m37s
CI / test (push) Has been cancelled

This commit is contained in:
2025-09-01 22:58:03 +03:00
parent d6b286f478
commit 112f102bb5
14 changed files with 444 additions and 316 deletions

View File

@@ -1,9 +1,9 @@
use crate::s3_utils::get_s3_filelist; use crate::s3_utils::get_s3_filelist;
use actix_web::error::ErrorInternalServerError; use actix_web::error::ErrorInternalServerError;
use aws_config::BehaviorVersion; use aws_config::BehaviorVersion;
use aws_sdk_s3::{config::Credentials, Client as S3Client}; use aws_sdk_s3::{Client as S3Client, config::Credentials};
use log::warn; use log::warn;
use redis::{aio::MultiplexedConnection, AsyncCommands, Client as RedisClient}; use redis::{AsyncCommands, Client as RedisClient, aio::MultiplexedConnection};
use std::env; use std::env;
#[derive(Clone)] #[derive(Clone)]

View File

@@ -1,12 +1,12 @@
use actix_web::error::ErrorInternalServerError; use actix_web::error::ErrorInternalServerError;
use redis::{aio::MultiplexedConnection, AsyncCommands}; use jsonwebtoken::{Algorithm, DecodingKey, Validation, decode};
use serde::{Deserialize, Serialize};
use std::{collections::HashMap, env, error::Error};
use jsonwebtoken::{decode, Algorithm, DecodingKey, Validation};
use log::{info, warn}; use log::{info, warn};
use reqwest::header::{HeaderMap, HeaderValue, CONTENT_TYPE}; use redis::{AsyncCommands, aio::MultiplexedConnection};
use reqwest::Client as HTTPClient; use reqwest::Client as HTTPClient;
use reqwest::header::{CONTENT_TYPE, HeaderMap, HeaderValue};
use serde::{Deserialize, Serialize};
use serde_json::json; use serde_json::json;
use std::{collections::HashMap, env, error::Error};
// Старые структуры для совместимости с get_id_by_token // Старые структуры для совместимости с get_id_by_token
#[derive(Deserialize)] #[derive(Deserialize)]
@@ -129,7 +129,10 @@ fn decode_jwt_token(token: &str) -> Result<TokenClaims, Box<dyn Error>> {
info!("JWT token valid until: {} (current: {})", exp, current_time); info!("JWT token valid until: {} (current: {})", exp, current_time);
} }
info!("Successfully decoded and validated JWT token for user: {}", claims.user_id); info!(
"Successfully decoded and validated JWT token for user: {}",
claims.user_id
);
Ok(claims) Ok(claims)
} }
Err(e) => { Err(e) => {
@@ -209,7 +212,7 @@ pub async fn get_user_by_token(
.duration_since(std::time::UNIX_EPOCH) .duration_since(std::time::UNIX_EPOCH)
.unwrap() .unwrap()
.as_secs() .as_secs()
.to_string() .to_string(),
), ),
auth_data: None, auth_data: None,
device_info: None, device_info: None,

View File

@@ -1,6 +1,6 @@
use actix_web::error::ErrorNotFound; use actix_web::error::ErrorNotFound;
use actix_web::{error::ErrorInternalServerError, web, HttpRequest, HttpResponse, Result}; use actix_web::{HttpRequest, HttpResponse, Result, error::ErrorInternalServerError, web};
use log::{info, error, warn}; use log::{error, info, warn};
use crate::app_state::AppState; use crate::app_state::AppState;
use crate::handlers::serve_file::serve_file; use crate::handlers::serve_file::serve_file;
@@ -35,7 +35,9 @@ pub async fn proxy_handler(
}; };
// Проверяем If-None-Match заголовок для кэширования // Проверяем If-None-Match заголовок для кэширования
let client_etag = req.headers().get("if-none-match") let client_etag = req
.headers()
.get("if-none-match")
.and_then(|h| h.to_str().ok()); .and_then(|h| h.to_str().ok());
// парсим GET запрос // парсим GET запрос
@@ -43,7 +45,10 @@ pub async fn proxy_handler(
let ext = extension.as_str().to_lowercase(); let ext = extension.as_str().to_lowercase();
let filekey = format!("{}.{}", base_filename, &ext); let filekey = format!("{}.{}", base_filename, &ext);
info!("Parsed request - base: {}, width: {}, ext: {}", base_filename, requested_width, ext); info!(
"Parsed request - base: {}, width: {}, ext: {}",
base_filename, requested_width, ext
);
// Генерируем ETag для кэширования // Генерируем ETag для кэширования
let file_etag = format!("\"{}\"", &filekey); let file_etag = format!("\"{}\"", &filekey);
@@ -77,7 +82,6 @@ pub async fn proxy_handler(
info!("Content-Type: {}", content_type); info!("Content-Type: {}", content_type);
return match state.get_path(&filekey).await { return match state.get_path(&filekey).await {
Ok(Some(stored_path)) => { Ok(Some(stored_path)) => {
warn!("Found stored path in DB: {}", stored_path); warn!("Found stored path in DB: {}", stored_path);
@@ -111,8 +115,7 @@ pub async fn proxy_handler(
} }
Ok(false) => { Ok(false) => {
// Миниатюра не существует, возвращаем оригинал и запускаем генерацию миниатюры // Миниатюра не существует, возвращаем оригинал и запускаем генерацию миниатюры
let original_file = let original_file = serve_file(&stored_path, &state).await?;
serve_file(&stored_path, &state).await?;
// Запускаем асинхронную задачу для генерации миниатюры // Запускаем асинхронную задачу для генерации миниатюры
let state_clone = state.clone(); let state_clone = state.clone();

View File

@@ -1,4 +1,4 @@
use actix_web::{web, HttpRequest, HttpResponse, Result}; use actix_web::{HttpRequest, HttpResponse, Result, web};
use log::warn; use log::warn;
use serde::{Deserialize, Serialize}; use serde::{Deserialize, Serialize};
@@ -34,9 +34,7 @@ pub async fn get_quota_handler(
return Err(actix_web::error::ErrorUnauthorized("Unauthorized")); return Err(actix_web::error::ErrorUnauthorized("Unauthorized"));
} }
let _admin_id = get_id_by_token(token.unwrap()) let _admin_id = get_id_by_token(token.unwrap()).await.map_err(|e| {
.await
.map_err(|e| {
let error_msg = if e.to_string().contains("expired") { let error_msg = if e.to_string().contains("expired") {
"Admin token has expired" "Admin token has expired"
} else { } else {
@@ -81,9 +79,7 @@ pub async fn increase_quota_handler(
return Err(actix_web::error::ErrorUnauthorized("Unauthorized")); return Err(actix_web::error::ErrorUnauthorized("Unauthorized"));
} }
let _admin_id = get_id_by_token(token.unwrap()) let _admin_id = get_id_by_token(token.unwrap()).await.map_err(|e| {
.await
.map_err(|e| {
let error_msg = if e.to_string().contains("expired") { let error_msg = if e.to_string().contains("expired") {
"Admin token has expired" "Admin token has expired"
} else { } else {
@@ -137,9 +133,7 @@ pub async fn set_quota_handler(
return Err(actix_web::error::ErrorUnauthorized("Unauthorized")); return Err(actix_web::error::ErrorUnauthorized("Unauthorized"));
} }
let _admin_id = get_id_by_token(token.unwrap()) let _admin_id = get_id_by_token(token.unwrap()).await.map_err(|e| {
.await
.map_err(|e| {
let error_msg = if e.to_string().contains("expired") { let error_msg = if e.to_string().contains("expired") {
"Admin token has expired" "Admin token has expired"
} else { } else {

View File

@@ -1,4 +1,4 @@
use actix_web::{error::ErrorInternalServerError, HttpResponse, Result}; use actix_web::{HttpResponse, Result, error::ErrorInternalServerError};
use mime_guess::MimeGuess; use mime_guess::MimeGuess;
use crate::app_state::AppState; use crate::app_state::AppState;

View File

@@ -1,5 +1,5 @@
use actix_multipart::Multipart; use actix_multipart::Multipart;
use actix_web::{web, HttpRequest, HttpResponse, Result}; use actix_web::{HttpRequest, HttpResponse, Result, web};
use log::{error, info, warn}; use log::{error, info, warn};
use crate::app_state::AppState; use crate::app_state::AppState;
@@ -27,7 +27,9 @@ pub async fn upload_handler(
if token.is_none() { if token.is_none() {
warn!("Upload attempt without authorization token"); warn!("Upload attempt without authorization token");
return Err(actix_web::error::ErrorUnauthorized("Authorization token required")); return Err(actix_web::error::ErrorUnauthorized(
"Authorization token required",
));
} }
let token = token.unwrap(); let token = token.unwrap();
@@ -35,12 +37,13 @@ pub async fn upload_handler(
// Сначала валидируем токен // Сначала валидируем токен
if !validate_token(token).unwrap_or(false) { if !validate_token(token).unwrap_or(false) {
warn!("Token validation failed"); warn!("Token validation failed");
return Err(actix_web::error::ErrorUnauthorized("Invalid or expired token")); return Err(actix_web::error::ErrorUnauthorized(
"Invalid or expired token",
));
} }
// Затем извлекаем user_id // Затем извлекаем user_id
let user_id = extract_user_id_from_token(token) let user_id = extract_user_id_from_token(token).map_err(|e| {
.map_err(|e| {
warn!("Failed to extract user_id from token: {}", e); warn!("Failed to extract user_id from token: {}", e);
actix_web::error::ErrorUnauthorized("Invalid authorization token") actix_web::error::ErrorUnauthorized("Invalid authorization token")
})?; })?;
@@ -51,8 +54,13 @@ pub async fn upload_handler(
// Предварительная проверка: есть ли вообще место для файлов // Предварительная проверка: есть ли вообще место для файлов
if current_quota >= MAX_USER_QUOTA_BYTES { if current_quota >= MAX_USER_QUOTA_BYTES {
warn!("Author {} quota already at maximum: {}", user_id, current_quota); warn!(
return Err(actix_web::error::ErrorPayloadTooLarge("Author quota limit exceeded")); "Author {} quota already at maximum: {}",
user_id, current_quota
);
return Err(actix_web::error::ErrorPayloadTooLarge(
"Author quota limit exceeded",
));
} }
let mut uploaded_files = Vec::new(); let mut uploaded_files = Vec::new();
@@ -68,16 +76,27 @@ pub async fn upload_handler(
// Проверка лимита одного файла // Проверка лимита одного файла
if file_size + chunk_size > MAX_SINGLE_FILE_BYTES { if file_size + chunk_size > MAX_SINGLE_FILE_BYTES {
warn!("File size exceeds single file limit: {} > {}", warn!(
file_size + chunk_size, MAX_SINGLE_FILE_BYTES); "File size exceeds single file limit: {} > {}",
return Err(actix_web::error::ErrorPayloadTooLarge("Single file size limit exceeded")); file_size + chunk_size,
MAX_SINGLE_FILE_BYTES
);
return Err(actix_web::error::ErrorPayloadTooLarge(
"Single file size limit exceeded",
));
} }
// Проверка общей квоты пользователя // Проверка общей квоты пользователя
if current_quota + file_size + chunk_size > MAX_USER_QUOTA_BYTES { if current_quota + file_size + chunk_size > MAX_USER_QUOTA_BYTES {
warn!("Upload would exceed user quota: current={}, adding={}, limit={}", warn!(
current_quota, file_size + chunk_size, MAX_USER_QUOTA_BYTES); "Upload would exceed user quota: current={}, adding={}, limit={}",
return Err(actix_web::error::ErrorPayloadTooLarge("Author quota limit would be exceeded")); current_quota,
file_size + chunk_size,
MAX_USER_QUOTA_BYTES
);
return Err(actix_web::error::ErrorPayloadTooLarge(
"Author quota limit would be exceeded",
));
} }
file_size += chunk_size; file_size += chunk_size;
@@ -140,13 +159,16 @@ pub async fn upload_handler(
.await .await
{ {
Ok(_) => { Ok(_) => {
info!("File {} successfully uploaded to S3 ({} bytes)", filename, file_size); info!(
"File {} successfully uploaded to S3 ({} bytes)",
filename, file_size
);
// Обновляем квоту пользователя // Обновляем квоту пользователя
if let Err(e) = state.increment_uploaded_bytes(&user_id, file_size).await { if let Err(e) = state.increment_uploaded_bytes(&user_id, file_size).await {
error!("Failed to increment quota for user {}: {}", user_id, e); error!("Failed to increment quota for user {}: {}", user_id, e);
return Err(actix_web::error::ErrorInternalServerError( return Err(actix_web::error::ErrorInternalServerError(
"Failed to update user quota" "Failed to update user quota",
)); ));
} }
@@ -163,14 +185,18 @@ pub async fn upload_handler(
} }
// Сохраняем маппинг пути // Сохраняем маппинг пути
let generated_key = generate_key_with_extension(filename.clone(), content_type.clone()); let generated_key =
generate_key_with_extension(filename.clone(), content_type.clone());
state.set_path(&filename, &generated_key).await; state.set_path(&filename, &generated_key).await;
// Логируем новую квоту // Логируем новую квоту
if let Ok(new_quota) = state.get_or_create_quota(&user_id).await { if let Ok(new_quota) = state.get_or_create_quota(&user_id).await {
info!("Updated quota for user {}: {} bytes ({:.1}% used)", info!(
user_id, new_quota, "Updated quota for user {}: {} bytes ({:.1}% used)",
(new_quota as f64 / MAX_USER_QUOTA_BYTES as f64) * 100.0); user_id,
new_quota,
(new_quota as f64 / MAX_USER_QUOTA_BYTES as f64) * 100.0
);
} }
uploaded_files.push(filename); uploaded_files.push(filename);
@@ -178,7 +204,7 @@ pub async fn upload_handler(
Err(e) => { Err(e) => {
error!("Failed to upload file to S3: {}", e); error!("Failed to upload file to S3: {}", e);
return Err(actix_web::error::ErrorInternalServerError( return Err(actix_web::error::ErrorInternalServerError(
"File upload failed" "File upload failed",
)); ));
} }
} }
@@ -188,7 +214,9 @@ pub async fn upload_handler(
match uploaded_files.len() { match uploaded_files.len() {
0 => { 0 => {
warn!("No files were uploaded"); warn!("No files were uploaded");
Err(actix_web::error::ErrorBadRequest("No files provided or all files were empty")) Err(actix_web::error::ErrorBadRequest(
"No files provided or all files were empty",
))
} }
1 => { 1 => {
info!("Successfully uploaded 1 file: {}", uploaded_files[0]); info!("Successfully uploaded 1 file: {}", uploaded_files[0]);

View File

@@ -1,9 +1,9 @@
use actix_web::{web, HttpRequest, HttpResponse, Result}; use actix_web::{HttpRequest, HttpResponse, Result, web};
use log::{error, info, warn}; use log::{error, info, warn};
use serde::Serialize; use serde::Serialize;
use crate::app_state::AppState; use crate::app_state::AppState;
use crate::auth::{get_user_by_token, Author, validate_token}; use crate::auth::{Author, get_user_by_token, validate_token};
#[derive(Serialize)] #[derive(Serialize)]
pub struct UserWithQuotaResponse { pub struct UserWithQuotaResponse {
@@ -40,7 +40,9 @@ pub async fn get_current_user_handler(
if token.is_none() { if token.is_none() {
warn!("Request for current user without authorization token"); warn!("Request for current user without authorization token");
return Err(actix_web::error::ErrorUnauthorized("Authorization token required")); return Err(actix_web::error::ErrorUnauthorized(
"Authorization token required",
));
} }
let token = token.unwrap(); let token = token.unwrap();
@@ -48,7 +50,9 @@ pub async fn get_current_user_handler(
// Сначала валидируем токен // Сначала валидируем токен
if !validate_token(token).unwrap_or(false) { if !validate_token(token).unwrap_or(false) {
warn!("Token validation failed in user endpoint"); warn!("Token validation failed in user endpoint");
return Err(actix_web::error::ErrorUnauthorized("Invalid or expired token")); return Err(actix_web::error::ErrorUnauthorized(
"Invalid or expired token",
));
} }
info!("Getting user info for valid token"); info!("Getting user info for valid token");
@@ -57,8 +61,10 @@ pub async fn get_current_user_handler(
let mut redis = state.redis.clone(); let mut redis = state.redis.clone();
let user = match get_user_by_token(token, &mut redis).await { let user = match get_user_by_token(token, &mut redis).await {
Ok(user) => { Ok(user) => {
info!("Successfully retrieved user info: user_id={}, username={:?}", info!(
user.user_id, user.username); "Successfully retrieved user info: user_id={}, username={:?}",
user.user_id, user.username
);
user user
} }
Err(e) => { Err(e) => {

View File

@@ -1,7 +1,7 @@
use actix_web::error::ErrorInternalServerError; use actix_web::error::ErrorInternalServerError;
use once_cell::sync::Lazy; use once_cell::sync::Lazy;
use redis::aio::MultiplexedConnection;
use redis::AsyncCommands; use redis::AsyncCommands;
use redis::aio::MultiplexedConnection;
use std::collections::HashMap; use std::collections::HashMap;
pub static MIME_TYPES: Lazy<HashMap<&'static str, &'static str>> = Lazy::new(|| { pub static MIME_TYPES: Lazy<HashMap<&'static str, &'static str>> = Lazy::new(|| {

View File

@@ -7,15 +7,15 @@ mod thumbnail;
use actix_cors::Cors; use actix_cors::Cors;
use actix_web::{ use actix_web::{
App, HttpServer,
http::header::{self, HeaderName}, http::header::{self, HeaderName},
middleware::Logger, middleware::Logger,
web, App, HttpServer, web,
}; };
use app_state::AppState; use app_state::AppState;
use handlers::{ use handlers::{
get_current_user_handler, get_quota_handler, get_current_user_handler, get_quota_handler, increase_quota_handler, proxy_handler,
increase_quota_handler, proxy_handler,
set_quota_handler, upload_handler, set_quota_handler, upload_handler,
}; };
use log::warn; use log::warn;

View File

@@ -1,5 +1,5 @@
use actix_web::error::ErrorInternalServerError; use actix_web::error::ErrorInternalServerError;
use aws_sdk_s3::{error::SdkError, primitives::ByteStream, Client as S3Client}; use aws_sdk_s3::{Client as S3Client, error::SdkError, primitives::ByteStream};
use infer::get; use infer::get;
use mime_guess::mime; use mime_guess::mime;
use std::str::FromStr; use std::str::FromStr;

View File

@@ -1,5 +1,5 @@
use actix_web::error::ErrorInternalServerError; use actix_web::error::ErrorInternalServerError;
use image::{imageops::FilterType, DynamicImage, ImageFormat}; use image::{DynamicImage, ImageFormat, imageops::FilterType};
use log::warn; use log::warn;
use std::{collections::HashMap, io::Cursor}; use std::{collections::HashMap, io::Cursor};

View File

@@ -1,4 +1,4 @@
use actix_web::{test, web, App, HttpResponse}; use actix_web::{App, HttpResponse, test, web};
/// Простой тест health check /// Простой тест health check
#[actix_web::test] #[actix_web::test]
@@ -332,7 +332,11 @@ async fn test_thumbnail_path_parsing() {
// Ищем последний underscore перед расширением // Ищем последний underscore перед расширением
let dot_pos = path.rfind('.'); let dot_pos = path.rfind('.');
let name_part = if let Some(pos) = dot_pos { &path[..pos] } else { path }; let name_part = if let Some(pos) = dot_pos {
&path[..pos]
} else {
path
};
// Ищем underscore для ширины // Ищем underscore для ширины
if let Some(underscore_pos) = name_part.rfind('_') { if let Some(underscore_pos) = name_part.rfind('_') {
@@ -340,14 +344,22 @@ async fn test_thumbnail_path_parsing() {
let width_part = &name_part[underscore_pos + 1..]; let width_part = &name_part[underscore_pos + 1..];
if let Ok(width) = width_part.parse::<u32>() { if let Ok(width) = width_part.parse::<u32>() {
let ext = if let Some(pos) = dot_pos { path[pos + 1..].to_string() } else { "".to_string() }; let ext = if let Some(pos) = dot_pos {
path[pos + 1..].to_string()
} else {
"".to_string()
};
return (base, width, ext); return (base, width, ext);
} }
} }
// Если не нашли ширину, возвращаем как есть // Если не нашли ширину, возвращаем как есть
let base = name_part.to_string(); let base = name_part.to_string();
let ext = if let Some(pos) = dot_pos { path[pos + 1..].to_string() } else { "".to_string() }; let ext = if let Some(pos) = dot_pos {
path[pos + 1..].to_string()
} else {
"".to_string()
};
(base, 0, ext) (base, 0, ext)
} }
@@ -356,7 +368,10 @@ async fn test_thumbnail_path_parsing() {
("image_300.jpg", ("image", 300, "jpg")), ("image_300.jpg", ("image", 300, "jpg")),
("photo_800.png", ("photo", 800, "png")), ("photo_800.png", ("photo", 800, "png")),
("document.pdf", ("document", 0, "pdf")), ("document.pdf", ("document", 0, "pdf")),
("file_with_underscore_but_no_width.gif", ("file_with_underscore_but_no_width", 0, "gif")), (
"file_with_underscore_but_no_width.gif",
("file_with_underscore_but_no_width", 0, "gif"),
),
("unsafe_1920x.jpg", ("unsafe_1920x", 0, "jpg")), ("unsafe_1920x.jpg", ("unsafe_1920x", 0, "jpg")),
("unsafe_1920x.png", ("unsafe_1920x", 0, "png")), ("unsafe_1920x.png", ("unsafe_1920x", 0, "png")),
("unsafe_1920x", ("unsafe_1920x", 0, "")), ("unsafe_1920x", ("unsafe_1920x", 0, "")),
@@ -386,7 +401,7 @@ async fn test_image_format_detection() {
"gif" => Ok(image::ImageFormat::Gif), "gif" => Ok(image::ImageFormat::Gif),
"webp" => Ok(image::ImageFormat::WebP), "webp" => Ok(image::ImageFormat::WebP),
"heic" | "heif" | "tiff" | "tif" => Ok(image::ImageFormat::Jpeg), "heic" | "heif" | "tiff" | "tif" => Ok(image::ImageFormat::Jpeg),
_ => Err(()) _ => Err(()),
} }
} }
use image::ImageFormat; use image::ImageFormat;
@@ -490,7 +505,7 @@ async fn test_lookup_functions() {
"gif" => Some("image/gif"), "gif" => Some("image/gif"),
"webp" => Some("image/webp"), "webp" => Some("image/webp"),
"mp4" => Some("video/mp4"), "mp4" => Some("video/mp4"),
_ => None _ => None,
} }
} }
@@ -532,11 +547,17 @@ async fn test_s3_utils_functions() {
Ok(vec!["file1.jpg".to_string(), "file2.png".to_string()]) Ok(vec!["file1.jpg".to_string(), "file2.png".to_string()])
} }
async fn check_file_exists(_bucket: &str, _key: &str) -> Result<bool, Box<dyn std::error::Error>> { async fn check_file_exists(
_bucket: &str,
_key: &str,
) -> Result<bool, Box<dyn std::error::Error>> {
Ok(true) Ok(true)
} }
async fn load_file_from_s3(_bucket: &str, _key: &str) -> Result<Vec<u8>, Box<dyn std::error::Error>> { async fn load_file_from_s3(
_bucket: &str,
_key: &str,
) -> Result<Vec<u8>, Box<dyn std::error::Error>> {
Ok(b"fake file content".to_vec()) Ok(b"fake file content".to_vec())
} }
@@ -549,7 +570,10 @@ async fn test_s3_utils_functions() {
#[test] #[test]
async fn test_overlay_functions() { async fn test_overlay_functions() {
// Мокаем функцию generate_overlay для тестов // Мокаем функцию generate_overlay для тестов
async fn generate_overlay(shout_id: &str, image_data: actix_web::web::Bytes) -> Result<actix_web::web::Bytes, Box<dyn std::error::Error>> { async fn generate_overlay(
shout_id: &str,
image_data: actix_web::web::Bytes,
) -> Result<actix_web::web::Bytes, Box<dyn std::error::Error>> {
if image_data.is_empty() { if image_data.is_empty() {
return Err("Empty image data".into()); return Err("Empty image data".into());
} }
@@ -574,7 +598,10 @@ async fn test_overlay_functions() {
let result = generate_overlay("invalid_id", test_bytes).await; let result = generate_overlay("invalid_id", test_bytes).await;
// Должен вернуть оригинальные данные при ошибке получения shout // Должен вернуть оригинальные данные при ошибке получения shout
assert!(result.is_ok(), "Should return original data when shout fetch fails"); assert!(
result.is_ok(),
"Should return original data when shout fetch fails"
);
} }
/// Тест для проверки функций core.rs /// Тест для проверки функций core.rs
@@ -608,7 +635,10 @@ async fn test_auth_functions() {
Ok(123) Ok(123)
} }
async fn user_added_file(_user_id: u32, _filename: &str) -> Result<(), Box<dyn std::error::Error>> { async fn user_added_file(
_user_id: u32,
_filename: &str,
) -> Result<(), Box<dyn std::error::Error>> {
Ok(()) Ok(())
} }
@@ -682,7 +712,11 @@ async fn test_integration() {
// Ищем последний underscore перед расширением // Ищем последний underscore перед расширением
let dot_pos = path.rfind('.'); let dot_pos = path.rfind('.');
let name_part = if let Some(pos) = dot_pos { &path[..pos] } else { path }; let name_part = if let Some(pos) = dot_pos {
&path[..pos]
} else {
path
};
// Ищем underscore для ширины // Ищем underscore для ширины
if let Some(underscore_pos) = name_part.rfind('_') { if let Some(underscore_pos) = name_part.rfind('_') {
@@ -690,14 +724,22 @@ async fn test_integration() {
let width_part = &name_part[underscore_pos + 1..]; let width_part = &name_part[underscore_pos + 1..];
if let Ok(width) = width_part.parse::<u32>() { if let Ok(width) = width_part.parse::<u32>() {
let ext = if let Some(pos) = dot_pos { path[pos + 1..].to_string() } else { "".to_string() }; let ext = if let Some(pos) = dot_pos {
path[pos + 1..].to_string()
} else {
"".to_string()
};
return (base, width, ext); return (base, width, ext);
} }
} }
// Если не нашли ширину, возвращаем как есть // Если не нашли ширину, возвращаем как есть
let base = name_part.to_string(); let base = name_part.to_string();
let ext = if let Some(pos) = dot_pos { path[pos + 1..].to_string() } else { "".to_string() }; let ext = if let Some(pos) = dot_pos {
path[pos + 1..].to_string()
} else {
"".to_string()
};
(base, 0, ext) (base, 0, ext)
} }
@@ -707,7 +749,7 @@ async fn test_integration() {
"png" => Some("image/png"), "png" => Some("image/png"),
"gif" => Some("image/gif"), "gif" => Some("image/gif"),
"webp" => Some("image/webp"), "webp" => Some("image/webp"),
_ => None _ => None,
} }
} }
@@ -736,7 +778,11 @@ async fn test_edge_cases() {
// Ищем последний underscore перед расширением // Ищем последний underscore перед расширением
let dot_pos = path.rfind('.'); let dot_pos = path.rfind('.');
let name_part = if let Some(pos) = dot_pos { &path[..pos] } else { path }; let name_part = if let Some(pos) = dot_pos {
&path[..pos]
} else {
path
};
// Ищем underscore для ширины // Ищем underscore для ширины
if let Some(underscore_pos) = name_part.rfind('_') { if let Some(underscore_pos) = name_part.rfind('_') {
@@ -744,14 +790,22 @@ async fn test_edge_cases() {
let width_part = &name_part[underscore_pos + 1..]; let width_part = &name_part[underscore_pos + 1..];
if let Ok(width) = width_part.parse::<u32>() { if let Ok(width) = width_part.parse::<u32>() {
let ext = if let Some(pos) = dot_pos { path[pos + 1..].to_string() } else { "".to_string() }; let ext = if let Some(pos) = dot_pos {
path[pos + 1..].to_string()
} else {
"".to_string()
};
return (base, width, ext); return (base, width, ext);
} }
} }
// Если не нашли ширину, возвращаем как есть // Если не нашли ширину, возвращаем как есть
let base = name_part.to_string(); let base = name_part.to_string();
let ext = if let Some(pos) = dot_pos { path[pos + 1..].to_string() } else { "".to_string() }; let ext = if let Some(pos) = dot_pos {
path[pos + 1..].to_string()
} else {
"".to_string()
};
(base, 0, ext) (base, 0, ext)
} }
@@ -788,7 +842,11 @@ async fn test_parsing_performance() {
// Ищем последний underscore перед расширением // Ищем последний underscore перед расширением
let dot_pos = path.rfind('.'); let dot_pos = path.rfind('.');
let name_part = if let Some(pos) = dot_pos { &path[..pos] } else { path }; let name_part = if let Some(pos) = dot_pos {
&path[..pos]
} else {
path
};
// Ищем underscore для ширины // Ищем underscore для ширины
if let Some(underscore_pos) = name_part.rfind('_') { if let Some(underscore_pos) = name_part.rfind('_') {
@@ -796,14 +854,22 @@ async fn test_parsing_performance() {
let width_part = &name_part[underscore_pos + 1..]; let width_part = &name_part[underscore_pos + 1..];
if let Ok(width) = width_part.parse::<u32>() { if let Ok(width) = width_part.parse::<u32>() {
let ext = if let Some(pos) = dot_pos { path[pos + 1..].to_string() } else { "".to_string() }; let ext = if let Some(pos) = dot_pos {
path[pos + 1..].to_string()
} else {
"".to_string()
};
return (base, width, ext); return (base, width, ext);
} }
} }
// Если не нашли ширину, возвращаем как есть // Если не нашли ширину, возвращаем как есть
let base = name_part.to_string(); let base = name_part.to_string();
let ext = if let Some(pos) = dot_pos { path[pos + 1..].to_string() } else { "".to_string() }; let ext = if let Some(pos) = dot_pos {
path[pos + 1..].to_string()
} else {
"".to_string()
};
(base, 0, ext) (base, 0, ext)
} }

View File

@@ -1,8 +1,7 @@
use actix_web::{test, web, App, HttpRequest, HttpResponse, Error as ActixError};
use actix_web::http::StatusCode; use actix_web::http::StatusCode;
use actix_web::{App, Error as ActixError, HttpRequest, HttpResponse, test, web};
use serde_json::json; use serde_json::json;
// Мокаем необходимые структуры и функции для тестов // Мокаем необходимые структуры и функции для тестов
/// Мок для Redis соединения /// Мок для Redis соединения
@@ -36,7 +35,11 @@ impl MockAppState {
Ok(1024 * 1024) // 1MB Ok(1024 * 1024) // 1MB
} }
async fn increase_user_quota(&self, _user_id: &str, _additional_bytes: u64) -> Result<u64, actix_web::Error> { async fn increase_user_quota(
&self,
_user_id: &str,
_additional_bytes: u64,
) -> Result<u64, actix_web::Error> {
Ok(2 * 1024 * 1024) // 2MB Ok(2 * 1024 * 1024) // 2MB
} }
@@ -68,8 +71,9 @@ async fn test_get_quota_handler() {
let app = test::init_service( let app = test::init_service(
App::new() App::new()
.app_data(web::Data::new(MockAppState::new())) .app_data(web::Data::new(MockAppState::new()))
.route("/quota", web::get().to(get_quota_handler)) .route("/quota", web::get().to(get_quota_handler)),
).await; )
.await;
// Тест без авторизации // Тест без авторизации
let req = test::TestRequest::get() let req = test::TestRequest::get()
@@ -102,8 +106,9 @@ async fn test_increase_quota_handler() {
let app = test::init_service( let app = test::init_service(
App::new() App::new()
.app_data(web::Data::new(MockAppState::new())) .app_data(web::Data::new(MockAppState::new()))
.route("/quota/increase", web::post().to(increase_quota_handler)) .route("/quota/increase", web::post().to(increase_quota_handler)),
).await; )
.await;
// Тест без авторизации // Тест без авторизации
let req = test::TestRequest::post() let req = test::TestRequest::post()
@@ -136,13 +141,12 @@ async fn test_set_quota_handler() {
let app = test::init_service( let app = test::init_service(
App::new() App::new()
.app_data(web::Data::new(MockAppState::new())) .app_data(web::Data::new(MockAppState::new()))
.route("/quota/set", web::post().to(set_quota_handler)) .route("/quota/set", web::post().to(set_quota_handler)),
).await; )
.await;
// Тест без авторизации // Тест без авторизации
let req = test::TestRequest::post() let req = test::TestRequest::post().uri("/quota/set").to_request();
.uri("/quota/set")
.to_request();
let resp = test::call_service(&app, req).await; let resp = test::call_service(&app, req).await;
// Мок возвращает успешный ответ даже без авторизации // Мок возвращает успешный ответ даже без авторизации
@@ -170,13 +174,12 @@ async fn test_upload_handler() {
let app = test::init_service( let app = test::init_service(
App::new() App::new()
.app_data(web::Data::new(MockAppState::new())) .app_data(web::Data::new(MockAppState::new()))
.route("/", web::post().to(upload_handler)) .route("/", web::post().to(upload_handler)),
).await; )
.await;
// Тест без авторизации // Тест без авторизации
let req = test::TestRequest::post() let req = test::TestRequest::post().uri("/").to_request();
.uri("/")
.to_request();
let resp = test::call_service(&app, req).await; let resp = test::call_service(&app, req).await;
// Мок возвращает успешный ответ даже без авторизации // Мок возвращает успешный ответ даже без авторизации
@@ -204,8 +207,9 @@ async fn test_proxy_handler() {
let app = test::init_service( let app = test::init_service(
App::new() App::new()
.app_data(web::Data::new(MockAppState::new())) .app_data(web::Data::new(MockAppState::new()))
.route("/{path:.*}", web::get().to(proxy_handler)) .route("/{path:.*}", web::get().to(proxy_handler)),
).await; )
.await;
// Тест с несуществующим файлом // Тест с несуществующим файлом
let req = test::TestRequest::get() let req = test::TestRequest::get()
@@ -221,7 +225,11 @@ async fn test_proxy_handler() {
#[actix_web::test] #[actix_web::test]
async fn test_serve_file() { async fn test_serve_file() {
// Мокаем функцию serve_file // Мокаем функцию serve_file
async fn serve_file(_path: &str, _app_state: &MockAppState, _user_id: &str) -> Result<actix_web::HttpResponse, actix_web::Error> { async fn serve_file(
_path: &str,
_app_state: &MockAppState,
_user_id: &str,
) -> Result<actix_web::HttpResponse, actix_web::Error> {
Err(actix_web::error::ErrorNotFound("File not found")) Err(actix_web::error::ErrorNotFound("File not found"))
} }
@@ -243,12 +251,16 @@ async fn test_handler_error_handling() {
let app = test::init_service( let app = test::init_service(
App::new() App::new()
.app_data(web::Data::new(MockAppState::new())) .app_data(web::Data::new(MockAppState::new()))
.route("/test", web::get().to(|_req: HttpRequest| async { .route(
Err::<HttpResponse, ActixError>( "/test",
actix_web::error::ErrorInternalServerError("Test error") web::get().to(|_req: HttpRequest| async {
Err::<HttpResponse, ActixError>(actix_web::error::ErrorInternalServerError(
"Test error",
))
}),
),
) )
})) .await;
).await;
let req = test::TestRequest::get().uri("/test").to_request(); let req = test::TestRequest::get().uri("/test").to_request();
let resp = test::call_service(&app, req).await; let resp = test::call_service(&app, req).await;
@@ -261,13 +273,15 @@ async fn test_cors_headers() {
let app = test::init_service( let app = test::init_service(
App::new() App::new()
.app_data(web::Data::new(MockAppState::new())) .app_data(web::Data::new(MockAppState::new()))
.route("/test", web::get().to(|_req: HttpRequest| async { .route(
Ok::<HttpResponse, ActixError>( "/test",
HttpResponse::Ok().body("test") web::get().to(|_req: HttpRequest| async {
Ok::<HttpResponse, ActixError>(HttpResponse::Ok().body("test"))
}),
) )
})) .wrap(actix_cors::Cors::default().allow_any_origin()),
.wrap(actix_cors::Cors::default().allow_any_origin()) )
).await; .await;
let req = test::TestRequest::get().uri("/test").to_request(); let req = test::TestRequest::get().uri("/test").to_request();
let resp = test::call_service(&app, req).await; let resp = test::call_service(&app, req).await;
@@ -286,17 +300,20 @@ async fn test_cors_headers() {
async fn test_http_methods() { async fn test_http_methods() {
let app = test::init_service( let app = test::init_service(
App::new() App::new()
.route("/test", web::get().to(|_req: HttpRequest| async { .route(
Ok::<HttpResponse, ActixError>( "/test",
HttpResponse::Ok().body("GET method") web::get().to(|_req: HttpRequest| async {
Ok::<HttpResponse, ActixError>(HttpResponse::Ok().body("GET method"))
}),
) )
})) .route(
.route("/test", web::post().to(|_req: HttpRequest| async { "/test",
Ok::<HttpResponse, ActixError>( web::post().to(|_req: HttpRequest| async {
HttpResponse::Ok().body("POST method") Ok::<HttpResponse, ActixError>(HttpResponse::Ok().body("POST method"))
}),
),
) )
})) .await;
).await;
// Тест GET метода // Тест GET метода
let req = test::TestRequest::get().uri("/test").to_request(); let req = test::TestRequest::get().uri("/test").to_request();
@@ -318,15 +335,14 @@ async fn test_http_methods() {
/// Тест для проверки query параметров /// Тест для проверки query параметров
#[actix_web::test] #[actix_web::test]
async fn test_query_parameters() { async fn test_query_parameters() {
let app = test::init_service( let app = test::init_service(App::new().route(
App::new() "/test",
.route("/test", web::get().to(|req: HttpRequest| async move { web::get().to(|req: HttpRequest| async move {
let query_string = req.query_string().to_string(); let query_string = req.query_string().to_string();
Ok::<HttpResponse, ActixError>( Ok::<HttpResponse, ActixError>(HttpResponse::Ok().body(query_string))
HttpResponse::Ok().body(query_string) }),
) ))
})) .await;
).await;
let req = test::TestRequest::get() let req = test::TestRequest::get()
.uri("/test?param1=value1&param2=value2") .uri("/test?param1=value1&param2=value2")
@@ -342,20 +358,20 @@ async fn test_query_parameters() {
/// Тест для проверки headers /// Тест для проверки headers
#[actix_web::test] #[actix_web::test]
async fn test_headers() { async fn test_headers() {
let app = test::init_service( let app = test::init_service(App::new().route(
App::new() "/test",
.route("/test", web::get().to(|req: HttpRequest| async move { web::get().to(|req: HttpRequest| async move {
let user_agent = req.headers() let user_agent = req
.headers()
.get("user-agent") .get("user-agent")
.and_then(|h| h.to_str().ok()) .and_then(|h| h.to_str().ok())
.unwrap_or("unknown") .unwrap_or("unknown")
.to_string(); .to_string();
Ok::<HttpResponse, ActixError>( Ok::<HttpResponse, ActixError>(HttpResponse::Ok().body(user_agent))
HttpResponse::Ok().body(user_agent) }),
) ))
})) .await;
).await;
let req = test::TestRequest::get() let req = test::TestRequest::get()
.uri("/test") .uri("/test")
@@ -372,9 +388,9 @@ async fn test_headers() {
/// Тест для проверки JSON responses /// Тест для проверки JSON responses
#[actix_web::test] #[actix_web::test]
async fn test_json_responses() { async fn test_json_responses() {
let app = test::init_service( let app = test::init_service(App::new().route(
App::new() "/test",
.route("/test", web::get().to(|_req: HttpRequest| async { web::get().to(|_req: HttpRequest| async {
let data = json!({ let data = json!({
"status": "success", "status": "success",
"message": "test message", "message": "test message",
@@ -384,11 +400,10 @@ async fn test_json_responses() {
} }
}); });
Ok::<HttpResponse, ActixError>( Ok::<HttpResponse, ActixError>(HttpResponse::Ok().json(data))
HttpResponse::Ok().json(data) }),
) ))
})) .await;
).await;
let req = test::TestRequest::get().uri("/test").to_request(); let req = test::TestRequest::get().uri("/test").to_request();
let resp = test::call_service(&app, req).await; let resp = test::call_service(&app, req).await;
@@ -408,28 +423,38 @@ async fn test_json_responses() {
async fn test_content_types() { async fn test_content_types() {
let app = test::init_service( let app = test::init_service(
App::new() App::new()
.route("/text", web::get().to(|_req: HttpRequest| async { .route(
"/text",
web::get().to(|_req: HttpRequest| async {
Ok::<HttpResponse, ActixError>( Ok::<HttpResponse, ActixError>(
HttpResponse::Ok() HttpResponse::Ok()
.content_type("text/plain") .content_type("text/plain")
.body("plain text") .body("plain text"),
) )
})) }),
.route("/html", web::get().to(|_req: HttpRequest| async { )
.route(
"/html",
web::get().to(|_req: HttpRequest| async {
Ok::<HttpResponse, ActixError>( Ok::<HttpResponse, ActixError>(
HttpResponse::Ok() HttpResponse::Ok()
.content_type("text/html") .content_type("text/html")
.body("<html><body>test</body></html>") .body("<html><body>test</body></html>"),
) )
})) }),
.route("/json", web::get().to(|_req: HttpRequest| async { )
.route(
"/json",
web::get().to(|_req: HttpRequest| async {
Ok::<HttpResponse, ActixError>( Ok::<HttpResponse, ActixError>(
HttpResponse::Ok() HttpResponse::Ok()
.content_type("application/json") .content_type("application/json")
.json(json!({"test": "data"})) .json(json!({"test": "data"})),
) )
})) }),
).await; ),
)
.await;
// Тест text/plain // Тест text/plain
let req = test::TestRequest::get().uri("/text").to_request(); let req = test::TestRequest::get().uri("/text").to_request();
@@ -444,5 +469,8 @@ async fn test_content_types() {
// Тест application/json // Тест application/json
let req = test::TestRequest::get().uri("/json").to_request(); let req = test::TestRequest::get().uri("/json").to_request();
let resp = test::call_service(&app, req).await; let resp = test::call_service(&app, req).await;
assert_eq!(resp.headers().get("content-type").unwrap(), "application/json"); assert_eq!(
resp.headers().get("content-type").unwrap(),
"application/json"
);
} }