我试图在模型中删除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).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
这是堆栈跟踪:
NoMethodError in 'SftpToServer should return a file stream'
private method `open' called for #<Object:0x10c572620>
/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:
我用Google搜索了这个,但我无法弄清楚为什么RSpec会将sftp.file.open视为私有方法......
提前感谢任何想法!
答案 0 :(得分:1)
所以,我想到了发生了什么 - 我没有将open()方法描述为@sftp_mock对象的一部分,所以@sftp.file.open看起来像一个私有方法。修复此问题需要让@ sftp_mock.file返回@sftp_mock,以便它可以链接到open()方法,如下所示:
@sftp_mock.stub!(:file) { @sftp_mock }
@sftp_mock.should_receive(:open).with("some/pathname/filename").and_return(@file)
我也错过了@file声明......完整的spec文件是:
require 'spec_helper'
describe "SftpToServer" do
let(:ftp) { BatchTask::SftpToServer.new }
it "should return a file stream" do
@file = File.new("#{RAILS_ROOT}/spec/files/express_checkout_tdr.csv", "r")
@sftp_mock = mock('sftp')
@entry = File.stubs(:reads).with("filename").returns(@file)
@entry_mock = mock('entry')
@entry_mock.stub(:name).and_return("filename")
@sftp_mock.stub_chain(:dir, :glob).and_yield(@entry_mock)
Net::SFTP.stub(:start).and_yield(@sftp_mock)
@sftp_mock.stub!(:file) { @sftp_mock }
@sftp_mock.should_receive(:open).with("some/pathname/filename").and_return(@file)
ftp.get_file_stream("ftp.test.com", "user", "password", "some/pathname", "filename").should be_kind_of(IO)
end
end
这个其他stackoverflow问题帮助我理解了这个概念:Stub chain together with should_receive
另外,如果有更好的方法来模拟和存根SFTP,请随时发表评论。这个规范的例子有点难看!