rails-response-dumper 3.0.1 → 4.1.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 33844c7b9a9480388327762b8a297daab7e464c47933b8559325283aa29fc5b3
-  data.tar.gz: 6b8984033d0040101a4577e9a8c9aedfb3bba372510c03655ddbc402eb07e016
+  metadata.gz: b36497e5dbc803078e07b1303d4df1ec11d23deb80881a3ab9ec76fe4ed532bf
+  data.tar.gz: 93643ba27e9f5c10d41595466d1864a194fb9409dd55f046bb9b01fdc1c327fb
 SHA512:
-  metadata.gz: 4c4aa80837f7f3ec7d70b5e245692e50ac8d612499deba1d269dc54800bae589aa9143ee33ad2f665b6a647b1a741bb77ffb83acc9aa1b12169150fcd543a796
-  data.tar.gz: 033d6714b3e5febebbf5aa063e7a538bda9ae101a4cbef7b196a446c567c5f4b4341bd6dca328f2ba084e8e3a378462b9c8caa909b111545e165de656db703fc
+  metadata.gz: c2ecfb86d177c8564eaec6f131397fd3ec3724895b1eb689e0a526b532055e638f36d908942b424efe727bb0646327be7948a064c5305121de7bd27bd6f0d652
+  data.tar.gz: 43aae543d827a60d995f5ff28db8308ed511b0d2f3a099d9694a7bbb4f5e21652a000bef0b753b10677c0f24518f62c4358f8e1e20d38ffffbc7cdf271a03cf1

data/lib/rails_response_dumper/defined.rb

@@ -6,6 +6,7 @@ require_relative 'dump_block'
 module RailsResponseDumper
   class Defined
     attr_accessor :name, :klass
+    attr_reader :before_block, :after_block
 
     delegate :include, to: :klass
 
@@ -22,8 +23,20 @@ module RailsResponseDumper
       self.class.dumpers << self
     end
 
-    def dump(name, status_code: :ok, &block)
-      blocks << DumpBlock.new(name, Rack::Utils::SYMBOL_TO_STATUS_CODE[status_code], block)
+    def dump(name, status_codes: [:ok], &block)
+      blocks << DumpBlock.new(
+        name,
+        status_codes.map { |status_code| Rack::Utils::SYMBOL_TO_STATUS_CODE[status_code] },
+        block
+      )
+    end
+
+    def before(&block)
+      @before_block = block
+    end
+
+    def after(&block)
+      @after_block = block
     end
 
     def blocks

data/lib/rails_response_dumper/dump_block.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module RailsResponseDumper
-  DumpBlock = Struct.new('DumpBlock', :name, :expected_status_code, :block)
+  DumpBlock = Struct.new('DumpBlock', :name, :expected_status_codes, :block)
 end

data/lib/rails_response_dumper/runner.rb

@@ -20,28 +20,45 @@ module RailsResponseDumper
           dumper.mock_setup
           begin
             rollback_after do
-              dumper.instance_eval(&dump_block.block)
+              dumper.instance_eval(&defined.before_block) if defined.before_block
+              begin
+                dumper.instance_eval(&dump_block.block)
+              ensure
+                dumper.instance_eval(&defined.after_block) if defined.after_block
+              end
             end
           ensure
             dumper.mock_teardown
           end
 
+          unless dumper.responses.count == dump_block.expected_status_codes.count
+            raise <<~ERROR.squish
+              #{defined.name}.#{dump_block.name} received
+              #{dumper.responses.count} responses
+              (expected #{dump_block.expected_status_codes.count})
+            ERROR
+          end
+
           klass_path = defined.name.underscore
           dumper_dir = "#{dumps_dir}/#{klass_path}/#{dump_block.name}"
           FileUtils.mkdir_p dumper_dir
 
           dumper.responses.each_with_index do |response, index|
-            unless response.status == dump_block.expected_status_code
+            unless response.status == dump_block.expected_status_codes[index]
               raise <<~ERROR.squish
                 #{defined.name}.#{dump_block.name} has unexpected status
                 code #{response.status} #{response.status_message}
-                (expected #{dump_block.expected_status_code})
+                (expected #{dump_block.expected_status_codes[index]})
               ERROR
             end
 
-            mime = response.content_type.split(/ *; */).first
-            extension = MIME::Types[mime].first.preferred_extension
-            File.write("#{dumper_dir}/#{index}.#{extension}", response.body)
+            if response.content_type
+              mime = response.content_type.split(/ *; */).first
+              extension = ".#{MIME::Types[mime].first.preferred_extension}"
+            else
+              extension = ''
+            end
+            File.write("#{dumper_dir}/#{index}#{extension}", response.body)
           end
         end
       end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: rails-response-dumper
 version: !ruby/object:Gem::Version
-  version: 3.0.1
+  version: 4.1.0
 platform: ruby
 authors:
 - Pioneer Valley Books
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2022-03-29 00:00:00.000000000 Z
+date: 2022-10-18 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: mime-types