tests: refactor tumblr tests to use assert_downloaded.

This commit is contained in:
evazion
2017-06-21 11:08:04 -05:00
parent 5684ba3b1d
commit 16a51533bc

View File

@@ -3,32 +3,17 @@ require 'test_helper'
module Downloads
class TumblrTest < ActiveSupport::TestCase
context "a download for a tumblr 500 sample" do
setup do
@source = "http://24.media.tumblr.com/fc328250915434e66e8e6a92773f79d0/tumblr_mf4nshfibc1s0oswoo1_500.jpg"
@tempfile = Tempfile.new("danbooru-test")
@download = Downloads::File.new(@source, @tempfile.path)
@download.download!
end
should "instead change the source to the 1280 version" do
assert_equal("http://24.media.tumblr.com/fc328250915434e66e8e6a92773f79d0/tumblr_mf4nshfibc1s0oswoo1_1280.jpg", @download.source)
end
should "instead download the 1280 version" do
assert_equal(196_617, ::File.size(@tempfile.path))
@source = "http://24.media.tumblr.com/fc328250915434e66e8e6a92773f79d0/tumblr_mf4nshfibc1s0oswoo1_500.jpg"
assert_rewritten("http://24.media.tumblr.com/fc328250915434e66e8e6a92773f79d0/tumblr_mf4nshfibc1s0oswoo1_1280.jpg", @source)
assert_downloaded(196_617, @source)
end
end
context "a download for a tumblr 500 image without a larger size" do
setup do
@source = "http://25.media.tumblr.com/tumblr_lxbzel2H5y1r9yjhso1_500.jpg"
@tempfile = Tempfile.new("danbooru-test")
@download = Downloads::File.new(@source, @tempfile.path)
@download.download!
end
should "download the 500 version" do
assert_equal(90_122, ::File.size(@tempfile.path))
@source = "http://25.media.tumblr.com/tumblr_lxbzel2H5y1r9yjhso1_500.jpg"
assert_downloaded(90_122, @source)
end
end
end