From 95a105feb913bcc05b4026088f7dde2ca464d8b1 Mon Sep 17 00:00:00 2001 From: Nick Sweeting Date: Sun, 15 Mar 2026 19:22:06 -0700 Subject: [PATCH] small fixes --- archivebox/api/tests.py | 11 +++++------ archivebox/misc/jsonl.py | 4 ++-- 2 files changed, 7 insertions(+), 8 deletions(-) diff --git a/archivebox/api/tests.py b/archivebox/api/tests.py index 0dba652c..f04fa248 100644 --- a/archivebox/api/tests.py +++ b/archivebox/api/tests.py @@ -1,6 +1,5 @@ import importlib from io import StringIO -from types import SimpleNamespace from archivebox.config.django import setup_django @@ -8,6 +7,7 @@ setup_django() User = importlib.import_module('django.contrib.auth.models').User TestCase = importlib.import_module('django.test').TestCase +RequestFactory = importlib.import_module('django.test').RequestFactory api_v1_cli = importlib.import_module('archivebox.api.v1_cli') ScheduleCommandSchema = api_v1_cli.ScheduleCommandSchema cli_schedule = api_v1_cli.cli_schedule @@ -23,11 +23,10 @@ class CLIScheduleAPITests(TestCase): ) def test_schedule_api_creates_schedule(self): - request = SimpleNamespace( - user=self.user, - stdout=StringIO(), - stderr=StringIO(), - ) + request = RequestFactory().post('/api/v1/cli/schedule') + request.user = self.user + request.stdout = StringIO() + request.stderr = StringIO() args = ScheduleCommandSchema( every='daily', import_path='https://example.com/feed.xml', diff --git a/archivebox/misc/jsonl.py b/archivebox/misc/jsonl.py index 826c7368..aaade4e8 100644 --- a/archivebox/misc/jsonl.py +++ b/archivebox/misc/jsonl.py @@ -25,7 +25,7 @@ __package__ = 'archivebox.misc' import sys import json import select -from typing import Iterator, Dict, Any, Optional, TextIO +from typing import Iterable, Iterator, Dict, Any, Optional, TextIO from pathlib import Path @@ -118,7 +118,7 @@ def read_file(path: Path) -> Iterator[Dict[str, Any]]: yield record -def read_args_or_stdin(args: tuple, stream: Optional[TextIO] = None) -> Iterator[Dict[str, Any]]: +def read_args_or_stdin(args: Iterable[str], stream: Optional[TextIO] = None) -> Iterator[Dict[str, Any]]: """ Read from CLI arguments if provided, otherwise from stdin.