diff --git a/archivebox/tests/test_ui_add_view.py b/archivebox/tests/test_ui_add_view.py index 29008680..3baabfbd 100644 --- a/archivebox/tests/test_ui_add_view.py +++ b/archivebox/tests/test_ui_add_view.py @@ -100,6 +100,9 @@ def test_add_view_admin_renders_plugin_config_grid(client, admin_user, monkeypat def test_add_view_hides_agent_link_when_opencode_is_disabled(client, admin_user): + from archivebox.machine.models import Machine + + Machine.from_json({"config": {"OPENCODE_ENABLED": False}}) client.force_login(admin_user) response = client.get(reverse("add"), HTTP_HOST=ADMIN_HOST) diff --git a/archivebox/tests/test_ui_admin_links.py b/archivebox/tests/test_ui_admin_links.py index 4a6f4174..7ace161a 100644 --- a/archivebox/tests/test_ui_admin_links.py +++ b/archivebox/tests/test_ui_admin_links.py @@ -240,6 +240,9 @@ def test_snapshot_admin_zip_links(): def test_admin_navigation_hides_agent_link_when_opencode_is_disabled(client, admin_user): + from archivebox.machine.models import Machine + + Machine.from_json({"config": {"OPENCODE_ENABLED": False}}) client.force_login(admin_user) response = client.get(reverse("admin:index"), HTTP_HOST="admin.archivebox.localhost:8000")