Add a rules controller

app/services/hub_load.rb

@@ -35,6 +35,20 @@ class HubLoad
     }
   end
 
+  # Test method for different load levels
+  def self.test_sampling(load_level)
+    rates = SAMPLING_RATES[load_level] || SAMPLING_RATES[:normal]
+
+    {
+      allowed_requests: rates[:allowed],
+      blocked_requests: rates[:blocked],
+      rate_limited_requests: rates[:rate_limited],
+      effective_until: next_sync_time,
+      load_level: load_level,
+      queue_depth: THRESHOLDS[load_level].first + 100
+    }
+  end
+
   # Calculate when sampling should be rechecked (next agent sync)
   def self.next_sync_time
     10.seconds.from_now.iso8601(3)