reset connection for pixiv tests

This commit is contained in:
Albert Yi
2018-05-09 14:59:18 -07:00
parent c144159dd9
commit d4a2521eec
2 changed files with 9 additions and 1 deletions

View File

@@ -2,6 +2,13 @@ require 'test_helper'
module Downloads
class PixivTest < ActiveSupport::TestCase
def teardown
super
# need to reset the connection
Sources::Strategies::Pixiv.new.agent.shutdown
end
context "An ugoira site for pixiv" do
setup do
@download = Downloads::File.new("http://www.pixiv.net/member_illust.php?mode=medium&illust_id=62247364")

View File

@@ -1,11 +1,12 @@
require 'test_helper'
require 'resolv-replace'
module Sources
class PixivTest < ActiveSupport::TestCase
def get_source(source)
@site = Sources::Site.new(source)
@site.get
# need to reset the connection
@site.strategy.agent.shutdown
@site
end