use domain::entities::{QuotaDefinition, TimePeriod, UsageType};
use domain::value_objects::SystemId;

#[test]
fn quota_with_rules() {
    let mut q = QuotaDefinition::new(SystemId::new());
    assert!(q.is_enforced);
    assert!(q.rules.is_empty());

    q.add_rule(UsageType::StorageBytes, 1_000_000, TimePeriod::Monthly);
    q.add_unlimited_rule(UsageType::ApiCalls);

    assert_eq!(q.rules.len(), 2);
    assert!(!q.rules[0].is_unlimited);
    assert_eq!(q.rules[0].limit_value, 1_000_000);
    assert!(q.rules[1].is_unlimited);
}