瀏覽代碼

test: Improve oneshot test

Cristian 5 年之前
父節點
當前提交
0158efb1d0
共有 1 個文件被更改,包括 2 次插入0 次删除
  1. 2 0
      tests/test_oneshot.py

+ 2 - 0
tests/test_oneshot.py

@@ -8,10 +8,12 @@ def test_oneshot_command_exists(tmp_path, disable_extractors_dict):
     assert not "invalid choice: 'oneshot'" in process.stderr.decode("utf-8")
 
 def test_oneshot_commad_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)
     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