From b1dbfcb73f22ebbb68e511650ede01db9cd87809 Mon Sep 17 00:00:00 2001 From: JDC Date: Fri, 13 Nov 2020 14:17:12 -0500 Subject: [PATCH] Add test remove tag filter --- tests/test_remove.py | 23 +++++++++++++++++++++++ 1 file changed, 23 insertions(+) diff --git a/tests/test_remove.py b/tests/test_remove.py index 0fb16e2a..c9c63385 100644 --- a/tests/test_remove.py +++ b/tests/test_remove.py @@ -70,6 +70,29 @@ def test_remove_domain(tmp_path, process, disable_extractors_dict): assert count == 0 + +def test_remove_tag(tmp_path, process, disable_extractors_dict): + subprocess.run(['archivebox', 'add', 'http://127.0.0.1:8080/static/example.com.html'], capture_output=True, env=disable_extractors_dict) + subprocess.run(['archivebox', 'add', 'http://127.0.0.1:8080/static/iana.org.html'], capture_output=True, env=disable_extractors_dict) + assert list((tmp_path / "archive").iterdir()) != [] + + conn = sqlite3.connect("index.sqlite3") + c = conn.cursor() + c.execute("INSERT INTO core_tag (id, name, slug) VALUES (2, 'test-tag', 'test-tag')") + snapshot_ids = c.execute("SELECT id from core_snapshot") + c.executemany('INSERT INTO core_snapshot_tags (snapshot_id, tag_id) VALUES (?, 2)', list(snapshot_ids)) + conn.commit() + + remove_process = subprocess.run(['archivebox', 'remove', '--filter-type=tag', 'test-tag', '--yes', '--delete'], capture_output=True) + + assert len(list((tmp_path / "archive").iterdir())) == 0 + + count = c.execute("SELECT COUNT() from core_snapshot").fetchone()[0] + conn.commit() + conn.close() + + assert count == 0 + def test_remove_before(tmp_path, process, disable_extractors_dict): subprocess.run(['archivebox', 'add', 'http://127.0.0.1:8080/static/example.com.html'], capture_output=True, env=disable_extractors_dict) subprocess.run(['archivebox', 'add', 'http://127.0.0.1:8080/static/iana.org.html'], capture_output=True, env=disable_extractors_dict)