Merge pull request #4488 from nonamethanks/add_weibo_support

Add Weibo support
2020-05-27 16:53:14 -05:00
parent 2c60a51f64 5c7307a1c9
commit feeea6602c
4 changed files with 312 additions and 1 deletions
--- a/test/unit/sources/weibo_test.rb
+++ b/test/unit/sources/weibo_test.rb
@@ -0,0 +1,92 @@
+require 'test_helper'
+
+module Sources
+  class WeiboTest < ActiveSupport::TestCase
+    context "A post with multiple pictures" do
+      setup do
+        @site = Sources::Strategies.find("https://www.weibo.com/5501756072/J2UNKfbqV?type=comment#_rnd1590548401855")
+      end
+
+      should "extract all the image urls" do
+        urls = %w[
+          https://wx1.sinaimg.cn/large/0060kO5aly1gezsyt5xvhj30ok0sgtc9.jpg
+          https://wx3.sinaimg.cn/large/0060kO5aly1gezsyuaas1j30go0sgjtj.jpg
+          https://wx3.sinaimg.cn/large/0060kO5aly1gezsys1ai9j30gi0sg0v9.jpg
+        ]
+        assert_equal(urls, @site.image_urls)
+      end
+
+      should "get the correct commentary" do
+        assert_not_nil(@site.artist_commentary_desc)
+      end
+
+      should "get the profile url" do
+        assert_equal("https://www.weibo.com/u/5501756072", @site.profile_url)
+      end
+
+      should "set the right source" do
+        assert_equal("https://www.weibo.com/5501756072/J2UNKfbqV", @site.canonical_url)
+      end
+
+      should "download an image" do
+        assert_downloaded(134_721, @site.image_url)
+      end
+
+      should "get the tags" do
+        tags = [
+          %w[fgo https://s.weibo.com/weibo/fgo],
+          %w[Alter组 https://s.weibo.com/weibo/Alter组]
+        ]
+        assert_equal(tags, @site.tags)
+      end
+
+      should "find the correct artist" do
+        @artist = FactoryBot.create(:artist, name: "nipi27", url_string: "https://www.weibo.com/u/5501756072")
+        assert_equal([@artist], @site.artists)
+      end
+    end
+
+    context "A deleted or not existing picture" do
+      should "still find the artist name" do
+        site = Sources::Strategies.find("https://www.weibo.com/5501756072/AsdAsdAsd")
+        artist = FactoryBot.create(:artist, name: "nipi27", url_string: "https://www.weibo.com/u/5501756072")
+
+        assert_equal([artist], site.artists)
+      end
+    end
+
+    context "A post with video" do
+      should "get the correct video" do
+        site = Sources::Strategies.find("https://www.weibo.com/5501756072/IF9fugHzj")
+
+        assert_downloaded(7_676_656, site.image_url)
+      end
+    end
+
+    context "A direct image sample upload" do
+      should "get the largest version" do
+        sample = Sources::Strategies.find("https://wx3.sinaimg.cn/mw690/a00fa34cly1gf62g2n8z3j21yu2jo1ky.jpg")
+
+        assert_equal("https://wx3.sinaimg.cn/large/a00fa34cly1gf62g2n8z3j21yu2jo1ky.jpg", sample.image_url)
+      end
+    end
+
+    context "An album url for a post with multiple pictures" do
+      should "upload the right picture rather than just the first" do
+        site = Sources::Strategies.find("http://photo.weibo.com/2125874520/wbphotos/large/mid/4194742441135220/pid/7eb64558gy1fnbryb5nzoj20dw10419t")
+
+        assert_equal("https://wx4.sinaimg.cn/large/7eb64558gy1fnbryb5nzoj20dw10419t.jpg", site.image_url)
+      end
+    end
+
+    context "An upload from the batch bookmarklet" do
+      should "set the right source" do
+        url = "https://wx1.sinaimg.cn/large/7eb64558gy1fnbryriihwj20dw104wtu.jpg"
+        ref = "http://photo.weibo.com/2125874520/wbphotos/large/mid/4194742441135220/pid/7eb64558gy1fnbryb5nzoj20dw10419t"
+        site = Sources::Strategies.find(url, ref)
+
+        assert_equal("https://www.weibo.com/2125874520/FDKGo4Lk0", site.canonical_url)
+      end
+    end
+  end
+end