danbooru/test/unit/sources/fanbox_test.rb

require 'test_helper'

module Sources
  class FanboxTest < ActiveSupport::TestCase
    context "A free Pixiv Fanbox post" do
      setup do
        @post1 = Sources::Strategies.find("https://yanmi0308.fanbox.cc/posts/1141325")
        @post2 = Sources::Strategies.find("https://chanxco.fanbox.cc/posts/209386")
        @post3 = Sources::Strategies.find("https://downloads.fanbox.cc/images/post/209386/w/1200/8dRNHXkFqAwSt31W2Bg8fSdL.jpeg")

        assert_nothing_raised { @post1.to_h }
        assert_nothing_raised { @post2.to_h }
        assert_nothing_raised { @post3.to_h }
      end

      should "get the image urls" do
        # "images" in api response
        images1 = %w[
          https://downloads.fanbox.cc/images/post/1141325/q7GaJ0A9J5Uz8kvEAUizHJoN.png
          https://downloads.fanbox.cc/images/post/1141325/LMJz0sAig5h9D3rPZGCEGniZ.png
          https://downloads.fanbox.cc/images/post/1141325/dRSz380Uf3N8s4pT2ADEXBco.png
          https://downloads.fanbox.cc/images/post/1141325/h48L2mbm39qqNUB1abLAvzvg.png
        ]
        assert_equal(images1, @post1.image_urls)

        # "imageMapi" in api response (embedded pics)
        images2 = %w[
          https://downloads.fanbox.cc/images/post/209386/Q8rZ0iMHpcmJDACEzNGjTj9E.jpeg
          https://downloads.fanbox.cc/images/post/209386/8dRNHXkFqAwSt31W2Bg8fSdL.jpeg
          https://downloads.fanbox.cc/images/post/209386/AGGWF0JxytFcNL2ybPKBaqp7.jpeg
        ]
        assert_equal(images2, @post2.image_urls)
      end

      should "get the commentary" do
        # Normal commentary
        assert_equal("栗山やんみ（デザイン）", @post1.artist_commentary_title)

        body1 = "˗ˋˏ Special Thanks ˎˊ˗   (敬称略)\n\n🎨キャラクターデザイン\n特急みかん  https://twitter.com/tokkyuumikan\n\n🤖3Dモデリング\n（仮）  https://twitter.com/Admiral_TMP\n\n⚙プログラミング\n神無月ユズカ  https://twitter.com/Kannaduki_Yzk\n\n🎧OP・EDミュージック\n卓球少年  https://twitter.com/takkyuu_s\n\n📻BGM\nC  https://twitter.com/nica2c\n\n🖌ロゴデザイン\nてづかもり  https://twitter.com/tezkamori\n\n🎨SDキャラクター\nAZU。  https://twitter.com/tokitou_aaa"
        assert_equal(body1, @post1.artist_commentary_desc)

        # With embedded pics
        assert_equal("水着BBちゃん＋アラフィフ＋ライダーさん", @post2.artist_commentary_title)
        assert_equal("水着BBちゃん＋アラフィフ＋ライダーさん", @post3.artist_commentary_title)

        body2 = "今週のらくがきまとめ\n\nhttps://downloads.fanbox.cc/images/post/209386/Q8rZ0iMHpcmJDACEzNGjTj9E.jpeg\n水着BBちゃん\n第一再臨もなかなかセクシー\nhttps://downloads.fanbox.cc/images/post/209386/8dRNHXkFqAwSt31W2Bg8fSdL.jpeg\nアラフィフ\n男キャラも描いていこうと練習中\n新宿での軽いキャラも好き\nhttps://downloads.fanbox.cc/images/post/209386/AGGWF0JxytFcNL2ybPKBaqp7.jpeg\nライダーさん\nつい眼鏡も描いてしまう\n\n＃FGO\n"
        assert_equal(body2, @post2.artist_commentary_desc)
        assert_equal(body2, @post3.artist_commentary_desc)
      end

      should "get the right page url" do
        assert_equal("https://yanmi0308.fanbox.cc/posts/1141325", @post1.page_url)
        assert_equal("https://chanxco.fanbox.cc/posts/209386", @post2.page_url)
        assert_equal("https://chanxco.fanbox.cc/posts/209386", @post3.page_url)
      end

      should "correctly download the right image" do
        assert_downloaded(431_225, @post1.image_urls[0])
        assert_downloaded(753_048, @post1.image_urls[1])
        assert_downloaded(589_327, @post1.image_urls[2])
        assert_downloaded(178_739, @post1.image_urls[3])

        assert_downloaded(245_678, @post2.image_urls[0])
        assert_downloaded(320_056, @post2.image_urls[1])
        assert_downloaded(666_681, @post2.image_urls[2])

        assert_downloaded(320_056, @post3.image_urls.sole)
      end

      should "get the tags" do
        tags = [
          ["栗山やんみ", "https://fanbox.cc/tags/栗山やんみ"], ["VTuber", "https://fanbox.cc/tags/VTuber"], ["三面図", "https://fanbox.cc/tags/三面図"],
          ["イラスト", "https://fanbox.cc/tags/イラスト"], ["ロゴデザイン", "https://fanbox.cc/tags/ロゴデザイン"], ["モデリング", "https://fanbox.cc/tags/モデリング"]
        ]
        assert_equal(tags, @post1.tags)
      end

      should "find the correct artist" do
        @artist1 = FactoryBot.create(:artist, name: "yanmi", url_string: @post1.url)
        @artist2 = FactoryBot.create(:artist, name: "chanxco", url_string: @post2.url)
        assert_equal([@artist1], @post1.artists)
        assert_equal([@artist2], @post2.artists)
        assert_equal([@artist2], @post3.artists)
      end

      should "find the right artist names" do
        assert_equal("yanmi0308", @post1.artist_name)
        assert_equal("栗山やんみ", @post1.display_name)
        assert_equal("chanxco", @post2.artist_name)
        assert_equal("CHANxCO", @post2.display_name)
        assert_equal(@post2.artist_name, @post3.artist_name)
        assert_equal(@post2.display_name, @post3.display_name)
      end
    end

    context "an age-restricted fanbox post" do
      should "work" do
        @source = Sources::Strategies.find("https://mfr.fanbox.cc/posts/1306390")

        assert_nothing_raised { @source.to_h }
        assert_equal("mfr", @source.artist_name)
        assert_equal(["https://downloads.fanbox.cc/images/post/1306390/VOXblkyvltL5fRhMoR7RdSkk.png"], @source.image_urls)
      end
    end

    context "A link in the old format" do
      should "still work" do
        post = Sources::Strategies.find("https://www.pixiv.net/fanbox/creator/1566167/post/39714")
        assert_nothing_raised { post.to_h }
        assert_equal("https://omu001.fanbox.cc", post.profile_url)
        assert_equal("https://omu001.fanbox.cc/posts/39714", post.page_url)
        artist = FactoryBot.create(:artist, name: "omu", url_string: "https://omu001.fanbox.cc")
        assert_equal([artist], post.artists)
      end
    end

    context "A cover image" do
      should "still work" do
        post = Sources::Strategies.find("https://pixiv.pximg.net/c/1620x580_90_a2_g5/fanbox/public/images/creator/1566167/cover/QqxYtuWdy4XWQx1ZLIqr4wvA.jpeg")
        assert_nothing_raised { post.to_h }
        assert_downloaded(750_484, post.image_urls.sole)
        assert_equal("https://omu001.fanbox.cc", post.profile_url)
        assert_equal(post.profile_url, post.canonical_url)
        artist = FactoryBot.create(:artist, name: "omu", url_string: "https://omu001.fanbox.cc")
        assert_equal([artist], post.artists)
      end
    end

    context "A dead profile picture from the old domain" do
      should "still find the artist" do
        post = Sources::Strategies.find("https://pixiv.pximg.net/c/400x400_90_a2_g5/fanbox/public/images/creator/1566167/profile/Ix6bnJmTaOAFZhXHLbWyIY1e.jpeg")
        assert_equal("https://omu001.fanbox.cc", post.profile_url)
        artist = FactoryBot.create(:artist, name: "omu", url_string: "https://omu001.fanbox.cc")
        assert_equal([artist], post.artists)
      end
    end

    context "generating page urls" do
      should "convert cover images to the profile url" do
        cover = "https://pixiv.pximg.net/c/400x400_90_a2_g5/fanbox/public/images/creator/1566167/profile/Ix6bnJmTaOAFZhXHLbWyIY1e.jpeg"
        assert_equal("https://www.pixiv.net/fanbox/creator/1566167", Source::URL.page_url(cover))
      end

      should "handle inconvertible urls" do
        assert_nil(Source::URL.page_url("https://pixiv.pximg.net/c/936x600_90_a2_g5/fanbox/public/images/plan/4635/cover/L6AZNneFuHW6r25CHHlkpHg4.jpeg"))
        assert_nil(Source::URL.page_url("https://downloads.fanbox.cc/images/post/39714/JvjJal8v1yLgc5DPyEI05YpT.png"))
      end
    end
  end
end