ruby - rspec - 如何存根具有多个用户输入的方法？

Question

我将如何存根一个使用 rspec 接受两个用户输入的方法？可能吗？

class Mirror
    def echo
        arr = []
        print "enter something: "
        arr[0] = gets.chomp
        print "enter something: "
        arr[1] = gets.chomp

        return arr
    end
end



describe Mirror do

    it "should echo" do
        @mirror = Mirror.new
        @mirror.stub!(:gets){   "foo\n" }
        @mirror.stub!(:gets){   "bar\n" }
        arr = @mirror.echo
        #@mirror.should_receive(:puts).with("phrase")
        arr.should eql ["foo", "bar"]

    end

end

使用这些规范，@mirror.echo 的返回值是 ["bar", "bar"] 意味着第一个存根被覆盖或以其他方式忽略。

我也尝试过使用 @mirror.stub!(:gets){"foo\nbar\n"} 和 @mirror.echo 返回 ["foo\nbar\n","foo\nbar\n"]

Answer 1

您可以使用and_return方法在每次调用方法时返回不同的值。

@mirror.stub!(:gets).and_return("foo\n", "bar\n")

你的代码看起来像这样

it "should echo" do
  @mirror = Mirror.new
  @mirror.stub!(:gets).and_return("foo\n", "bar\n")
  @mirror.echo.should eql ["foo", "bar"]
end

使用示例and_return

counter.stub(:count).and_return(1,2,3)
counter.count # => 1
counter.count # => 2
counter.count # => 3
counter.count # => 3
counter.count # => 3