test: throttle weight 0

jamiemccarthy · Dec 12, 2021 · 6ec96eb · 6ec96eb
1 parent 1c02f54
commit 6ec96eb
Showing 1 changed file with 20 additions and 0 deletions.
diff --git a/spec/acceptance/throttling_spec.rb b/spec/acceptance/throttling_spec.rb
@@ -68,6 +68,8 @@
     weight_proc = lambda do |request|
       if request.env["X-APIKey"] == "private-secret"
         3
+      elsif request.env["X-Lightweight"].present?
+	0
       else
         2
       end
@@ -80,6 +82,12 @@
 
     assert_equal 200, last_response.status
 
+    2.times do
+      get "/", {}, "REMOTE_ADDR" => "1.2.3.4", "X-Lightweight" => "1"
+
+      assert_equal 200, last_response.status
+    end
+
     get "/", {}, "REMOTE_ADDR" => "1.2.3.4"
 
     assert_equal 200, last_response.status
@@ -90,6 +98,12 @@
     assert_nil last_response.headers["Retry-After"]
     assert_equal "Retry later\n", last_response.body
 
+    get "/", {}, "REMOTE_ADDR" => "1.2.3.4", "X-Lightweight" => "1"
+
+    assert_equal 429, last_response.status
+    assert_nil last_response.headers["Retry-After"]
+    assert_equal "Retry later\n", last_response.body
+
     get "/", {}, "REMOTE_ADDR" => "5.6.7.8", "X-APIKey" => "private-secret"
 
     assert_equal 200, last_response.status
@@ -105,6 +119,12 @@
 
       assert_equal 200, last_response.status
     end
+
+    5.times do
+      get "/", {}, "REMOTE_ADDR" => "9.10.11.12", "X-Lightweight" => "1"
+
+      assert_equal 200, last_response.status
+    end
   end
 
   it "returns correct Retry-After header if enabled" do