[danbooru] add test cases

This commit is contained in:
Mike Fährmann
2015-12-12 15:59:26 +01:00
parent 5304e5beef
commit c8fdfbec42

View File

@@ -19,13 +19,25 @@ class DanbooruTagExtractor(DanbooruExtractor, booru.BooruTagExtractor):
"""Extract images from danbooru based on search-tags"""
subcategory = "tag"
pattern = [r"(?:https?://)?(?:www\.)?danbooru.donmai.us/posts\?(?:utf8=%E2%9C%93&)?tags=([^&]+)"]
test = [("https://danbooru.donmai.us/posts?tags=heath_ledger", {
"url": "a261c33f117c7395f0eac54091075e67c8e66fca",
"keyword": "86c1e0fc2879befe90cf8b4404264338b8063685",
})]
class DanbooruPoolExtractor(DanbooruExtractor, booru.BooruPoolExtractor):
"""Extract image-pools from danbooru"""
subcategory = "pool"
pattern = [r"(?:https?://)?(?:www\.)?danbooru.donmai.us/pools/(\d+)"]
test = [("https://danbooru.donmai.us/pools/7659", {
"url": "db522c7f74d42ec1940fb23d1a58ed8edb65f9ae",
"keyword": "bf8256e88ac0d032808e9f5aa165c9a1fa7e451c",
})]
class DanbooruPostExtractor(DanbooruExtractor, booru.BooruPostExtractor):
"""Extract single images from danbooru"""
subcategory = "post"
pattern = [r"(?:https?://)?(?:www\.)?danbooru.donmai.us/posts/(\d+)"]
test = [("https://danbooru.donmai.us/posts/1534703", {
"url": "3a87a761dad69dfa038499aeacef679a3912816b",
"keyword": "0a2f4dd6b8384fcbc805302040234536895a2dc3",
})]