style: cargo fmt --all
This commit is contained in:
@@ -1,9 +1,9 @@
|
||||
use std::sync::Arc;
|
||||
use application::storage::{CheckQuotaHandler, CheckQuotaQuery};
|
||||
use application::testing::{InMemoryQuotaRepository, InMemoryUsageLedgerRepository};
|
||||
use application::storage::{CheckQuotaQuery, CheckQuotaHandler};
|
||||
use domain::entities::{QuotaDefinition, TimePeriod, UsageLedgerEntry, UsageType};
|
||||
use domain::ports::UsageLedgerRepository;
|
||||
use domain::value_objects::SystemId;
|
||||
use std::sync::Arc;
|
||||
|
||||
#[tokio::test]
|
||||
async fn returns_allowed() {
|
||||
@@ -17,11 +17,14 @@ async fn returns_allowed() {
|
||||
quota_repo.save("a).await.unwrap();
|
||||
|
||||
let handler = CheckQuotaHandler::new(quota_repo, ledger_repo);
|
||||
let result = handler.execute(CheckQuotaQuery {
|
||||
user_id: user,
|
||||
usage_type: UsageType::StorageBytes,
|
||||
requested_amount: 5_000,
|
||||
}).await.unwrap();
|
||||
let result = handler
|
||||
.execute(CheckQuotaQuery {
|
||||
user_id: user,
|
||||
usage_type: UsageType::StorageBytes,
|
||||
requested_amount: 5_000,
|
||||
})
|
||||
.await
|
||||
.unwrap();
|
||||
|
||||
assert!(result.allowed);
|
||||
assert_eq!(result.limit, 10_000);
|
||||
@@ -44,11 +47,14 @@ async fn returns_denied() {
|
||||
ledger_repo.record(&entry).await.unwrap();
|
||||
|
||||
let handler = CheckQuotaHandler::new(quota_repo, ledger_repo);
|
||||
let result = handler.execute(CheckQuotaQuery {
|
||||
user_id: user,
|
||||
usage_type: UsageType::StorageBytes,
|
||||
requested_amount: 200,
|
||||
}).await.unwrap();
|
||||
let result = handler
|
||||
.execute(CheckQuotaQuery {
|
||||
user_id: user,
|
||||
usage_type: UsageType::StorageBytes,
|
||||
requested_amount: 200,
|
||||
})
|
||||
.await
|
||||
.unwrap();
|
||||
|
||||
assert!(!result.allowed);
|
||||
assert_eq!(result.current_usage, 900);
|
||||
@@ -60,11 +66,14 @@ async fn returns_unlimited_when_no_quota() {
|
||||
let ledger_repo = Arc::new(InMemoryUsageLedgerRepository::new());
|
||||
|
||||
let handler = CheckQuotaHandler::new(quota_repo, ledger_repo);
|
||||
let result = handler.execute(CheckQuotaQuery {
|
||||
user_id: SystemId::new(),
|
||||
usage_type: UsageType::StorageBytes,
|
||||
requested_amount: 999_999,
|
||||
}).await.unwrap();
|
||||
let result = handler
|
||||
.execute(CheckQuotaQuery {
|
||||
user_id: SystemId::new(),
|
||||
usage_type: UsageType::StorageBytes,
|
||||
requested_amount: 999_999,
|
||||
})
|
||||
.await
|
||||
.unwrap();
|
||||
|
||||
assert!(result.allowed);
|
||||
assert!(result.is_unlimited);
|
||||
|
||||
Reference in New Issue
Block a user