Browse Source

test extract flag on oneshot

Nick Sweeting 5 years ago
parent
commit
e03d17c208
1 changed files with 13 additions and 3 deletions
  1. 13 3
      tests/test_oneshot.py

+ 13 - 3
tests/test_oneshot.py

@@ -9,11 +9,21 @@ def test_oneshot_command_exists(tmp_path, disable_extractors_dict):
 
 def test_oneshot_command_saves_page_in_right_folder(tmp_path, disable_extractors_dict):
     disable_extractors_dict.update({"SAVE_DOM": "true"})
-    process = subprocess.run(["archivebox", "oneshot", f"--out-dir={tmp_path}", "http://127.0.0.1:8080/static/example.com.html"],
-                              capture_output=True, env=disable_extractors_dict)
+    process = subprocess.run(
+        [
+            "archivebox",
+            "oneshot",
+            f"--out-dir={tmp_path}",
+            "--extract=title,favicon,dom",
+            "http://127.0.0.1:8080/static/example.com.html",
+        ],
+        capture_output=True,
+        env=disable_extractors_dict,
+    )
+    print(process.stdout)
     items = ' '.join([str(x) for x in tmp_path.iterdir()])
     current_path = ' '.join([str(x) for x in Path.cwd().iterdir()])
     assert "index.json" in items
     assert not "index.sqlite3" in current_path
     assert "output.html" in items
-    
+