測試輸入\輸出

Question

我對用戶輸入的代碼：

class Z 
    def self.input() 
    val = $stdin.gets.chomp 
    if val == "123" 
     p "Ok" 
    else 
     p "none" 
    end 
    end 
end 

我想測試不同的數據：

describe "Z" do 
    it "should receive none" 
    Object.stub!(:gets){"das"} 
    Z.input 
    Object.should_receive(:p).with("none") 
    end 
end 

，但我得到一個錯誤：

Failure/Error: Object.should_receive(:p).with("none") 
    (<Object (class)>).p("none") 
     expected: 1 time 
     received: 0 times 

如何測試輸出？ 謝謝。

Answer 1

試試這個：

describe Z do 
    it "should print into stdout" do 
    $stdin.stub(:gets).and_return("das") 
    $stdout.should_receive(:p).with("none") 
    Z.input 
    end 
end