From 1234d93292ca935107fc8779eabb604735461883 Mon Sep 17 00:00:00 2001 From: nonamethanks Date: Tue, 25 May 2021 12:13:01 +0200 Subject: [PATCH] Nijie: get correct image when using bookmarklet --- app/logical/sources/strategies/nijie.rb | 1 + test/unit/sources/nijie_test.rb | 10 ++++++++++ 2 files changed, 11 insertions(+) diff --git a/app/logical/sources/strategies/nijie.rb b/app/logical/sources/strategies/nijie.rb index bebec0720..d5cfd0b0c 100644 --- a/app/logical/sources/strategies/nijie.rb +++ b/app/logical/sources/strategies/nijie.rb @@ -86,6 +86,7 @@ module Sources def image_url return to_full_image_url(url) if url =~ IMAGE_URL || url =~ DOJIN_URL + return url if url =~ IMAGE_BASE_URL image_urls.first end diff --git a/test/unit/sources/nijie_test.rb b/test/unit/sources/nijie_test.rb index ba10b6b2f..a5ed9461e 100644 --- a/test/unit/sources/nijie_test.rb +++ b/test/unit/sources/nijie_test.rb @@ -362,5 +362,15 @@ module Sources assert_equal(bad_source3, Sources::Strategies.normalize_source(bad_source3)) end end + + context "an unsupported image url" do + should "not break the bookmarklet" do + image_url = "https://pic.nijie.net/01/nijie_picture/diff/main/201207181053373205_0.jpg" + ref = "https://nijie.info/view_popup.php?id=18858&#diff_1" + source = Sources::Strategies.find(image_url, ref) + + assert_equal(image_url, source.image_url) + end + end end end