Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 2 additions & 0 deletions quicklendx-contracts/src/contract.rs
Original file line number Diff line number Diff line change
Expand Up @@ -134,6 +134,8 @@ impl QuickLendXContract {
// This is the primary anti-spam control: only vetted businesses may write
// invoice data to on-chain storage.
crate::verification::require_business_not_pending(&env, &business)?;
// Enforce per-business invoice cap.
ProtocolLimitsContract::check_invoice_limit(&env, &business)?;

let invoice_id: BytesN<32> = env
.crypto()
Expand Down
21 changes: 21 additions & 0 deletions quicklendx-contracts/src/events.rs
Original file line number Diff line number Diff line change
Expand Up @@ -652,6 +652,27 @@ pub struct ProtocolInitialized {
pub timestamp: u64,
}

// ============================================================================
// Pause Control Events

#[contractevent]
pub struct Paused {
pub admin: Address,
}

#[contractevent]
pub struct Unpaused {
pub admin: Address,
}

pub fn emit_paused(env: &Env, admin: &Address) {
Paused { admin: admin.clone() }.publish(env);
}

pub fn emit_unpaused(env: &Env, admin: &Address) {
Unpaused { admin: admin.clone() }.publish(env);
}

// ============================================================================
// Invoice Event Emitters
// ============================================================================
Expand Down
13 changes: 12 additions & 1 deletion quicklendx-contracts/src/pause.rs
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
use crate::admin::AdminStorage;
use crate::admin::AdminStorage;
use crate::errors::QuickLendXError;
use soroban_sdk::{symbol_short, Address, Env, Symbol};

Expand All @@ -24,7 +24,18 @@ impl PauseControl {
pub fn set_paused(env: &Env, admin: &Address, paused: bool) -> Result<(), QuickLendXError> {
admin.require_auth();
AdminStorage::require_admin(env, admin)?;
// Check current paused state to ensure idempotency
let current: bool = Self::is_paused(env);
if current == paused {
return Ok(());
}
Self::apply_paused(env, paused);
// Emit appropriate event based on state transition
if paused {
crate::events::emit_paused(env, admin);
} else {
crate::events::emit_unpaused(env, admin);
}
Ok(())
}

Expand Down
5 changes: 5 additions & 0 deletions quicklendx-contracts/src/protocol_limits.rs
Original file line number Diff line number Diff line change
Expand Up @@ -226,6 +226,11 @@ impl ProtocolLimitsContract {
})
}

/// Get the maximum active invoices per business.
pub fn get_max_invoices_per_business(env: Env) -> u32 {
Self::get_protocol_limits(env).max_invoices_per_business
}

/// @notice Validate invoice amount and due date against configured limits.
pub fn validate_invoice(env: Env, amount: i128, due_date: u64) -> Result<(), QuickLendXError> {
let current_time = env.ledger().timestamp();
Expand Down
32 changes: 32 additions & 0 deletions quicklendx-contracts/src/test_max_invoices_per_business.rs
Original file line number Diff line number Diff line change
Expand Up @@ -3,6 +3,9 @@ mod test_max_invoices_per_business {
use crate::errors::QuickLendXError;
use crate::protocol_limits::is_active_status;
use crate::types::InvoiceStatus;
use crate::{QuickLendXContract, QuickLendXContractClient};
use soroban_sdk::{Address, Env, String, Vec};
use crate::types::InvoiceCategory;

// Core logic test extracted from check_invoice_limit architecture
fn enforce_limit_logic(active_count: u32, limit: u32) -> Result<(), QuickLendXError> {
Expand Down Expand Up @@ -53,4 +56,33 @@ mod test_max_invoices_per_business {
assert_eq!(is_active_status(&InvoiceStatus::Cancelled), false);
assert_eq!(is_active_status(&InvoiceStatus::Refunded), false);
}

#[test]
fn test_store_invoice_respects_cap() {
let env = Env::default();
env.mock_all_auths();
let contract_id = env.register(QuickLendXContract, ());
let client = QuickLendXContractClient::new(&env, &contract_id);

let admin = Address::generate(&env);
client.set_admin(&admin);
// set low invoice cap = 2
client.set_protocol_limits(&admin, 1_000i128, 1_000i128, 0u32, 365u64, 86400u64, 2).unwrap();

let business = Address::generate(&env);
let currency = Address::generate(&env);
let due_date = env.ledger().timestamp() + 86_400;
// verify business
client.submit_kyc_application(&business, &String::from_str(&env, "biz"));
client.verify_business(&admin, &business);

// store two invoices successfully
for _ in 0..2 {
client.store_invoice(&business, 1_000i128, &currency, &due_date, &String::from_str(&env, "inv"), &InvoiceCategory::Services, &Vec::new(&env)).unwrap();
}

// third invoice should fail
let result = client.try_store_invoice(&business, 1_000i128, &currency, &due_date, &String::from_str(&env, "inv3"), &InvoiceCategory::Services, &Vec::new(&env));
assert_eq!(result, Err(Ok(QuickLendXError::MaxInvoicesPerBusinessExceeded)));
}
}
35 changes: 34 additions & 1 deletion quicklendx-contracts/src/test_pause.rs
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,7 @@ use crate::errors::QuickLendXError;
use crate::invoice::InvoiceCategory;
use crate::{QuickLendXContract, QuickLendXContractClient};
use soroban_sdk::testutils::{Address as _, Ledger, MockAuth, MockAuthInvoke};
use soroban_sdk::{token, Address, Env, IntoVal, String, Vec};
use soroban_sdk::{token, Address, Env, IntoVal, String, Vec, Symbol, xdr};

/// Standard test setup: registers contract, initializes admin, generates test addresses.
pub fn setup_contract_with_admin() -> (Env, QuickLendXContractClient<'static>, Address, Address) {
Expand Down Expand Up @@ -184,6 +184,39 @@ fn test_pause_allows_admin_rotation_and_new_admin_unpause() {
assert!(!client.is_paused());
}

#[test]
fn test_pause_idempotent_no_duplicate_events() {
let env = Env::default();
let (client, admin, _business, _investor, _currency) = setup(&env);
// First pause should emit one Paused event
client.pause(&admin);
assert!(client.is_paused());
// Count Paused events
let paused_topic = Symbol::new(&env, "paused");
let topic_xdr = xdr::ScVal::try_from_val(&env, &paused_topic).unwrap();
let count1 = env.events()
.all()
.events()
.iter()
.filter(|e| match &e.body {
soroban_sdk::xdr::ContractEventBody::V0(body) => body.topics.first() == Some(&topic_xdr),
})
.count();
assert_eq!(count1, 1);
// Second pause should not emit another event
client.pause(&admin);
let count2 = env.events()
.all()
.events()
.iter()
.filter(|e| match &e.body {
soroban_sdk::xdr::ContractEventBody::V0(body) => body.topics.first() == Some(&topic_xdr),
})
.count();
assert_eq!(count2, 1);
assert!(client.is_paused());
}

#[test]
fn test_pause_allows_emergency_withdraw_lifecycle() {
let env = Env::default();
Expand Down
Loading