Class: CouchPotato::RSpec::StubView::ViewStub

Inherits:
Object
  • Object
show all
Includes:
RSpec::Mocks::ExampleMethods
Defined in:
lib/couch_potato/rspec/stub_db.rb

Instance Method Summary collapse

Constructor Details

#initialize(clazz, view, db) ⇒ ViewStub

Returns a new instance of ViewStub.



11
12
13
14
15
 
# File 'lib/couch_potato/rspec/stub_db.rb', line 11

def initialize(clazz, view, db)
  @clazz = clazz
  @view = view
  @db = db
end

Instance Method Details

#and_return(return_value) ⇒ Object 



23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
 
# File 'lib/couch_potato/rspec/stub_db.rb', line 23

def and_return(return_value)
  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)
  allow(@db).to receive(:view).with(view_stub).and_return(return_value)
  return unless return_value.respond_to?(:first)

  allow(@db).to receive(:first).with(view_stub).and_return(return_value.first)
  allow(@db)
    .to receive(:view_in_batches) do |_view, batch_size: CouchPotato::Database.default_batch_size, &block|
      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)
  end
end

#with(*args, &block) ⇒ Object 



17
18
19
20
21
 
# File 'lib/couch_potato/rspec/stub_db.rb', line 17

def with(*args, &block)
  @args = args
  and_return(block.call) if block
  self
end