diff --git a/lib/feed_handler.rb b/lib/feed_handler.rb index af1d356..7d9b5ed 100644 --- a/lib/feed_handler.rb +++ b/lib/feed_handler.rb @@ -42,7 +42,13 @@ class FeedHandler if !valid_url?(address) raise InvalidUrl.new("\"%s\" is not a valid URL" % address) end - open(address, "User-Agent" => "Noosfero/#{Noosfero::VERSION}", &block) + header = {"User-Agent" => "Noosfero/#{Noosfero::VERSION}"} + if address.starts_with?("https://") + header.merge!(:proxy => ENV['FEED_HTTPS_PROXY']) if ENV['FEED_HTTPS_PROXY'] + else + header.merge!(:proxy => ENV['FEED_HTTP_PROXY']) if ENV['FEED_HTTP_PROXY'] + end + open(address, header, &block) end return content rescue Exception => ex diff --git a/test/unit/feed_handler_test.rb b/test/unit/feed_handler_test.rb index 0eebf36..784e94a 100644 --- a/test/unit/feed_handler_test.rb +++ b/test/unit/feed_handler_test.rb @@ -149,4 +149,30 @@ class FeedHandlerTest < ActiveSupport::TestCase end end + should 'set proxy when FEED_HTTP_PROXY is setted from env' do + ENV.stubs('[]').with('FEED_HTTP_PROXY').returns('http://127.0.0.1:3128') + handler.expects(:open).with('http://site.org/feed.xml', {"User-Agent" => "Noosfero/#{Noosfero::VERSION}", :proxy => 'http://127.0.0.1:3128'}, anything).returns('bli content') + assert_equal 'bli content', handler.fetch('http://site.org/feed.xml') + end + + should 'set proxy when FEED_HTTPS_PROXY is setted from env' do + ENV.stubs('[]').with('FEED_HTTPS_PROXY').returns('http://127.0.0.1:3128') + handler.expects(:open).with('https://site.org/feed.xml', {"User-Agent" => "Noosfero/#{Noosfero::VERSION}", :proxy => 'http://127.0.0.1:3128'}, anything).returns('bli content') + assert_equal 'bli content', handler.fetch('https://site.org/feed.xml') + end + + should 'use https proxy for https address when both env variables were defined' do + ENV.stubs('[]').with('FEED_HTTPS_PROXY').returns('http://127.0.0.2:3128') + ENV.stubs('[]').with('FEED_HTTP_PROXY').returns('http://127.0.0.1:3128') + handler.expects(:open).with('https://site.org/feed.xml', {"User-Agent" => "Noosfero/#{Noosfero::VERSION}", :proxy => 'http://127.0.0.2:3128'}, anything).returns('bli content') + assert_equal 'bli content', handler.fetch('https://site.org/feed.xml') + end + + should 'use http proxy for http address when both env variables were defined' do + ENV.stubs('[]').with('FEED_HTTPS_PROXY').returns('http://127.0.0.2:3128') + ENV.stubs('[]').with('FEED_HTTP_PROXY').returns('http://127.0.0.1:3128') + handler.expects(:open).with('http://site.org/feed.xml', {"User-Agent" => "Noosfero/#{Noosfero::VERSION}", :proxy => 'http://127.0.0.1:3128'}, anything).returns('bli content') + assert_equal 'bli content', handler.fetch('http://site.org/feed.xml') + end + end -- libgit2 0.21.2