Procházet zdrojové kódy

tests: add test for mercury-parser

ttimasdf před 5 roky
rodič
revize
e3329be291
2 změnil soubory, kde provedl 9 přidání a 0 odebrání
  1. 1 0
      tests/fixtures.py
  2. 8 0
      tests/test_extractors.py

+ 1 - 0
tests/fixtures.py

@@ -16,6 +16,7 @@ def disable_extractors_dict():
         "USE_WGET": "false",
         "USE_SINGLEFILE": "false",
         "USE_READABILITY": "false",
+        "USE_MERCURY": "false",
         "SAVE_PDF": "false",
         "SAVE_SCREENSHOT": "false",
         "SAVE_DOM": "false",

+ 8 - 0
tests/test_extractors.py

@@ -30,6 +30,14 @@ def test_readability_works(tmp_path, process, disable_extractors_dict):
     output_file = archived_item_path / "readability" / "content.html"
     assert output_file.exists()
 
+def test_mercury_works(tmp_path, process, disable_extractors_dict):
+    disable_extractors_dict.update({"USE_MERCURY": "true"})
+    add_process = subprocess.run(['archivebox', 'add', 'http://127.0.0.1:8080/static/example.com.html'],
+                                  capture_output=True, env=disable_extractors_dict)
+    archived_item_path = list(tmp_path.glob("archive/**/*"))[0]
+    output_file = archived_item_path / "mercury" / "content.html"
+    assert output_file.exists()
+
 def test_readability_works_with_wget(tmp_path, process, disable_extractors_dict):
     disable_extractors_dict.update({"USE_READABILITY": "true", "USE_WGET": "true"})
     add_process = subprocess.run(['archivebox', 'add', 'http://127.0.0.1:8080/static/example.com.html'],