class ChefSpec::Stubs::DataBagItemStub

def initialize(bag, id, &block)
  @bag   = bag.to_s
  @id    = id
  @block = block
end

def signature
  if @block
    "stub_data_bag_item(#{@bag.inspect}, #{@id.inspect}) { # Ruby code }"
  else
    "stub_data_bag_item(#{@bag.inspect}, #{@id.inspect}).and_return(#{@value})"
  end
end