如何指定文件下载

Question

我正在研究一个库，需要能够使用RestClient从远程API下载插件文件。该库首先获取一个插件列表，然后将每个插件作为原始文件下载，并将每个插件保存在plugins目录中。

这是我到目前为止，但它是失败的我:

require 'yaml'
module Monitaur
  class Client
    attr_accessor :logger, :client_key, :server_url, :config, :raw_config,
                  :plugin_manifest
    def initialize
      load_config
      @plugin_manifest ||= []
    end
    def run
      get_plugin_manifest
      sync_plugins
    end
    def get_plugin_manifest
      res = RestClient.get("#{server_url}/nodes/#{client_key}/plugins")
      @plugin_manifest = JSON.parse(res)
    end
    def sync_plugins
      @plugin_manifest.each do |plugin|
        res = RestClient.get("#{server_url}/plugins/#{plugin['name']}")
        File.open(File.join(Monitaur.plugin_dir, "#{plugin['name']}.rb"), "w+") do |file|
          file.write res.body
        end
      end
    end
    def load_config
      if File.exist?(Monitaur.config_file_path) && File.readable?(Monitaur.config_file_path)
        @raw_config = YAML.load_file(Monitaur.config_file_path)
      else
        raise IOError, "Cannot open or read #{Monitaur.config_file_path}"
      end
      @server_url = raw_config['server_url']
      @client_key = raw_config['client_key']
    end

  end
end

和client_spec.rb

require 'spec_helper'
module Monitaur
  describe Client do
    let(:server_url) { "http://api.monitaurapp.com" }
    let(:client_key) { "asdf1234" }
    describe "#load_config" do
      let(:client) { Monitaur::Client.new }
      before do
        File.open(Monitaur.config_file_path, "w") do |file|
          file.puts "server_url: http://api.monitaurapp.com"
          file.puts "client_key: asdf1234"
        end 
      end
      it "loads up the configuration file" do
        client.load_config
        client.server_url.should == "http://api.monitaurapp.com"
        client.client_key.should == "asdf1234"
      end
    end
    describe "#get_plugin_manifest" do
      let(:client) { Monitaur::Client.new }
      before do
        stub_get_plugin_manifest
      end
      it "retrieves a plugins manifest from the server" do
        client.get_plugin_manifest
        client.plugin_manifest.should == plugin_manifest_response
      end
    end
    describe "#sync_plugins" do
      let(:client) { Monitaur::Client.new }
      let(:foo_plugin) { mock('foo_plugin') }
      let(:bar_plugin) { mock('bar_plugin') }
      before do
        FileUtils.mkdir("/tmp")
        File.open("/tmp/foo_plugin.rb", "w+") do |file|
          file.write %|
          class FooPlugin < Monitaur::Plugin
            name "foo_plugin"
            desc "A test plugin to determine whether plugin sync works"
            def run
              { :foo => 'foo' }
            end
          end
          |
        end
        File.open("/tmp/bar_plugin.rb", "w+") do |file|
          file.write %|
          class BarPlugin < Monitaur::Plugin
            name "bar_plugin"
            desc "A test plugin to determine whether plugin sync works"
            def run
              { :bar => 'bar' }
            end
          end
          |
        end
        Monitaur.install
        stub_get_plugin_manifest
        stub_sync_plugins
        client.get_plugin_manifest
      end
      it "downloads plugins to the cache directory" do
        File.should_receive(:open).
          with(File.join(Monitaur.plugin_dir, "foo_plugin.rb"), "w+")
          and_yield(foo_plugin)
        client.sync_plugins
        File.exist?("/home/user/.monitaur/cache/plugins/foo_plugin.rb").should be_true
        File.exist?("/home/user/.monitaur/cache/plugins/bar_plugin.rb").should be_true
      end
    end
  end
end
def stub_get_plugin_manifest
  stub_request(:get, "#{server_url}/nodes/#{client_key}/plugins").
    to_return(
      :status => 200,
      :body => %Q{
        [
          {
            "name": "foo_plugin",
            "checksum": "qwer5678"
          },
          {
            "name": "bar_plugin",
            "checksum": "hjkl4321"
          }
        ]
      }
    )
end
def plugin_manifest_response
  [
    {
      "name" => "foo_plugin",
      "checksum" => "qwer5678"
    },
    {
      "name" => "bar_plugin",
      "checksum" => "hjkl4321"
    }
  ]
end
def stub_sync_plugins
  stub_request(:get, "#{server_url}/plugins/foo_plugin").
    to_return(:body => File.open('/tmp/foo_plugin.rb').read)
  stub_request(:get, "#{server_url}/plugins/bar_plugin").
    to_return(:body => File.open('/tmp/bar_plugin.rb').read)
end

如何测试下载过程?

Answer 1

我使用FakeWeb用于此目的，因为如果其他网站关闭或其他原因，您的规格确实不需要失败。请参阅文档中的"重播录制的响应"。我们所做的是curl页面，将其保存为固定装置，并在规格中重播。