rack-berater 0.0.1 → 0.0.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: a1fad5e57124897f72b0fac11bf460c6b3dc279d057f56f22127ec32274d782d
-  data.tar.gz: aa7e1c738a450ff6fb72aa9b92e1211ed51c544f801a2ffef475a2a89805db5c
+  metadata.gz: 2a5812fba3c2523be30274bcc192d67ba7cebbb1d2b52dcc0ca403d69c39630c
+  data.tar.gz: 46152472d4af26a86e506a9150ff9c56743490707b630ecddc94beb0e63f7448
 SHA512:
-  metadata.gz: 0ec05fa53c35aea22a9c29e93b5d509c8a161364542641ada1af1c421320d671cd634dc8ab74abc597f38a056fb3d8638b69786cf82e26357b0ace12bf483ee9
-  data.tar.gz: c9f1c61d05ac9f297b88fad401fdf635cd553a15d2040a2fc7e45282fe2776e00967b8e7181cec5a621c5cb69771676cf7d70a34ee63e6be2b74dd7784b7b7d2
+  metadata.gz: '079407a6fd8d63c849809c972ef15739de09bd490c1e3a4046ae047dbba8b36cb7e5f0606c17b5e5a6a74ce8ec9fa95f61139afc3937a5c30643fe85ff59e91d'
+  data.tar.gz: 7d6ede65719ed13181c25aca35a48c941ee9c4777caa4f56eede0552b4acdc39d2989069659217327a0b3bc494d0c178b1ee1faa21e42f5c70dd4067f4fb077e

data/lib/rack/berater/handler.rb

@@ -5,6 +5,9 @@ module Rack
         @app = app
         @options = {
           status_code: options.fetch(:status_code, 429),
+          headers: {
+            Rack::CONTENT_TYPE => "text/plain",
+          }.update(options.fetch(:headers, {})),
           body: options.fetch(:body, true),
         }
       end
@@ -13,6 +16,7 @@ module Rack
         @app.call(env)
       rescue ::Berater::Overloaded => e
         code = @options[:status_code]
+
         body = case @options[:body]
           when true
             Rack::Utils::HTTP_STATUS_CODES[code]
@@ -26,9 +30,11 @@ module Rack
             raise ArgumentError, "invalid :body option: #{@options[:body]}"
           end
 
+        headers = body ? @options[:headers] : {}
+
         [
           code,
-          {},
+          headers,
           [ body ].compact,
         ]
       end

data/lib/rack/berater/version.rb

@@ -1,5 +1,5 @@
 module Rack
   module Berater
-    VERSION = "0.0.1"
+    VERSION = "0.0.2"
   end
 end

data/spec/handler_spec.rb

@@ -11,8 +11,20 @@ describe Rack::Berater::Handler do
   let(:response) { get "/" }
 
   shared_examples "works nominally" do
-    it { expect(response.status).to eq 200 }
-    it { expect(response.body).to eq "OK" }
+    it "has the correct status code" do
+      expect(response.status).to eq 200
+    end
+
+    it "has the correct headers" do
+      expect(response.headers).to eq({
+        "Content-Type" => "text/plain",
+        "Content-Length" => "2",
+      })
+    end
+
+    it "has the correct body" do
+      expect(response.body).to eq "OK"
+    end
   end
 
   context "without Handler" do
@@ -61,6 +73,10 @@ describe Rack::Berater::Handler do
       it "should not send a body" do
         expect(response.body).to be_empty
       end
+
+      it "should not send the Content-Type header" do
+        expect(response.headers.keys).not_to include(Rack::CONTENT_TYPE)
+      end
     end
 
     context "with body nil" do
@@ -94,5 +110,17 @@ describe Rack::Berater::Handler do
         response
       end
     end
+
+    context "with custom headers" do
+      let(:options) { { headers: { Rack::CACHE_CONTROL => "no-cache" } } }
+
+      it "should contain the default headers" do
+        expect(response.headers.keys).to include(Rack::CONTENT_TYPE)
+      end
+
+      it "should also contain custom header" do
+        expect(response.headers).to include(options[:headers])
+      end
+    end
   end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: rack-berater
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.2
 platform: ruby
 authors:
 - Daniel Pepper
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2021-09-12 00:00:00.000000000 Z
+date: 2021-09-13 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: berater
@@ -108,7 +108,7 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         version: '0'
-description: "..."
+description: limit incoming requests
 email:
 executables: []
 extensions: []