d9d3c1dfe4
Rename Sources::Strategies to Source::Extractor. A Source::Extractor represents a thing that extracts information from a given URL.
18 lines
682 B
Ruby
18 lines
682 B
Ruby
module DownloadTestHelper
|
|
def assert_downloaded(expected_filesize, source, referer = nil)
|
|
strategy = Source::Extractor.find(source, referer)
|
|
file = strategy.download_file!(strategy.image_urls.sole)
|
|
assert_equal(expected_filesize, file.size, "Tested source URL: #{source}")
|
|
end
|
|
|
|
def assert_rewritten(expected_source, test_source, test_referer = nil)
|
|
strategy = Source::Extractor.find(test_source, test_referer)
|
|
rewritten_source = strategy.image_urls.sole
|
|
assert_match(expected_source, rewritten_source, "Tested source URL: #{test_source}")
|
|
end
|
|
|
|
def assert_not_rewritten(source, referer = nil)
|
|
assert_rewritten(source, source, referer)
|
|
end
|
|
end
|