warc2zim/tests/test_warc_to_zim.py

827 lines
28 KiB
Python
Raw Normal View History

2020-08-11 03:41:33 +00:00
#!/usr/bin/env python
# vim: ai ts=4 sts=4 et sw=4 nu
import io
2020-12-09 10:55:59 +00:00
import json
import os
Directly store entries using their potentially reduced path. Before, we were storing a entry using its full path and potentially create a redirect entry (using reduced path) pointing to the full path entry. Now, path reduction is part of normalization and so we directly store entries using their (potentially) reduced path. Converted `test/data/video-vimeo.warc.gz` goes from : ``` A/404.html A/index.html A/load.js A/sw.js A/topFrame.html H/f.vimeocdn.com/js_opt/modules/utils/vuid.min.js H/f.vimeocdn.com/p/3.45.3/css/player.css H/f.vimeocdn.com/p/3.45.3/js/player.js H/i.vimeocdn.com/player/354746.png?mw=200&mh=200 H/i.vimeocdn.com/video/797382244-0106ae13e902e09d0f02d8f404fa80581f38d1b8b7846b3f8e87ef391ffb8c99-d.jpg?mw=80&q=85 H/i.vimeocdn.com/video/797382244-0106ae13e902e09d0f02d8f404fa80581f38d1b8b7846b3f8e87ef391ffb8c99-d?mw=1280&mh=720&q=70 H/oembed.link/favicon.ico H/oembed.link/https://vimeo.com/347119375 H/player.vimeo.com/video/347119375?h=1699409fe2&app_id=122963 H/vod-progressive.akamaized.net/exp=1635528595~acl=%2Fvimeo-prod-skyfire-std-us%2F01%2F4423%2F13%2F347119375%2F1398505169.mp4~hmac=27c31f1990aab5e5429f7f7db5b2dcbcf8d2f5c92184d53102da36920d33d53e/vimeo-prod-skyfire-std-us/01/4423/13/347119375/1398505169.mp4 f.vimeocdn.com/js_opt/modules/utils/vuid.min.js f.vimeocdn.com/p/3.45.3/css/player.css f.vimeocdn.com/p/3.45.3/js/player.js i.vimeocdn.com/player/354746.png?mw=200&mh=200 i.vimeocdn.com/video/797382244-0106ae13e902e09d0f02d8f404fa80581f38d1b8b7846b3f8e87ef391ffb8c99-d.jpg?mw=80&q=85 i.vimeocdn.com/video/797382244-0106ae13e902e09d0f02d8f404fa80581f38d1b8b7846b3f8e87ef391ffb8c99-d?mw=1280&mh=720&q=70 oembed.link/favicon.ico oembed.link/https://vimeo.com/347119375 player.vimeo.com/video/347119375?h=1699409fe2&app_id=122963 vimeo-cdn.fuzzy.replayweb.page/01/4423/13/347119375/1398505169.mp4 vimeo.fuzzy.replayweb.page/video/347119375 vod-progressive.akamaized.net/exp=1635528595~acl=%2Fvimeo-prod-skyfire-std-us%2F01%2F4423%2F13%2F347119375%2F1398505169.mp4~hmac=27c31f1990aab5e5429f7f7db5b2dcbcf8d2f5c92184d53102da36920d33d53e/vimeo-prod-skyfire-std-us/01/4423/13/347119375/1398505169.mp4 ``` to : ``` A/404.html A/index.html A/load.js A/sw.js A/topFrame.html H/f.vimeocdn.com/js_opt/modules/utils/vuid.min.js H/f.vimeocdn.com/p/3.45.3/css/player.css H/f.vimeocdn.com/p/3.45.3/js/player.js H/i.vimeocdn.com/player/354746.png?mw=200&mh=200 H/i.vimeocdn.com/video/797382244-0106ae13e902e09d0f02d8f404fa80581f38d1b8b7846b3f8e87ef391ffb8c99-d.jpg?mw=80&q=85 H/i.vimeocdn.com/video/797382244-0106ae13e902e09d0f02d8f404fa80581f38d1b8b7846b3f8e87ef391ffb8c99-d?mw=1280&mh=720&q=70 H/oembed.link/favicon.ico H/oembed.link/https://vimeo.com/347119375 H/player.vimeo.com/video/347119375?h=1699409fe2&app_id=122963 H/vimeo-cdn.fuzzy.replayweb.page/01/4423/13/347119375/1398505169.mp4 f.vimeocdn.com/js_opt/modules/utils/vuid.min.js f.vimeocdn.com/p/3.45.3/css/player.css f.vimeocdn.com/p/3.45.3/js/player.js i.vimeocdn.com/player/354746.png?mw=200&mh=200 i.vimeocdn.com/video/797382244-0106ae13e902e09d0f02d8f404fa80581f38d1b8b7846b3f8e87ef391ffb8c99-d.jpg?mw=80&q=85 i.vimeocdn.com/video/797382244-0106ae13e902e09d0f02d8f404fa80581f38d1b8b7846b3f8e87ef391ffb8c99-d?mw=1280&mh=720&q=70 oembed.link/favicon.ico oembed.link/https://vimeo.com/347119375 player.vimeo.com/video/347119375?h=1699409fe2&app_id=122963 vimeo-cdn.fuzzy.replayweb.page/01/4423/13/347119375/1398505169.mp4 vimeo.fuzzy.replayweb.page/video/347119375 ``` Notice that `vod-progressive.akamaized.net` is not present. It is "replaced" by `vimeo-cdn.fuzzy.replayweb.page/01/4423/13/347119375/1398505169.mp4` which is now a plain entry instead of a redirect to `vod-progressive.akamaized.net[...]`.
2023-11-14 16:12:17 +01:00
import re
import time
from urllib.parse import unquote
import pytest
2021-01-12 16:17:46 +00:00
import requests
from zimscraperlib.image.convertion import convert_image, resize_image
from zimscraperlib.zim import Archive
from warc2zim.__about__ import __version__
from warc2zim.converter import iter_warc_records
2023-11-28 15:27:01 +01:00
from warc2zim.main import main
from warc2zim.url_rewriting import HttpUrl, ZimPath, normalize
from warc2zim.utils import get_record_url
TEST_DATA_DIR = os.path.join(os.path.dirname(os.path.realpath(__file__)), "data")
SCRAPER_SUFFIX = " + zimit x.y.z-devw"
# ============================================================================
CMDLINES = [
["example-response.warc"],
2020-12-09 10:55:59 +00:00
["example-response.warc", "--progress-file", "progress.json"],
["example-response.warc", "--scraper-suffix", SCRAPER_SUFFIX],
["example-revisit.warc.gz"],
[
"example-revisit.warc.gz",
"-u",
"http://example.iana.org/",
"--lang",
"eng",
],
2020-08-02 23:04:32 +00:00
[
"example-utf8.warc",
"-u",
"https://httpbin.org/anything/utf8=%E2%9C%93?query=test&a=b&1=%E2%9C%93",
],
["single-page-test.warc"],
]
@pytest.fixture(params=CMDLINES, ids=[" ".join(cmds) for cmds in CMDLINES])
def cmdline(request):
return request.param
# ============================================================================
FUZZYCHECKS = [
{
"filename": "video-yt.warc.gz",
"entries": [
Directly store entries using their potentially reduced path. Before, we were storing a entry using its full path and potentially create a redirect entry (using reduced path) pointing to the full path entry. Now, path reduction is part of normalization and so we directly store entries using their (potentially) reduced path. Converted `test/data/video-vimeo.warc.gz` goes from : ``` A/404.html A/index.html A/load.js A/sw.js A/topFrame.html H/f.vimeocdn.com/js_opt/modules/utils/vuid.min.js H/f.vimeocdn.com/p/3.45.3/css/player.css H/f.vimeocdn.com/p/3.45.3/js/player.js H/i.vimeocdn.com/player/354746.png?mw=200&mh=200 H/i.vimeocdn.com/video/797382244-0106ae13e902e09d0f02d8f404fa80581f38d1b8b7846b3f8e87ef391ffb8c99-d.jpg?mw=80&q=85 H/i.vimeocdn.com/video/797382244-0106ae13e902e09d0f02d8f404fa80581f38d1b8b7846b3f8e87ef391ffb8c99-d?mw=1280&mh=720&q=70 H/oembed.link/favicon.ico H/oembed.link/https://vimeo.com/347119375 H/player.vimeo.com/video/347119375?h=1699409fe2&app_id=122963 H/vod-progressive.akamaized.net/exp=1635528595~acl=%2Fvimeo-prod-skyfire-std-us%2F01%2F4423%2F13%2F347119375%2F1398505169.mp4~hmac=27c31f1990aab5e5429f7f7db5b2dcbcf8d2f5c92184d53102da36920d33d53e/vimeo-prod-skyfire-std-us/01/4423/13/347119375/1398505169.mp4 f.vimeocdn.com/js_opt/modules/utils/vuid.min.js f.vimeocdn.com/p/3.45.3/css/player.css f.vimeocdn.com/p/3.45.3/js/player.js i.vimeocdn.com/player/354746.png?mw=200&mh=200 i.vimeocdn.com/video/797382244-0106ae13e902e09d0f02d8f404fa80581f38d1b8b7846b3f8e87ef391ffb8c99-d.jpg?mw=80&q=85 i.vimeocdn.com/video/797382244-0106ae13e902e09d0f02d8f404fa80581f38d1b8b7846b3f8e87ef391ffb8c99-d?mw=1280&mh=720&q=70 oembed.link/favicon.ico oembed.link/https://vimeo.com/347119375 player.vimeo.com/video/347119375?h=1699409fe2&app_id=122963 vimeo-cdn.fuzzy.replayweb.page/01/4423/13/347119375/1398505169.mp4 vimeo.fuzzy.replayweb.page/video/347119375 vod-progressive.akamaized.net/exp=1635528595~acl=%2Fvimeo-prod-skyfire-std-us%2F01%2F4423%2F13%2F347119375%2F1398505169.mp4~hmac=27c31f1990aab5e5429f7f7db5b2dcbcf8d2f5c92184d53102da36920d33d53e/vimeo-prod-skyfire-std-us/01/4423/13/347119375/1398505169.mp4 ``` to : ``` A/404.html A/index.html A/load.js A/sw.js A/topFrame.html H/f.vimeocdn.com/js_opt/modules/utils/vuid.min.js H/f.vimeocdn.com/p/3.45.3/css/player.css H/f.vimeocdn.com/p/3.45.3/js/player.js H/i.vimeocdn.com/player/354746.png?mw=200&mh=200 H/i.vimeocdn.com/video/797382244-0106ae13e902e09d0f02d8f404fa80581f38d1b8b7846b3f8e87ef391ffb8c99-d.jpg?mw=80&q=85 H/i.vimeocdn.com/video/797382244-0106ae13e902e09d0f02d8f404fa80581f38d1b8b7846b3f8e87ef391ffb8c99-d?mw=1280&mh=720&q=70 H/oembed.link/favicon.ico H/oembed.link/https://vimeo.com/347119375 H/player.vimeo.com/video/347119375?h=1699409fe2&app_id=122963 H/vimeo-cdn.fuzzy.replayweb.page/01/4423/13/347119375/1398505169.mp4 f.vimeocdn.com/js_opt/modules/utils/vuid.min.js f.vimeocdn.com/p/3.45.3/css/player.css f.vimeocdn.com/p/3.45.3/js/player.js i.vimeocdn.com/player/354746.png?mw=200&mh=200 i.vimeocdn.com/video/797382244-0106ae13e902e09d0f02d8f404fa80581f38d1b8b7846b3f8e87ef391ffb8c99-d.jpg?mw=80&q=85 i.vimeocdn.com/video/797382244-0106ae13e902e09d0f02d8f404fa80581f38d1b8b7846b3f8e87ef391ffb8c99-d?mw=1280&mh=720&q=70 oembed.link/favicon.ico oembed.link/https://vimeo.com/347119375 player.vimeo.com/video/347119375?h=1699409fe2&app_id=122963 vimeo-cdn.fuzzy.replayweb.page/01/4423/13/347119375/1398505169.mp4 vimeo.fuzzy.replayweb.page/video/347119375 ``` Notice that `vod-progressive.akamaized.net` is not present. It is "replaced" by `vimeo-cdn.fuzzy.replayweb.page/01/4423/13/347119375/1398505169.mp4` which is now a plain entry instead of a redirect to `vod-progressive.akamaized.net[...]`.
2023-11-14 16:12:17 +01:00
"youtube.fuzzy.replayweb.page/get_video_info?video_id=aT-Up5Y4uRI",
"youtube.fuzzy.replayweb.page/videoplayback?id=o-AE3bg3qVNY-gAWwYgL52vgpHKJe9ijdbu2eciNi5Uo_w",
2021-10-29 16:55:45 +00:00
],
},
{
"filename": "video-yt-2.warc.gz",
"entries": [
Directly store entries using their potentially reduced path. Before, we were storing a entry using its full path and potentially create a redirect entry (using reduced path) pointing to the full path entry. Now, path reduction is part of normalization and so we directly store entries using their (potentially) reduced path. Converted `test/data/video-vimeo.warc.gz` goes from : ``` A/404.html A/index.html A/load.js A/sw.js A/topFrame.html H/f.vimeocdn.com/js_opt/modules/utils/vuid.min.js H/f.vimeocdn.com/p/3.45.3/css/player.css H/f.vimeocdn.com/p/3.45.3/js/player.js H/i.vimeocdn.com/player/354746.png?mw=200&mh=200 H/i.vimeocdn.com/video/797382244-0106ae13e902e09d0f02d8f404fa80581f38d1b8b7846b3f8e87ef391ffb8c99-d.jpg?mw=80&q=85 H/i.vimeocdn.com/video/797382244-0106ae13e902e09d0f02d8f404fa80581f38d1b8b7846b3f8e87ef391ffb8c99-d?mw=1280&mh=720&q=70 H/oembed.link/favicon.ico H/oembed.link/https://vimeo.com/347119375 H/player.vimeo.com/video/347119375?h=1699409fe2&app_id=122963 H/vod-progressive.akamaized.net/exp=1635528595~acl=%2Fvimeo-prod-skyfire-std-us%2F01%2F4423%2F13%2F347119375%2F1398505169.mp4~hmac=27c31f1990aab5e5429f7f7db5b2dcbcf8d2f5c92184d53102da36920d33d53e/vimeo-prod-skyfire-std-us/01/4423/13/347119375/1398505169.mp4 f.vimeocdn.com/js_opt/modules/utils/vuid.min.js f.vimeocdn.com/p/3.45.3/css/player.css f.vimeocdn.com/p/3.45.3/js/player.js i.vimeocdn.com/player/354746.png?mw=200&mh=200 i.vimeocdn.com/video/797382244-0106ae13e902e09d0f02d8f404fa80581f38d1b8b7846b3f8e87ef391ffb8c99-d.jpg?mw=80&q=85 i.vimeocdn.com/video/797382244-0106ae13e902e09d0f02d8f404fa80581f38d1b8b7846b3f8e87ef391ffb8c99-d?mw=1280&mh=720&q=70 oembed.link/favicon.ico oembed.link/https://vimeo.com/347119375 player.vimeo.com/video/347119375?h=1699409fe2&app_id=122963 vimeo-cdn.fuzzy.replayweb.page/01/4423/13/347119375/1398505169.mp4 vimeo.fuzzy.replayweb.page/video/347119375 vod-progressive.akamaized.net/exp=1635528595~acl=%2Fvimeo-prod-skyfire-std-us%2F01%2F4423%2F13%2F347119375%2F1398505169.mp4~hmac=27c31f1990aab5e5429f7f7db5b2dcbcf8d2f5c92184d53102da36920d33d53e/vimeo-prod-skyfire-std-us/01/4423/13/347119375/1398505169.mp4 ``` to : ``` A/404.html A/index.html A/load.js A/sw.js A/topFrame.html H/f.vimeocdn.com/js_opt/modules/utils/vuid.min.js H/f.vimeocdn.com/p/3.45.3/css/player.css H/f.vimeocdn.com/p/3.45.3/js/player.js H/i.vimeocdn.com/player/354746.png?mw=200&mh=200 H/i.vimeocdn.com/video/797382244-0106ae13e902e09d0f02d8f404fa80581f38d1b8b7846b3f8e87ef391ffb8c99-d.jpg?mw=80&q=85 H/i.vimeocdn.com/video/797382244-0106ae13e902e09d0f02d8f404fa80581f38d1b8b7846b3f8e87ef391ffb8c99-d?mw=1280&mh=720&q=70 H/oembed.link/favicon.ico H/oembed.link/https://vimeo.com/347119375 H/player.vimeo.com/video/347119375?h=1699409fe2&app_id=122963 H/vimeo-cdn.fuzzy.replayweb.page/01/4423/13/347119375/1398505169.mp4 f.vimeocdn.com/js_opt/modules/utils/vuid.min.js f.vimeocdn.com/p/3.45.3/css/player.css f.vimeocdn.com/p/3.45.3/js/player.js i.vimeocdn.com/player/354746.png?mw=200&mh=200 i.vimeocdn.com/video/797382244-0106ae13e902e09d0f02d8f404fa80581f38d1b8b7846b3f8e87ef391ffb8c99-d.jpg?mw=80&q=85 i.vimeocdn.com/video/797382244-0106ae13e902e09d0f02d8f404fa80581f38d1b8b7846b3f8e87ef391ffb8c99-d?mw=1280&mh=720&q=70 oembed.link/favicon.ico oembed.link/https://vimeo.com/347119375 player.vimeo.com/video/347119375?h=1699409fe2&app_id=122963 vimeo-cdn.fuzzy.replayweb.page/01/4423/13/347119375/1398505169.mp4 vimeo.fuzzy.replayweb.page/video/347119375 ``` Notice that `vod-progressive.akamaized.net` is not present. It is "replaced" by `vimeo-cdn.fuzzy.replayweb.page/01/4423/13/347119375/1398505169.mp4` which is now a plain entry instead of a redirect to `vod-progressive.akamaized.net[...]`.
2023-11-14 16:12:17 +01:00
"youtube.fuzzy.replayweb.page/youtubei/v1/player?videoId=aT-Up5Y4uRI",
"youtube.fuzzy.replayweb.page/videoplayback?id=o-AGDtIqpFRmvgVVZk96wgGyFxL_SFSdpBxs0iBHatQpRD",
2021-10-29 16:55:45 +00:00
],
},
{
"filename": "video-vimeo.warc.gz",
"entries": [
Directly store entries using their potentially reduced path. Before, we were storing a entry using its full path and potentially create a redirect entry (using reduced path) pointing to the full path entry. Now, path reduction is part of normalization and so we directly store entries using their (potentially) reduced path. Converted `test/data/video-vimeo.warc.gz` goes from : ``` A/404.html A/index.html A/load.js A/sw.js A/topFrame.html H/f.vimeocdn.com/js_opt/modules/utils/vuid.min.js H/f.vimeocdn.com/p/3.45.3/css/player.css H/f.vimeocdn.com/p/3.45.3/js/player.js H/i.vimeocdn.com/player/354746.png?mw=200&mh=200 H/i.vimeocdn.com/video/797382244-0106ae13e902e09d0f02d8f404fa80581f38d1b8b7846b3f8e87ef391ffb8c99-d.jpg?mw=80&q=85 H/i.vimeocdn.com/video/797382244-0106ae13e902e09d0f02d8f404fa80581f38d1b8b7846b3f8e87ef391ffb8c99-d?mw=1280&mh=720&q=70 H/oembed.link/favicon.ico H/oembed.link/https://vimeo.com/347119375 H/player.vimeo.com/video/347119375?h=1699409fe2&app_id=122963 H/vod-progressive.akamaized.net/exp=1635528595~acl=%2Fvimeo-prod-skyfire-std-us%2F01%2F4423%2F13%2F347119375%2F1398505169.mp4~hmac=27c31f1990aab5e5429f7f7db5b2dcbcf8d2f5c92184d53102da36920d33d53e/vimeo-prod-skyfire-std-us/01/4423/13/347119375/1398505169.mp4 f.vimeocdn.com/js_opt/modules/utils/vuid.min.js f.vimeocdn.com/p/3.45.3/css/player.css f.vimeocdn.com/p/3.45.3/js/player.js i.vimeocdn.com/player/354746.png?mw=200&mh=200 i.vimeocdn.com/video/797382244-0106ae13e902e09d0f02d8f404fa80581f38d1b8b7846b3f8e87ef391ffb8c99-d.jpg?mw=80&q=85 i.vimeocdn.com/video/797382244-0106ae13e902e09d0f02d8f404fa80581f38d1b8b7846b3f8e87ef391ffb8c99-d?mw=1280&mh=720&q=70 oembed.link/favicon.ico oembed.link/https://vimeo.com/347119375 player.vimeo.com/video/347119375?h=1699409fe2&app_id=122963 vimeo-cdn.fuzzy.replayweb.page/01/4423/13/347119375/1398505169.mp4 vimeo.fuzzy.replayweb.page/video/347119375 vod-progressive.akamaized.net/exp=1635528595~acl=%2Fvimeo-prod-skyfire-std-us%2F01%2F4423%2F13%2F347119375%2F1398505169.mp4~hmac=27c31f1990aab5e5429f7f7db5b2dcbcf8d2f5c92184d53102da36920d33d53e/vimeo-prod-skyfire-std-us/01/4423/13/347119375/1398505169.mp4 ``` to : ``` A/404.html A/index.html A/load.js A/sw.js A/topFrame.html H/f.vimeocdn.com/js_opt/modules/utils/vuid.min.js H/f.vimeocdn.com/p/3.45.3/css/player.css H/f.vimeocdn.com/p/3.45.3/js/player.js H/i.vimeocdn.com/player/354746.png?mw=200&mh=200 H/i.vimeocdn.com/video/797382244-0106ae13e902e09d0f02d8f404fa80581f38d1b8b7846b3f8e87ef391ffb8c99-d.jpg?mw=80&q=85 H/i.vimeocdn.com/video/797382244-0106ae13e902e09d0f02d8f404fa80581f38d1b8b7846b3f8e87ef391ffb8c99-d?mw=1280&mh=720&q=70 H/oembed.link/favicon.ico H/oembed.link/https://vimeo.com/347119375 H/player.vimeo.com/video/347119375?h=1699409fe2&app_id=122963 H/vimeo-cdn.fuzzy.replayweb.page/01/4423/13/347119375/1398505169.mp4 f.vimeocdn.com/js_opt/modules/utils/vuid.min.js f.vimeocdn.com/p/3.45.3/css/player.css f.vimeocdn.com/p/3.45.3/js/player.js i.vimeocdn.com/player/354746.png?mw=200&mh=200 i.vimeocdn.com/video/797382244-0106ae13e902e09d0f02d8f404fa80581f38d1b8b7846b3f8e87ef391ffb8c99-d.jpg?mw=80&q=85 i.vimeocdn.com/video/797382244-0106ae13e902e09d0f02d8f404fa80581f38d1b8b7846b3f8e87ef391ffb8c99-d?mw=1280&mh=720&q=70 oembed.link/favicon.ico oembed.link/https://vimeo.com/347119375 player.vimeo.com/video/347119375?h=1699409fe2&app_id=122963 vimeo-cdn.fuzzy.replayweb.page/01/4423/13/347119375/1398505169.mp4 vimeo.fuzzy.replayweb.page/video/347119375 ``` Notice that `vod-progressive.akamaized.net` is not present. It is "replaced" by `vimeo-cdn.fuzzy.replayweb.page/01/4423/13/347119375/1398505169.mp4` which is now a plain entry instead of a redirect to `vod-progressive.akamaized.net[...]`.
2023-11-14 16:12:17 +01:00
"vimeo.fuzzy.replayweb.page/video/347119375",
"vimeo-cdn.fuzzy.replayweb.page/01/4423/13/347119375/1398505169.mp4",
2021-10-29 16:55:45 +00:00
],
},
]
@pytest.fixture(params=FUZZYCHECKS, ids=[fuzzy["filename"] for fuzzy in FUZZYCHECKS])
def fuzzycheck(request):
return request.param
# ============================================================================
class TestWarc2Zim:
def list_articles(self, zimfile):
zim_fh = Archive(zimfile)
for x in range(zim_fh.entry_count):
yield zim_fh.get_entry_by_id(x)
def get_main_entry_with_redirect(self, zimfile):
zim_fh = Archive(zimfile)
if zim_fh.main_entry.is_redirect:
return zim_fh.main_entry.get_redirect_entry()
return zim_fh.main_entry
def get_metadata(self, zimfile, name):
zim_fh = Archive(zimfile)
return zim_fh.get_metadata(name)
def get_article(self, zimfile, path):
zim_fh = Archive(zimfile)
return zim_fh.get_content(path)
def get_article_raw(self, zimfile, path):
zim_fh = Archive(zimfile)
return zim_fh.get_item(path)
2024-04-30 14:24:58 +00:00
def assert_item_exist(self, zimfile, path):
zim_fh = Archive(zimfile)
assert zim_fh.get_item(path)
def assert_item_does_not_exist(self, zimfile, path):
zim_fh = Archive(zimfile)
try:
payload = zim_fh.get_item(path)
except KeyError:
payload = None
assert payload is None
def verify_warc_and_zim(self, warcfile, zimfile, verify_scraper_suffix):
assert os.path.isfile(warcfile)
assert os.path.isfile(zimfile)
# [TOFIX]
head_insert = b""
# track to avoid checking duplicates, which are not written to ZIM
warc_urls = set()
zim_fh = Archive(zimfile)
if verify_scraper_suffix:
assert (
f"warc2zim {__version__}{SCRAPER_SUFFIX}"
== zim_fh.get_text_metadata("Scraper")
)
else:
assert f"warc2zim {__version__}" == zim_fh.get_text_metadata("Scraper")
for record in iter_warc_records([warcfile]):
url = get_record_url(record)
if not url:
continue
if url in warc_urls:
continue
if record.rec_type not in (("response", "resource", "revisit")):
continue
# ignore revisit records that are to the same url
if (
record.rec_type == "revisit"
and record.rec_headers["WARC-Refers-To-Target-URI"] == url
):
continue
# parse headers as record, ensure headers match
url_no_scheme = url.split("//", 2)[1]
Directly store entries using their potentially reduced path. Before, we were storing a entry using its full path and potentially create a redirect entry (using reduced path) pointing to the full path entry. Now, path reduction is part of normalization and so we directly store entries using their (potentially) reduced path. Converted `test/data/video-vimeo.warc.gz` goes from : ``` A/404.html A/index.html A/load.js A/sw.js A/topFrame.html H/f.vimeocdn.com/js_opt/modules/utils/vuid.min.js H/f.vimeocdn.com/p/3.45.3/css/player.css H/f.vimeocdn.com/p/3.45.3/js/player.js H/i.vimeocdn.com/player/354746.png?mw=200&mh=200 H/i.vimeocdn.com/video/797382244-0106ae13e902e09d0f02d8f404fa80581f38d1b8b7846b3f8e87ef391ffb8c99-d.jpg?mw=80&q=85 H/i.vimeocdn.com/video/797382244-0106ae13e902e09d0f02d8f404fa80581f38d1b8b7846b3f8e87ef391ffb8c99-d?mw=1280&mh=720&q=70 H/oembed.link/favicon.ico H/oembed.link/https://vimeo.com/347119375 H/player.vimeo.com/video/347119375?h=1699409fe2&app_id=122963 H/vod-progressive.akamaized.net/exp=1635528595~acl=%2Fvimeo-prod-skyfire-std-us%2F01%2F4423%2F13%2F347119375%2F1398505169.mp4~hmac=27c31f1990aab5e5429f7f7db5b2dcbcf8d2f5c92184d53102da36920d33d53e/vimeo-prod-skyfire-std-us/01/4423/13/347119375/1398505169.mp4 f.vimeocdn.com/js_opt/modules/utils/vuid.min.js f.vimeocdn.com/p/3.45.3/css/player.css f.vimeocdn.com/p/3.45.3/js/player.js i.vimeocdn.com/player/354746.png?mw=200&mh=200 i.vimeocdn.com/video/797382244-0106ae13e902e09d0f02d8f404fa80581f38d1b8b7846b3f8e87ef391ffb8c99-d.jpg?mw=80&q=85 i.vimeocdn.com/video/797382244-0106ae13e902e09d0f02d8f404fa80581f38d1b8b7846b3f8e87ef391ffb8c99-d?mw=1280&mh=720&q=70 oembed.link/favicon.ico oembed.link/https://vimeo.com/347119375 player.vimeo.com/video/347119375?h=1699409fe2&app_id=122963 vimeo-cdn.fuzzy.replayweb.page/01/4423/13/347119375/1398505169.mp4 vimeo.fuzzy.replayweb.page/video/347119375 vod-progressive.akamaized.net/exp=1635528595~acl=%2Fvimeo-prod-skyfire-std-us%2F01%2F4423%2F13%2F347119375%2F1398505169.mp4~hmac=27c31f1990aab5e5429f7f7db5b2dcbcf8d2f5c92184d53102da36920d33d53e/vimeo-prod-skyfire-std-us/01/4423/13/347119375/1398505169.mp4 ``` to : ``` A/404.html A/index.html A/load.js A/sw.js A/topFrame.html H/f.vimeocdn.com/js_opt/modules/utils/vuid.min.js H/f.vimeocdn.com/p/3.45.3/css/player.css H/f.vimeocdn.com/p/3.45.3/js/player.js H/i.vimeocdn.com/player/354746.png?mw=200&mh=200 H/i.vimeocdn.com/video/797382244-0106ae13e902e09d0f02d8f404fa80581f38d1b8b7846b3f8e87ef391ffb8c99-d.jpg?mw=80&q=85 H/i.vimeocdn.com/video/797382244-0106ae13e902e09d0f02d8f404fa80581f38d1b8b7846b3f8e87ef391ffb8c99-d?mw=1280&mh=720&q=70 H/oembed.link/favicon.ico H/oembed.link/https://vimeo.com/347119375 H/player.vimeo.com/video/347119375?h=1699409fe2&app_id=122963 H/vimeo-cdn.fuzzy.replayweb.page/01/4423/13/347119375/1398505169.mp4 f.vimeocdn.com/js_opt/modules/utils/vuid.min.js f.vimeocdn.com/p/3.45.3/css/player.css f.vimeocdn.com/p/3.45.3/js/player.js i.vimeocdn.com/player/354746.png?mw=200&mh=200 i.vimeocdn.com/video/797382244-0106ae13e902e09d0f02d8f404fa80581f38d1b8b7846b3f8e87ef391ffb8c99-d.jpg?mw=80&q=85 i.vimeocdn.com/video/797382244-0106ae13e902e09d0f02d8f404fa80581f38d1b8b7846b3f8e87ef391ffb8c99-d?mw=1280&mh=720&q=70 oembed.link/favicon.ico oembed.link/https://vimeo.com/347119375 player.vimeo.com/video/347119375?h=1699409fe2&app_id=122963 vimeo-cdn.fuzzy.replayweb.page/01/4423/13/347119375/1398505169.mp4 vimeo.fuzzy.replayweb.page/video/347119375 ``` Notice that `vod-progressive.akamaized.net` is not present. It is "replaced" by `vimeo-cdn.fuzzy.replayweb.page/01/4423/13/347119375/1398505169.mp4` which is now a plain entry instead of a redirect to `vod-progressive.akamaized.net[...]`.
2023-11-14 16:12:17 +01:00
if "www.youtube.com/embed" in url_no_scheme:
# We know that those url are rewritten in zim. Don't check for them.
break
url_no_scheme = re.sub(r"\?\d+$", "?", url_no_scheme)
# remove user/password
if "@" in url_no_scheme:
at_index = url_no_scheme.index("@")
if at_index >= 0:
if "/" in url_no_scheme:
slash_index = url_no_scheme.index("/")
if at_index < slash_index:
url_no_scheme = url_no_scheme[at_index + 1 :]
else:
url_no_scheme = url_no_scheme[at_index + 1 :]
# remove trailing ?
if url_no_scheme.endswith("?"):
url_no_scheme = url_no_scheme[:-1]
# unquote url since everything is not encoded in ZIM
url_no_scheme = unquote(url_no_scheme)
# ensure payloads match
try:
payload = zim_fh.get_item(url_no_scheme)
except KeyError:
payload = None
if record.http_headers and record.http_headers.get("Content-Length") == "0":
2024-04-30 14:24:58 +00:00
if record.http_headers.get("Location"):
assert payload # this is a redirect, it must be handled
else:
assert not payload
elif record.rec_type == "revisit":
# We must have a payload
# We should check with the content of the targeted record...
# But difficult to test as we don't have it
assert payload
elif record.rec_type == "response":
# We must have a payload
assert payload
payload_content = payload.content.tobytes()
# if HTML, still need to account for the head insert, otherwise should
# have exact match
2023-02-02 15:32:10 +00:00
if payload.mimetype.startswith("text/html"):
assert head_insert in payload_content
elif record.rec_type == "resource":
# we do not want to embed resources "as-is"
assert not payload
warc_urls.add(url)
def rebuild_favicon_bytes(self, zim, favicon_path) -> bytes:
favicon_bytes = self.get_article(
zim,
favicon_path,
)
assert favicon_bytes
dst = io.BytesIO()
convert_image(
io.BytesIO(
favicon_bytes
), # pyright: ignore[reportGeneralTypeIssues, reportArgumentType]
dst, # pyright: ignore[reportGeneralTypeIssues, reportArgumentType]
fmt="PNG", # pyright: ignore[reportGeneralTypeIssues, reportArgumentType]
)
resize_image(dst, width=48, height=48, method="cover")
return dst.getvalue()
@pytest.mark.parametrize(
"url,zim_path",
[
("https://exemple.com", "exemple.com/"),
("https://exemple.com/", "exemple.com/"),
("http://example.com/resource", "example.com/resource"),
("http://example.com/resource/", "example.com/resource/"),
(
"http://example.com/resource/folder/sub.txt",
"example.com/resource/folder/sub.txt",
),
(
"http://example.com/resource/folder/sub",
"example.com/resource/folder/sub",
),
(
"http://example.com/resource/folder/sub?foo=bar",
"example.com/resource/folder/sub?foo=bar",
),
(
"http://example.com/resource/folder/sub?foo=bar#anchor1",
"example.com/resource/folder/sub?foo=bar",
),
("http://example.com/resource/#anchor1", "example.com/resource/"),
("http://example.com/resource/?foo=bar", "example.com/resource/?foo=bar"),
("http://example.com#anchor1", "example.com/"),
("http://example.com?foo=bar#anchor1", "example.com/?foo=bar"),
("http://example.com/?foo=bar", "example.com/?foo=bar"),
("http://example.com/?foo=ba+r", "example.com/?foo=ba r"),
(
"http://example.com/?foo=ba r",
"example.com/?foo=ba r",
), # situation where the ` ` has not been properly escaped in document
("http://example.com/?foo=ba%2Br", "example.com/?foo=ba+r"),
("http://example.com/?foo=ba+%2B+r", "example.com/?foo=ba + r"),
("http://example.com/#anchor1", "example.com/"),
(
"http://example.com/some/path/http://example.com//some/path",
"example.com/some/path/http://example.com//some/path",
),
(
"http://example.com/some/pa?th/http://example.com//some/path",
"example.com/some/pa?th/http://example.com//some/path",
),
(
"http://example.com/so?me/pa?th/http://example.com//some/path",
"example.com/so?me/pa?th/http://example.com//some/path",
),
("http://example.com/resource?", "example.com/resource"),
("http://example.com/resource#", "example.com/resource"),
("http://user@example.com/resource", "example.com/resource"),
("http://user:password@example.com/resource", "example.com/resource"),
("http://example.com:8080/resource", "example.com/resource"),
(
"http://foobargooglevideo.com/videoplayback?id=1576&key=value",
"youtube.fuzzy.replayweb.page/videoplayback?id=1576",
), # Fuzzy rule is applied in addition to path transformations
("https://xn--exmple-cva.com", "exémple.com/"),
("https://xn--exmple-cva.com/", "exémple.com/"),
("https://xn--exmple-cva.com/resource", "exémple.com/resource"),
("https://exémple.com/", "exémple.com/"),
("https://exémple.com/resource", "exémple.com/resource"),
# host_ip is an invalid hostname according to spec
("https://host_ip/", "host_ip/"),
("https://host_ip/resource", "host_ip/resource"),
("https://192.168.1.1/", "192.168.1.1/"),
("https://192.168.1.1/resource", "192.168.1.1/resource"),
("http://example.com/res%24urce", "example.com/res$urce"),
(
"http://example.com/resource?foo=b%24r",
"example.com/resource?foo=b$r",
),
("http://example.com/resource@300x", "example.com/resource@300x"),
("http://example.com:8080/resource", "example.com/resource"),
("http://user@example.com:8080/resource", "example.com/resource"),
("http://user:password@example.com:8080/resource", "example.com/resource"),
# the two URI below are an illustration of a potential collision (two
# differents URI leading to the same ZIM path)
(
"http://tmp.kiwix.org/ci/test-website/images/urlencoding1_ico%CC%82ne-"
"de%CC%81buter-Solidarite%CC%81-Nume%CC%81rique_1%40300x.png",
"tmp.kiwix.org/ci/test-website/images/urlencoding1_icône-débuter-"
"Solidarité-Numérique_1@300x.png",
),
(
"https://tmp.kiwix.org/ci/test-website/images/urlencoding1_ico%CC%82ne-"
"de%CC%81buter-Solidarite%CC%81-Nume%CC%81rique_1@300x.png",
"tmp.kiwix.org/ci/test-website/images/urlencoding1_icône-débuter-"
"Solidarité-Numérique_1@300x.png",
),
],
)
def test_normalize(self, url, zim_path):
assert normalize(HttpUrl(url)) == ZimPath(zim_path)
def test_warc_to_zim_specify_params_and_metadata(self, tmp_path):
zim_output = "zim-out-filename.zim"
2023-11-28 15:27:01 +01:00
main(
[
"-v",
os.path.join(TEST_DATA_DIR, "example-response.warc"),
"--name",
"example-response",
"--output",
str(tmp_path),
"--zim-file",
zim_output,
"--tags",
" foo ;bar; ; some;_foo:bar;_foo_,_bar_",
"--desc",
"test zim",
"--title",
"Some Title",
]
)
zim_output = tmp_path / zim_output
assert os.path.isfile(zim_output)
all_articles = {
article.path: article.title for article in self.list_articles(zim_output)
}
assert all_articles == {
# entries from WARC
"example.com/": "Example Domain",
"_zim_static/__wb_module_decl.js": "_zim_static/__wb_module_decl.js",
"_zim_static/wombat.js": "_zim_static/wombat.js",
"_zim_static/wombatSetup.js": "_zim_static/wombatSetup.js",
}
zim_fh = Archive(zim_output)
# ZIM metadata
assert list(zim_fh.metadata.keys()) == [
"Counter",
"Creator",
"Date",
"Description",
"Language",
"Name",
"Publisher",
"Scraper",
"Tags",
"Title",
]
assert zim_fh.has_fulltext_index
assert zim_fh.has_title_index
assert self.get_metadata(zim_output, "Description") == b"test zim"
# we compare sets of tags since tags ordering has no meaning
assert set(
self.get_metadata(zim_output, "Tags").decode("utf-8").split(";")
) == {
"_ftindex:yes",
"_category:other",
"some",
"foo",
"bar",
"_foo:bar",
"_foo_,_bar_",
}
assert self.get_metadata(zim_output, "Title") == b"Some Title"
def test_warc_to_zim_main(self, cmdline, tmp_path):
# intput filename
filename = cmdline[0]
# set intput filename (first arg) to absolute path from test dir
warcfile = os.path.join(TEST_DATA_DIR, filename)
cmdline[0] = warcfile
cmdline.extend(["--output", str(tmp_path), "--name", filename])
2023-11-28 15:27:01 +01:00
main(cmdline)
zimfile = filename + "_" + time.strftime("%Y-%m") + ".zim"
2020-12-09 10:55:59 +00:00
if "--progress-file" in cmdline:
with open(tmp_path / "progress.json") as fh:
2020-12-09 10:55:59 +00:00
progress = json.load(fh)
assert (
progress["written"] > 0
and progress["total"] > 0
and progress["written"] <= progress["total"]
)
self.verify_warc_and_zim(
warcfile, tmp_path / zimfile, "--scraper-suffix" in cmdline
)
def test_same_domain_only(self, tmp_path):
zim_output = "same-domain.zim"
2023-11-28 15:27:01 +01:00
main(
[
os.path.join(TEST_DATA_DIR, "example-revisit.warc.gz"),
"--favicon",
"http://example.com/favicon.ico",
"--include-domains",
"example.com/",
"--lang",
"eng",
"--zim-file",
zim_output,
"--name",
"same-domain",
"--output",
str(tmp_path),
]
)
zim_output = tmp_path / zim_output
for article in self.list_articles(zim_output):
url = article.path
# ignore the replay files, which have only one path segment
if not url.startswith("_zim_static/"):
assert url.startswith("example.com/")
def test_skip_self_redirect(self, tmp_path):
zim_output = "self-redir.zim"
2023-11-28 15:27:01 +01:00
main(
[
os.path.join(TEST_DATA_DIR, "self-redirect.warc"),
"--output",
str(tmp_path),
"--zim-file",
zim_output,
"--name",
"self-redir",
]
)
zim_output = tmp_path / zim_output
def test_include_domains_favicon_and_language(self, tmp_path):
zim_output = "spt.zim"
2023-11-28 15:27:01 +01:00
main(
[
os.path.join(TEST_DATA_DIR, "single-page-test.warc"),
"-i",
"reseau-canope.fr",
"--output",
str(tmp_path),
"--zim-file",
zim_output,
"--name",
"spt",
]
)
zim_output = tmp_path / zim_output
for article in self.list_articles(zim_output):
url = article.path
# ignore the replay files, which have only one path segment
if not url.startswith("_zim_static/"):
assert "reseau-canope.fr/" in url
# test detected language
assert self.get_metadata(zim_output, "Language") == b"fra"
# test detected favicon
zim_favicon = self.get_metadata(zim_output, "Illustration_48x48@1")
assert zim_favicon
assert (
self.rebuild_favicon_bytes(
zim_output,
"lesfondamentaux.reseau-canope.fr/fileadmin/template/img/favicon.ico",
)
== zim_favicon
)
# test default tags added ; we compare sets of tags since tags ordering has no
# meaning
assert set(
self.get_metadata(zim_output, "Tags").decode("utf-8").split(";")
) == {
"_ftindex:yes",
"_category:other",
}
def test_website_with_redirect(self, tmp_path):
zim_output = "kiwix.zim"
main(
[
os.path.join(TEST_DATA_DIR, "kiwix-with-redirects.warc.gz"),
"-u",
"http://www.kiwix.org",
"--output",
str(tmp_path),
"--zim-file",
zim_output,
"--name",
"kiwix",
]
)
zim_output = tmp_path / zim_output
# check that redirections have been followed
assert self.get_main_entry_with_redirect(zim_output).path == "kiwix.org/en/"
# test detected language
assert self.get_metadata(zim_output, "Language") == b"eng"
# test detected favicon
zim_favicon = self.get_metadata(zim_output, "Illustration_48x48@1")
assert zim_favicon
assert (
self.rebuild_favicon_bytes(
zim_output,
"kiwix.org/favicon.ico",
)
== zim_favicon
)
def test_all_warcs_root_dir(self, tmp_path):
zim_output = "test-all.zim"
2023-11-28 15:27:01 +01:00
main(
[
os.path.join(TEST_DATA_DIR),
"--output",
str(tmp_path),
"--zim-file",
zim_output,
"--name",
"test-all",
"--url",
"http://example.com",
]
)
zim_output = tmp_path / zim_output
# check articles from different warc records in tests/data dir
# from example.warc.gz
assert self.get_article(zim_output, "example.com/") != b""
# from single-page-test.warc
assert (
self.get_article(
zim_output, "lesfondamentaux.reseau-canope.fr/accueil.html"
)
!= b""
)
# timestamp fuzzy match from example-with-timestamp.warc
assert self.get_article(zim_output, "example.com/path.txt") != b""
def test_fuzzy_urls(self, tmp_path, fuzzycheck):
zim_output = fuzzycheck["filename"] + ".zim"
2023-11-28 15:27:01 +01:00
main(
[
os.path.join(TEST_DATA_DIR, fuzzycheck["filename"]),
"--output",
str(tmp_path),
"--zim-file",
zim_output,
"--name",
"test-fuzzy",
]
)
zim_output = tmp_path / zim_output
2021-10-29 02:20:01 +00:00
for entry in fuzzycheck["entries"]:
# This should be item and get_article_raw is eq to getItem and it will fail
# if it is not a item
Directly store entries using their potentially reduced path. Before, we were storing a entry using its full path and potentially create a redirect entry (using reduced path) pointing to the full path entry. Now, path reduction is part of normalization and so we directly store entries using their (potentially) reduced path. Converted `test/data/video-vimeo.warc.gz` goes from : ``` A/404.html A/index.html A/load.js A/sw.js A/topFrame.html H/f.vimeocdn.com/js_opt/modules/utils/vuid.min.js H/f.vimeocdn.com/p/3.45.3/css/player.css H/f.vimeocdn.com/p/3.45.3/js/player.js H/i.vimeocdn.com/player/354746.png?mw=200&mh=200 H/i.vimeocdn.com/video/797382244-0106ae13e902e09d0f02d8f404fa80581f38d1b8b7846b3f8e87ef391ffb8c99-d.jpg?mw=80&q=85 H/i.vimeocdn.com/video/797382244-0106ae13e902e09d0f02d8f404fa80581f38d1b8b7846b3f8e87ef391ffb8c99-d?mw=1280&mh=720&q=70 H/oembed.link/favicon.ico H/oembed.link/https://vimeo.com/347119375 H/player.vimeo.com/video/347119375?h=1699409fe2&app_id=122963 H/vod-progressive.akamaized.net/exp=1635528595~acl=%2Fvimeo-prod-skyfire-std-us%2F01%2F4423%2F13%2F347119375%2F1398505169.mp4~hmac=27c31f1990aab5e5429f7f7db5b2dcbcf8d2f5c92184d53102da36920d33d53e/vimeo-prod-skyfire-std-us/01/4423/13/347119375/1398505169.mp4 f.vimeocdn.com/js_opt/modules/utils/vuid.min.js f.vimeocdn.com/p/3.45.3/css/player.css f.vimeocdn.com/p/3.45.3/js/player.js i.vimeocdn.com/player/354746.png?mw=200&mh=200 i.vimeocdn.com/video/797382244-0106ae13e902e09d0f02d8f404fa80581f38d1b8b7846b3f8e87ef391ffb8c99-d.jpg?mw=80&q=85 i.vimeocdn.com/video/797382244-0106ae13e902e09d0f02d8f404fa80581f38d1b8b7846b3f8e87ef391ffb8c99-d?mw=1280&mh=720&q=70 oembed.link/favicon.ico oembed.link/https://vimeo.com/347119375 player.vimeo.com/video/347119375?h=1699409fe2&app_id=122963 vimeo-cdn.fuzzy.replayweb.page/01/4423/13/347119375/1398505169.mp4 vimeo.fuzzy.replayweb.page/video/347119375 vod-progressive.akamaized.net/exp=1635528595~acl=%2Fvimeo-prod-skyfire-std-us%2F01%2F4423%2F13%2F347119375%2F1398505169.mp4~hmac=27c31f1990aab5e5429f7f7db5b2dcbcf8d2f5c92184d53102da36920d33d53e/vimeo-prod-skyfire-std-us/01/4423/13/347119375/1398505169.mp4 ``` to : ``` A/404.html A/index.html A/load.js A/sw.js A/topFrame.html H/f.vimeocdn.com/js_opt/modules/utils/vuid.min.js H/f.vimeocdn.com/p/3.45.3/css/player.css H/f.vimeocdn.com/p/3.45.3/js/player.js H/i.vimeocdn.com/player/354746.png?mw=200&mh=200 H/i.vimeocdn.com/video/797382244-0106ae13e902e09d0f02d8f404fa80581f38d1b8b7846b3f8e87ef391ffb8c99-d.jpg?mw=80&q=85 H/i.vimeocdn.com/video/797382244-0106ae13e902e09d0f02d8f404fa80581f38d1b8b7846b3f8e87ef391ffb8c99-d?mw=1280&mh=720&q=70 H/oembed.link/favicon.ico H/oembed.link/https://vimeo.com/347119375 H/player.vimeo.com/video/347119375?h=1699409fe2&app_id=122963 H/vimeo-cdn.fuzzy.replayweb.page/01/4423/13/347119375/1398505169.mp4 f.vimeocdn.com/js_opt/modules/utils/vuid.min.js f.vimeocdn.com/p/3.45.3/css/player.css f.vimeocdn.com/p/3.45.3/js/player.js i.vimeocdn.com/player/354746.png?mw=200&mh=200 i.vimeocdn.com/video/797382244-0106ae13e902e09d0f02d8f404fa80581f38d1b8b7846b3f8e87ef391ffb8c99-d.jpg?mw=80&q=85 i.vimeocdn.com/video/797382244-0106ae13e902e09d0f02d8f404fa80581f38d1b8b7846b3f8e87ef391ffb8c99-d?mw=1280&mh=720&q=70 oembed.link/favicon.ico oembed.link/https://vimeo.com/347119375 player.vimeo.com/video/347119375?h=1699409fe2&app_id=122963 vimeo-cdn.fuzzy.replayweb.page/01/4423/13/347119375/1398505169.mp4 vimeo.fuzzy.replayweb.page/video/347119375 ``` Notice that `vod-progressive.akamaized.net` is not present. It is "replaced" by `vimeo-cdn.fuzzy.replayweb.page/01/4423/13/347119375/1398505169.mp4` which is now a plain entry instead of a redirect to `vod-progressive.akamaized.net[...]`.
2023-11-14 16:12:17 +01:00
self.get_article_raw(zim_output, entry)
def test_error_bad_main_page(self, tmp_path):
zim_output_not_created = "zim-out-not-created.zim"
with pytest.raises(KeyError, match="Unable to find WARC record for main page:"):
2023-11-28 15:27:01 +01:00
main(
[
"-v",
os.path.join(TEST_DATA_DIR, "example-response.warc"),
"-u",
"https://no-such-url.example.com",
"--output",
str(tmp_path),
"--name",
"bad",
"--zim-file",
zim_output_not_created,
]
)
def test_args_only(self):
# error, name required
with pytest.raises(SystemExit) as e:
2023-11-28 15:27:01 +01:00
main([])
assert e.value.code == 2
# error, no such output directory
with pytest.raises(SystemExit) as e:
2023-11-28 15:27:01 +01:00
main(["--name", "test", "--output", "/no-such-dir"])
2024-05-24 13:53:12 +00:00
assert e.value.code == 1
2024-04-18 10:38:16 +05:30
# error, name has invalid characters for Linux filesystem
with pytest.raises(SystemExit) as e:
main(["--name", "te/st", "--output", "./"])
2024-05-24 13:53:12 +00:00
assert e.value.code == 3
2024-04-18 10:38:16 +05:30
2024-01-31 15:13:32 +01:00
# success, special return code for no output files
2023-11-28 15:27:01 +01:00
assert main(["--name", "test", "--output", "./"]) == 100
2021-01-12 16:17:46 +00:00
def test_custom_css(self, tmp_path):
custom_css = b"* { background-color: red; }"
custom_css_path = tmp_path / "custom.css"
with open(custom_css_path, "wb") as fh:
fh.write(custom_css)
zim_output = "test-css.zim"
2023-11-28 15:27:01 +01:00
main(
2021-01-12 16:17:46 +00:00
[
os.path.join(TEST_DATA_DIR, "example-response.warc"),
"--output",
str(tmp_path),
"--zim-file",
zim_output,
"--name",
"test-css",
"--custom-css",
str(custom_css_path),
]
)
zim_output = tmp_path / zim_output
res = self.get_article(zim_output, "example.com/")
assert b"static_prefix" not in res
assert b"../_zim_static/custom.css" in res
2021-01-12 16:17:46 +00:00
res = self.get_article(zim_output, "_zim_static/custom.css")
2021-01-12 16:17:46 +00:00
assert custom_css == res
def test_custom_css_remote(self, tmp_path):
zim_output = "test-css.zim"
url = (
"https://cdn.jsdelivr.net/npm/bootstrap@4.5.3/dist/css/bootstrap-reboot.css"
)
2023-11-28 15:27:01 +01:00
main(
2021-01-12 16:17:46 +00:00
[
os.path.join(TEST_DATA_DIR, "example-response.warc"),
"--output",
str(tmp_path),
"--zim-file",
zim_output,
"--name",
"test-css",
"--custom-css",
url,
]
)
zim_output = tmp_path / zim_output
res = self.get_article(zim_output, "example.com/")
assert b"static_prefix" not in res
assert b"../_zim_static/custom.css" in res
2021-01-12 16:17:46 +00:00
res = self.get_article(zim_output, "_zim_static/custom.css")
assert res == requests.get(url, timeout=10).content
2024-04-30 14:24:58 +00:00
def test_http_return_codes(self, tmp_path):
zim_output = "test-http-return-codes.zim"
2021-01-12 16:17:46 +00:00
2024-04-30 14:24:58 +00:00
main(
[
os.path.join(TEST_DATA_DIR, "http-return-codes.warc.gz"),
"--output",
str(tmp_path),
"--zim-file",
zim_output,
"--name",
"test-http-return-codes",
]
)
zim_output = tmp_path / zim_output
for exising_website_items in [
"200-response",
"201-response",
"202-response",
"301-internal-redirect-ok",
"301-external-redirect-ok",
"302-internal-redirect-ok",
"302-external-redirect-ok",
"307-internal-redirect-ok",
"307-external-redirect-ok",
"308-internal-redirect-ok",
"308-external-redirect-ok",
]:
self.assert_item_exist(
zim_output, f"website.test.openzim.org/{exising_website_items}"
)
self.assert_item_exist(zim_output, "www.example.com/")
for ignored_website_items in [
"204-response",
"206-response",
"300-response",
"303-response",
"304-response",
"305-response",
"306-response",
"400-response",
"401-response",
"402-response",
"403-response",
"404-response",
"500-response",
"501-response",
"502-response",
"301-internal-redirect-ko",
"301-external-redirect-ko",
"302-internal-redirect-ko",
"302-external-redirect-ko",
"307-internal-redirect-ko",
"307-external-redirect-ko",
"308-internal-redirect-ko",
"308-external-redirect-ko",
]:
self.assert_item_does_not_exist(
zim_output, f"website.test.openzim.org/{ignored_website_items}"
)
def test_redirection_loops(self, tmp_path):
zim_output = "test-redir-loops.zim"
main(
[
os.path.join(TEST_DATA_DIR, "redir-loops.warc.gz"),
"--output",
str(tmp_path),
"--zim-file",
zim_output,
"--name",
"test-redir-loops",
]
)
zim_output = tmp_path / zim_output
for exising_website_items in [
"redirection-loops.html",
]:
self.assert_item_exist(
zim_output, f"website.test.openzim.org/{exising_website_items}"
)
for ignored_website_items in [
"/bad-redir-loop-A",
"/bad-redir-loop-B",
"/bad-redir-loop-C",
"/bad-redir-loop-D",
]:
self.assert_item_does_not_exist(
zim_output, f"website.test.openzim.org/{ignored_website_items}"
)
def test_content_resource_types(self, tmp_path):
zim_output = "tests_en_content-resource-types.zim"
main(
[
os.path.join(TEST_DATA_DIR, "content-resource-types.warc.gz"),
"--output",
str(tmp_path),
"--zim-file",
zim_output,
"--name",
"tests_en_content-resource-types",
]
)
zim_output = tmp_path / zim_output
res = self.get_article(
zim_output, "website.test.openzim.org/content-types/index.html"
)
assert b"<!-- WB Insert -->" in res # simple check that rewriting has been done
for js_file in [
"website.test.openzim.org/content-types/script1.js",
"website.test.openzim.org/content-types/script2.js",
]:
res = self.get_article(zim_output, js_file)
assert b"wombat" in res # simple check that rewriting has been done