编写rspec测试用例时如何覆盖ruby方法的局部变量？

Question

现有的Ruby代码：

def book_details
  book_count = Book.count
  while (book_count > 1)
    # Do something
  end
end

我想编写用于在book_details函数中设置book_count值的rspec测试用例。 可能吗？ 如果是，怎么办？

到目前为止，我尝试过的是：

it 'Should return 1 book details only' do
  allow(Model).to receive(book_count).and_return(1)
  # Do something
end

但这给＃提供了错误'undefined local variable or method`book_count'

Answer 1

假设在ruby代码中存在book_count = 1000，并且您想使用rspec将其重置为10，那么可以了：

allow(Book).to receive(:count).and_return(10)

以上将根据您对rspec的期望进行计数。

另外，要针对您的rspec contexts ， rspec是为methods而不是local variables编写的。 因此，请在rspec测试方法的期望值。

下面是您的method的简单rspec示例：

class Book

  def book_details
    book_count = Book.count
    while (book_count > 1)
      puts 'Book name'
    end
  end

end

describe 'Books' do

  describe '#book_details' do

    context 'when Books are less than one' do

      it 'should not return book details' do
        book_details.should eq 0
        book_details.should_not eq 'Book name'
      end

    end

    context 'when Books are more than one' do

      before do
        FactoryGirl.create(:book)
      end

      it 'should return book details' do
        book_details.should_not eq 0
        book_details.should eq 'Book name'
      end

    end

  end

end

Answer 2

答案基本上是：不。

如果您觉得有必要在方法中间重写变量，那您就错了。 该方法职责过多，需要分解。

就您的情况而言，一种使测试起作用的快速方法是：

allow(Book).to receive(:count).once.and_return(1)

或者，一种更简洁的方法是将该调用提取到一个新方法中，因此您只是在“控制”自己的行为（例如：

class Foo
  def book_details
    while (book_count > 1)
      # Do something
    end
  end

  def book_count
    Book.count
  end
end

# In your test:
let(:foo) { Foo.new }
allow(foo).to receive(:book_count).once.and_return(1)

...但是实际上，理想情况下，您应该在此处真正进行的操作是正确地测试该方法，根本不要模拟任何东西！ 由于您的代码正在与数据库进行交互，因此您的测试应创建实际记录，以使代码与之交互：

# Using FactoryBot, for example:
let!(:books) { create_list(:book, 5) }

Answer 3

Book.stub(:count).and_return(10)