Add a rules controller

app/services/dsn_authentication_service.rb

@@ -70,7 +70,7 @@ class DsnAuthenticationService
     raise AuthenticationError, "Invalid public_key" unless project
 
     # Verify project_id matches (supports both slug and ID)
-    project_matches = Project.find_by_project_id(project_id)
+    project_matches = Project.find_by(slug: project_id) || Project.find_by(id: project_id)
     raise AuthenticationError, "Invalid project_id" unless project_matches == project
 
     # Ensure project is enabled

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)