couch_potato-rspec 4.3.0 → 4.4.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: bfb9baa1278ce79d353157cf6494ea874279354b8d24f500b958566bfc4a9f59
-  data.tar.gz: '0596cfd3cfceb218b2a2bcf5cabeee7b5b2bb6959f89416bb61bc5e067e6bc0f'
+  metadata.gz: fa56480a95f50cbb7d4f9e4a2c5a16da89f17f696a6a06ffa65d45233e1c896d
+  data.tar.gz: f2583c19f0ebd21727517836353c5b28ec5b929220250880563a520299c57aac
 SHA512:
-  metadata.gz: 9517337195be2078727cf8f372de1fd5f371793ed137fb08fea26f475b7afcaca7deac3f05f720b3c46b807a40360dc83be49d7f2785e049ad71860839159bb8
-  data.tar.gz: 5bdaefe1478a853296612ad4d7d7530cb06ba6831109d623b06ea28136a03a21af23942cd5ce3925951d0f991229db329613ffc360b43b5fb185d13c30b8e53f
+  metadata.gz: 5bd117d88eae328561a20e843e19a78b0163c7b8c1e1ba4f6665944ab2422571fabc1500faa678542637a83e8e7b64b6b94ba63d4e3cc638f73de12988193662
+  data.tar.gz: 4c77c9d423f001f36b3c1e0308b266909d033a1a5b073738f55f453cb1f9edd966414cd7a79c92407ce9f8f1ef898f85a95440361fc3f34269b873db44fc4a67

data/lib/couch_potato/rspec/stub_db.rb

@@ -20,32 +20,38 @@ module CouchPotato::RSpec
         self
       end
 
-      def and_return(return_value)
+      def and_return(*return_values)
         view_stub = double("#{@clazz}.#{@view}(#{@args.try(:join, ', ')}) view")
         stub = allow(@clazz).to receive(@view)
         stub.with(*@args) if @args
         stub.and_return(view_stub)
 
-        view_return_value = if flex_view? && return_value.is_a?(Array)
-                              stub_flex_results(return_value)
-                            else
-                              return_value
-                            end
-        allow(@db).to receive(:view).with(view_stub).and_return(view_return_value)
-        return unless return_value.respond_to?(:first)
+        view_return_values = return_values.map do |return_value|
+          if flex_view? && return_value.is_a?(Array)
+            stub_flex_results(return_value)
+          else
+            return_value
+          end
+        end
+        allow(@db).to receive(:view).with(view_stub).and_return(*view_return_values)
+        return unless return_values.all? { |v| v.respond_to?(:first) }
 
-        allow(@db).to receive(:first).with(view_stub).and_return(return_value.first)
+        allow(@db).to receive(:first).with(view_stub).and_return(*return_values.map(&:first))
+        remaining_for_batches = return_values.dup
         allow(@db)
           .to receive(:view_in_batches) do |_view, batch_size: CouchPotato::Database.default_batch_size, &block|
+            return_value = remaining_for_batches.size > 1 ? remaining_for_batches.shift : remaining_for_batches.first
             batches = return_value.in_groups_of(batch_size, false)
             batches.each(&block)
           end
           .with(view_stub, any_args)
 
-        if return_value.first
-          allow(@db).to receive(:first!).with(view_stub).and_return(return_value.first)
-        else
-          allow(@db).to receive(:first!).with(view_stub).and_raise(CouchPotato::NotFound)
+        remaining_for_first = return_values.dup
+        allow(@db).to receive(:first!).with(view_stub) do
+          return_value = remaining_for_first.size > 1 ? remaining_for_first.shift : remaining_for_first.first
+          raise CouchPotato::NotFound unless return_value.first
+
+          return_value.first
         end
       end
 

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: couch_potato-rspec
 version: !ruby/object:Gem::Version
-  version: 4.3.0
+  version: 4.4.0
 platform: ruby
 authors:
 - Alexander Lang