2008-06-17 18:04:18 -04:00
|
|
|
require 'tempfile'
|
|
|
|
require 'rubygems'
|
|
|
|
require 'rubygems/remote_fetcher'
|
|
|
|
|
|
|
|
##
|
|
|
|
# A fake Gem::RemoteFetcher for use in tests or to avoid real live HTTP
|
|
|
|
# requests when testing code that uses RubyGems.
|
|
|
|
#
|
|
|
|
# Example:
|
|
|
|
#
|
|
|
|
# @fetcher = Gem::FakeFetcher.new
|
|
|
|
# @fetcher.data['http://gems.example.com/yaml'] = source_index.to_yaml
|
|
|
|
# Gem::RemoteFetcher.fetcher = @fetcher
|
2009-12-08 02:19:09 -05:00
|
|
|
#
|
2008-06-17 18:04:18 -04:00
|
|
|
# # invoke RubyGems code
|
2009-12-08 02:19:09 -05:00
|
|
|
#
|
2008-06-17 18:04:18 -04:00
|
|
|
# paths = @fetcher.paths
|
|
|
|
# assert_equal 'http://gems.example.com/yaml', paths.shift
|
|
|
|
# assert paths.empty?, paths.join(', ')
|
|
|
|
#
|
|
|
|
# See RubyGems' tests for more examples of FakeFetcher.
|
|
|
|
|
|
|
|
class Gem::FakeFetcher
|
|
|
|
|
|
|
|
attr_reader :data
|
2010-02-21 21:52:35 -05:00
|
|
|
attr_reader :last_request
|
2012-11-29 01:52:18 -05:00
|
|
|
attr_reader :api_endpoints
|
2008-06-17 18:04:18 -04:00
|
|
|
attr_accessor :paths
|
|
|
|
|
|
|
|
def initialize
|
|
|
|
@data = {}
|
|
|
|
@paths = []
|
2012-11-29 01:52:18 -05:00
|
|
|
@api_endpoints = {}
|
|
|
|
end
|
|
|
|
|
|
|
|
def api_endpoint(uri)
|
|
|
|
@api_endpoints[uri] || uri
|
2008-06-17 18:04:18 -04:00
|
|
|
end
|
|
|
|
|
2010-02-21 21:52:35 -05:00
|
|
|
def find_data(path)
|
2008-06-17 18:04:18 -04:00
|
|
|
path = path.to_s
|
|
|
|
@paths << path
|
2010-02-21 21:52:35 -05:00
|
|
|
raise ArgumentError, 'need full URI' unless path =~ %r'^https?://'
|
2008-06-17 18:04:18 -04:00
|
|
|
|
2008-09-25 06:13:50 -04:00
|
|
|
unless @data.key? path then
|
|
|
|
raise Gem::RemoteFetcher::FetchError.new("no data for #{path}", path)
|
2008-06-17 18:04:18 -04:00
|
|
|
end
|
|
|
|
|
2010-02-21 21:52:35 -05:00
|
|
|
@data[path]
|
|
|
|
end
|
|
|
|
|
|
|
|
def fetch_path path, mtime = nil
|
|
|
|
data = find_data(path)
|
2008-09-25 06:13:50 -04:00
|
|
|
|
2008-06-25 22:06:00 -04:00
|
|
|
if data.respond_to?(:call) then
|
|
|
|
data.call
|
|
|
|
else
|
2008-09-25 06:13:50 -04:00
|
|
|
if path.to_s =~ /gz$/ and not data.nil? and not data.empty? then
|
|
|
|
data = Gem.gunzip data
|
|
|
|
end
|
|
|
|
|
2008-06-25 22:06:00 -04:00
|
|
|
data
|
|
|
|
end
|
2008-06-17 18:04:18 -04:00
|
|
|
end
|
|
|
|
|
2012-12-06 00:13:08 -05:00
|
|
|
def cache_update_path uri, path = nil, update = true
|
2012-11-29 01:52:18 -05:00
|
|
|
if data = fetch_path(uri)
|
2012-12-06 00:13:08 -05:00
|
|
|
open(path, 'wb') { |io| io.write data } if path and update
|
2012-11-29 01:52:18 -05:00
|
|
|
data
|
|
|
|
else
|
|
|
|
Gem.read_binary(path) if path
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
2010-02-21 21:52:35 -05:00
|
|
|
# Thanks, FakeWeb!
|
|
|
|
def open_uri_or_path(path)
|
|
|
|
data = find_data(path)
|
|
|
|
body, code, msg = data
|
|
|
|
|
|
|
|
response = Net::HTTPResponse.send(:response_class, code.to_s).new("1.0", code.to_s, msg)
|
|
|
|
response.instance_variable_set(:@body, body)
|
|
|
|
response.instance_variable_set(:@read, true)
|
|
|
|
response
|
|
|
|
end
|
|
|
|
|
|
|
|
def request(uri, request_class, last_modified = nil)
|
|
|
|
data = find_data(uri)
|
|
|
|
body, code, msg = data
|
|
|
|
|
|
|
|
@last_request = request_class.new uri.request_uri
|
|
|
|
yield @last_request if block_given?
|
|
|
|
|
|
|
|
response = Net::HTTPResponse.send(:response_class, code.to_s).new("1.0", code.to_s, msg)
|
|
|
|
response.instance_variable_set(:@body, body)
|
|
|
|
response.instance_variable_set(:@read, true)
|
|
|
|
response
|
|
|
|
end
|
|
|
|
|
2008-06-17 18:04:18 -04:00
|
|
|
def fetch_size(path)
|
|
|
|
path = path.to_s
|
|
|
|
@paths << path
|
2008-09-25 06:13:50 -04:00
|
|
|
|
2008-06-17 18:04:18 -04:00
|
|
|
raise ArgumentError, 'need full URI' unless path =~ %r'^http://'
|
|
|
|
|
2008-09-25 06:13:50 -04:00
|
|
|
unless @data.key? path then
|
|
|
|
raise Gem::RemoteFetcher::FetchError.new("no data for #{path}", path)
|
2008-06-17 18:04:18 -04:00
|
|
|
end
|
|
|
|
|
2008-09-25 06:13:50 -04:00
|
|
|
data = @data[path]
|
|
|
|
|
2008-06-17 18:04:18 -04:00
|
|
|
data.respond_to?(:call) ? data.call : data.length
|
|
|
|
end
|
|
|
|
|
|
|
|
def download spec, source_uri, install_dir = Gem.dir
|
2011-05-31 23:45:05 -04:00
|
|
|
name = File.basename spec.cache_file
|
2012-11-29 01:52:18 -05:00
|
|
|
path = if Dir.pwd == install_dir then # see fetch_command
|
|
|
|
install_dir
|
|
|
|
else
|
|
|
|
File.join install_dir, "cache"
|
|
|
|
end
|
|
|
|
|
|
|
|
path = File.join path, name
|
2008-06-17 18:04:18 -04:00
|
|
|
|
|
|
|
Gem.ensure_gem_subdirectories install_dir
|
|
|
|
|
|
|
|
if source_uri =~ /^http/ then
|
|
|
|
File.open(path, "wb") do |f|
|
|
|
|
f.write fetch_path(File.join(source_uri, "gems", name))
|
|
|
|
end
|
|
|
|
else
|
|
|
|
FileUtils.cp source_uri, path
|
|
|
|
end
|
|
|
|
|
|
|
|
path
|
|
|
|
end
|
|
|
|
|
2011-03-01 04:41:32 -05:00
|
|
|
def download_to_cache dependency
|
2012-11-29 01:52:18 -05:00
|
|
|
found, _ = Gem::SpecFetcher.fetcher.spec_for_dependency dependency
|
2011-03-01 04:41:32 -05:00
|
|
|
|
|
|
|
return if found.empty?
|
|
|
|
|
2012-11-29 01:52:18 -05:00
|
|
|
spec, source = found.first
|
2011-03-01 04:41:32 -05:00
|
|
|
|
2012-11-29 01:52:18 -05:00
|
|
|
download spec, source.uri.to_s
|
2011-03-01 04:41:32 -05:00
|
|
|
end
|
|
|
|
|
2008-06-17 18:04:18 -04:00
|
|
|
end
|
|
|
|
|
|
|
|
# :stopdoc:
|
|
|
|
class Gem::RemoteFetcher
|
|
|
|
|
|
|
|
def self.fetcher=(fetcher)
|
|
|
|
@fetcher = fetcher
|
|
|
|
end
|
|
|
|
|
|
|
|
end
|
|
|
|
# :startdoc:
|
|
|
|
|
|
|
|
##
|
|
|
|
# A StringIO duck-typed class that uses Tempfile instead of String as the
|
|
|
|
# backing store.
|
2011-08-04 21:00:01 -04:00
|
|
|
#
|
|
|
|
# This is available when rubygems/test_utilities is required.
|
2008-06-17 18:04:18 -04:00
|
|
|
#--
|
|
|
|
# This class was added to flush out problems in Rubinius' IO implementation.
|
|
|
|
|
2009-12-30 03:59:16 -05:00
|
|
|
class TempIO < Tempfile
|
2008-06-17 18:04:18 -04:00
|
|
|
def initialize(string = '')
|
2009-12-30 03:59:16 -05:00
|
|
|
super "TempIO"
|
|
|
|
binmode
|
|
|
|
write string
|
|
|
|
rewind
|
2008-06-17 18:04:18 -04:00
|
|
|
end
|
|
|
|
|
|
|
|
def string
|
2009-12-30 03:59:16 -05:00
|
|
|
flush
|
|
|
|
Gem.read_binary path
|
2008-06-17 18:04:18 -04:00
|
|
|
end
|
|
|
|
end
|
|
|
|
|