2012-07-31 59 views
6

我正在嘗試從一個對象中剔除Net :: SFTP。這裏的模型:RSpec:Stubbing SFTP

class BatchTask 
    require 'net/sftp' 

    def get_file_stream(host, username, password, path_to_dir, filename) 
    raise ArgumentError if host.nil? or username.nil? or password.nil? or path_to_dir.nil? or filename.nil? 
    file_stream = nil 
    Net::SFTP.start(host, username, password) do |sftp| 
     sftp.dir.glob(path_to_dir, filename) do |entry| 
     # Verify the directory contents 
     raise RuntimeError(true), "file: #{path_to_dir}/#{filename} not found on SFTP server" if entry.nil? 
     file_stream = sftp.file.open("#{path_to_dir}/#{entry.name}") 
     end 
    end 
    file_stream 
    end 

end 

這裏的規格:

require 'spec_helper' 

describe "SftpToServer" do 
    let(:ftp) { BatchTask::SftpToServer.new } 

it "should return a file stream" do 
    @sftp_mock = mock('sftp') 
    @entry = File.stubs(:reads).with("filename").returns(@file) 
    @entry_mock = mock('entry') 
    @entry_mock.stub(:name).with("filename").and_return("filename") 
    @sftp_mock.stub_chain(:dir, :glob).and_yield(@entry_mock) 
    Net::SFTP.stub(:start).and_yield(@sftp_mock) 
    @sftp_mock.stub_chain(:file, :open).with("filename").and_yield(@file) 

    ftp.get_file_stream("ftp.test.com", "user", "password", "some/pathname", "filename").should be_kind_of(IO) 
    end 

end 

這裏的堆棧跟蹤:

Spec::Mocks::MockExpectationError in 'SftpToServer should return a file stream' 
Mock "entry" received :name with unexpected arguments 
    expected: ("filename") 
     got: (no args) 
/Users/app/models/batch_task/sftp_to_server.rb:12:in `get_file_stream' 
/Users/app/models/batch_task/sftp_to_server.rb:9:in `get_file_stream' 
/Users/app/models/batch_task/sftp_to_server.rb:8:in `get_file_stream' 
./spec/models/batch_task/sftp_to_server_spec.rb:15: 

首先,是我的方法就在這裏?我想刪除SFTP的功能,因爲我們可以確信它已經很好地測試過了。相反,我想專注於確保文件流由get_file_stream()中的「黑盒子」返回。

其次,我該如何正確地取出sftp.file.open()來完成此操作?

在此先感謝您的任何想法!

回答

7

首先嘲諷了SFTP是有兩個原因一個好主意:

  1. 你不是寫一個SFTP測試,重點測試只是你設置 了測試。
  2. 您可以在測試中刪除對網絡的任何依賴關係 - 您不希望由於某些事情無法控制而導致測試開始失敗。

至於錯誤去,這是你的直接問題:

@entry_mock.stub(:name).with("filename").and_return("filename") 

在這裏,我們磕碰entry.name("filename")而不僅僅是entry.name

將其更改爲:

@entry_mock.stub(:name).and_return("filename") 

,讓我知道你上車。

+0

@cheesewasel - 工作就像一個魅力!謝謝您的幫助! – Sly 2012-07-31 17:16:55

+0

完全沒問題。 :) – 2012-08-01 02:23:36