Retain Policy Freshness after Schedule Activation for Quotas

2026-10-31 21:02:01 +03:00 · 2026-10-19 23:11:25 +03:00
parent 6fdc2a5ba4
commit 0416de3c27
9 changed files with 506 additions and 15 deletions
@@ -0,0 +1,104 @@
+use super::*;
+
+#[tokio::test]
+async fn quota_period_b11_patch_preserves_active_policy_against_delayed_source() {
+    for accepted in [false, true] {
+        let fixture = Fixture::new();
+        // The startup account has no API-creation fence.
+        patch_user(
+            "admin",
+            patch(serde_json::json!({"data_quota_bytes":1024,"quota_period":"daily"})),
+            None,
+            &fixture.shared,
+        )
+        .await
+        .unwrap();
+        fixture.publish().await;
+        let stale = ProxyConfig::parse_source(&fixture.shared.config_path).unwrap();
+        let authority = fixture.shared.proxy_shared.user_admission();
+        let epoch = authority.epoch();
+        patch_user(
+            "admin",
+            patch(serde_json::json!({"quota_period":"monthly"})),
+            None,
+            &fixture.shared,
+        )
+        .await
+        .unwrap();
+        assert_eq!(authority.epoch(), epoch);
+        assert!(authority.has_fresh_quota_schedules());
+        fixture.publish().await;
+        assert!(!authority.has_fresh_quota_schedules());
+        let account = fixture.store.current_handle("admin").unwrap();
+        account.charge(700);
+        let checkpoint = account.schedule_checkpoint();
+        let stale = Arc::new(stale.prepare().await.unwrap().config);
+        let gate = fixture.shared.quota_state.lock().await;
+        fixture
+            .shared
+            .proxy_shared
+            .publish_quota_config(1, &stale, accepted);
+        drop(gate);
+        fixture
+            .shared
+            .quota_state
+            .process_schedules(crate::quota_state::now_epoch_secs(), false)
+            .await
+            .unwrap();
+        assert_eq!(account.schedule_checkpoint(), checkpoint);
+        assert_eq!(fixture.store.used("admin"), 700);
+        assert_eq!(
+            fixture
+                .shared
+                .quota_state
+                .schedule_status("admin", Some(UserQuotaPeriod::Monthly))
+                .quota_schedule_state,
+            "active"
+        );
+        assert_eq!(
+            fixture.store.current_handle("admin").unwrap().incarnation(),
+            account.incarnation()
+        );
+        fixture.publish().await;
+        assert!(!authority.has_fresh_quota_schedules());
+        fixture.finish().await;
+    }
+}
+
+#[tokio::test]
+async fn quota_period_b11_noop_patch_keeps_active_status_without_publication() {
+    let fixture = Fixture::new();
+    patch_user(
+        "admin",
+        patch(serde_json::json!({"data_quota_bytes":1024,"quota_period":"monthly"})),
+        None,
+        &fixture.shared,
+    )
+    .await
+    .unwrap();
+    fixture.publish().await;
+    let account = fixture.store.current_handle("admin").unwrap();
+    account.charge(700);
+    let checkpoint = account.schedule_checkpoint();
+    patch_user("admin", patch(serde_json::json!({})), None, &fixture.shared)
+        .await
+        .unwrap();
+    assert_eq!(account.schedule_checkpoint(), checkpoint);
+    assert_eq!(fixture.store.used("admin"), 700);
+    assert!(
+        !fixture
+            .shared
+            .proxy_shared
+            .user_admission()
+            .has_fresh_quota_schedules()
+    );
+    assert_eq!(
+        fixture
+            .shared
+            .quota_state
+            .schedule_status("admin", Some(UserQuotaPeriod::Monthly))
+            .quota_schedule_state,
+        "active"
+    );
+    fixture.finish().await;
+}