Commit 732da2a806db1092f788614f78dc5cf0f73f3c8e
Committed by
Joenio Costa
1 parent
57fc46ec
Exists in
staging
and in
32 other branches
Added a way to set a proxy for feed handler
This has commit 4ebcbe31 squashed in which added SSL support.
Showing
2 changed files
with
33 additions
and
1 deletions
Show diff stats
lib/feed_handler.rb
... | ... | @@ -42,7 +42,13 @@ class FeedHandler |
42 | 42 | if !valid_url?(address) |
43 | 43 | raise InvalidUrl.new("\"%s\" is not a valid URL" % address) |
44 | 44 | end |
45 | - open(address, "User-Agent" => "Noosfero/#{Noosfero::VERSION}", &block) | |
45 | + header = {"User-Agent" => "Noosfero/#{Noosfero::VERSION}"} | |
46 | + if address.starts_with?("https://") | |
47 | + header.merge!(:proxy => ENV['FEED_HTTPS_PROXY']) if ENV['FEED_HTTPS_PROXY'] | |
48 | + else | |
49 | + header.merge!(:proxy => ENV['FEED_HTTP_PROXY']) if ENV['FEED_HTTP_PROXY'] | |
50 | + end | |
51 | + open(address, header, &block) | |
46 | 52 | end |
47 | 53 | return content |
48 | 54 | rescue Exception => ex | ... | ... |
test/unit/feed_handler_test.rb
... | ... | @@ -149,4 +149,30 @@ class FeedHandlerTest < ActiveSupport::TestCase |
149 | 149 | end |
150 | 150 | end |
151 | 151 | |
152 | + should 'set proxy when FEED_HTTP_PROXY is setted from env' do | |
153 | + ENV.stubs('[]').with('FEED_HTTP_PROXY').returns('http://127.0.0.1:3128') | |
154 | + 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') | |
155 | + assert_equal 'bli content', handler.fetch('http://site.org/feed.xml') | |
156 | + end | |
157 | + | |
158 | + should 'set proxy when FEED_HTTPS_PROXY is setted from env' do | |
159 | + ENV.stubs('[]').with('FEED_HTTPS_PROXY').returns('http://127.0.0.1:3128') | |
160 | + 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') | |
161 | + assert_equal 'bli content', handler.fetch('https://site.org/feed.xml') | |
162 | + end | |
163 | + | |
164 | + should 'use https proxy for https address when both env variables were defined' do | |
165 | + ENV.stubs('[]').with('FEED_HTTPS_PROXY').returns('http://127.0.0.2:3128') | |
166 | + ENV.stubs('[]').with('FEED_HTTP_PROXY').returns('http://127.0.0.1:3128') | |
167 | + 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') | |
168 | + assert_equal 'bli content', handler.fetch('https://site.org/feed.xml') | |
169 | + end | |
170 | + | |
171 | + should 'use http proxy for http address when both env variables were defined' do | |
172 | + ENV.stubs('[]').with('FEED_HTTPS_PROXY').returns('http://127.0.0.2:3128') | |
173 | + ENV.stubs('[]').with('FEED_HTTP_PROXY').returns('http://127.0.0.1:3128') | |
174 | + 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') | |
175 | + assert_equal 'bli content', handler.fetch('http://site.org/feed.xml') | |
176 | + end | |
177 | + | |
152 | 178 | end | ... | ... |