diff --git a/app/logical/sources/strategies/art_station.rb b/app/logical/sources/strategies/art_station.rb index be73b183e..554fb0b44 100644 --- a/app/logical/sources/strategies/art_station.rb +++ b/app/logical/sources/strategies/art_station.rb @@ -54,7 +54,8 @@ module Sources::Strategies @json = JSON.parse(resp.body) @artist_name = json["user"]["username"] @profile_url = json["user"]["permalink"] - @image_urls = json["assets"].map do |x| + images = json["assets"].select { |asset| asset["asset_type"] == "image" } + @image_urls = images.map do |x| y, _, _ = image_url_rewriter.rewrite(x["image_url"], nil) y end diff --git a/test/unit/sources/art_station_test.rb b/test/unit/sources/art_station_test.rb index a3a329e4e..9361f10cd 100644 --- a/test/unit/sources/art_station_test.rb +++ b/test/unit/sources/art_station_test.rb @@ -70,5 +70,17 @@ module Sources assert_equal(url, @site.image_url) end end + + context "The source site for an ArtStation gallery" do + setup do + @site = Sources::Site.new("https://www.artstation.com/artwork/BDxrA") + @site.get + end + + should "get only image urls, not video urls" do + urls = %w[https://cdnb.artstation.com/p/assets/images/images/006/037/253/original/astri-lohne-sjursen-eva.jpg?1495573664] + assert_equal(urls, @site.image_urls) + end + end end end