mirror of
https://github.com/openswarm-ai/openswarm.git
synced 2026-09-22 09:34:53 +02:00
* [eric] ci: gitleaks-ignore the known historical secrets so our branch stops failing on leaks it didnt add * [eric] workflows: restore scheduled-tasks on the workflow line (revert removal, keep windows fixes + 1.1.69) * [eric] workflows: re-apply uncommitted scheduling wip (schedule pill, calendar view, slice) * [eric] ops: gitignore dev-team local state files * [eric] ops: backlog item for download-tracking visibility * [eric] ci: allowlist the cdp-routes redaction-test token in gitleaks * [aidan] feat/scheduled-tasks: keep step labels in sync on edit and show chevron on every step * [aidan] fix: schedule time in chat * [aidan] ux/workflows: add workflow step removal (#91) * [aidan] feat/scheduled-tasks: remember workflow tool permissions across runs * [aidan] feat/task-scheduling: add hourly and minute (15-min minimum) schedule intervals (#93) * [aidan] feat/scheduled-tasks: calendar, rename, and edit workflows (#94) * [aidan] bug: fix schedule button * [aidan] fix/agent-errors: surface provider rate limits * [aidan] ux/cards: click-to-rename for chat and workflow titles Single-click a card's title to enter edit mode inline. Commit on Enter/blur, cancel on Escape. Rename persists via PATCH for workflows and sessions. * [aidan] feat/workflows: seed build prompt for zero-step workflows When a new workflow has no steps, seed the agent with a prompt asking the user to describe what the workflow should do, rather than starting blank. * [aidan] feat/workflows: add-to-schedule popover for unscheduled workflows Clicking the "+" on an unscheduled workflow row opens a popover with two options: - Keep this schedule: enables the workflow's existing cadence and moves it to Scheduled - Change schedule: opens the scheduling editor to pick a different time * [aidan] ux/workflows: wire add-to-schedule popover and simplify New button - Made the "+" icon on unscheduled workflow rows clickable, opening a popover to keep or change the schedule - Removed AddIcon from toolbar "New" button (now reads "New" instead of "+ New") * [aidan] fix/scheduled-tasks: open schedule calendar when Schedule pill clicked Fixed the Schedule pill click being swallowed by the toolbar's dismiss handler. Exempted the toolbar pills via data-toolbar-pills so their click handlers fire. * [aidan] ux/workflows: open New workflow in agent build chat instead of empty card When creating a new workflow from the hub, open it in edit_agent view (with the agent builder chat) instead of a preview card. The workflow is created on the backend first so the embedded session has a real ID. * [aidan] feat/workflow-edit: add draft testing save flow * [aidan] ux/chat: remove continue chat button * [aidan] ux/workflows: polish workflow card interactions * [aidan] fix/workflow-scheduling: save unscheduled workflows as drafts * aidan ui: schedule naming changes * [aidan] ui: tool calling desc/naming * [aidan] ui: calendar sidebar naming * [aidan] ui: fix stop viewing closing chat * [aidan] feat/workflows: auto-name workflows and polish the build flow (#95) * [aidan] feat/workflow-auto-naming: auto-generate workflow titles from steps Generate a title + description from a workflow's steps (one aux call, reused for step labels) whenever it is still auto_named, so a workflow built in the Edit Agent names itself on commit instead of staying "New workflow". A manual rename sets auto_named=False and is never overwritten. Stream the aux call (non-streaming drops content on some 9router lanes) and fall back to a step-derived title when the model is unavailable. * [aidan] feat/workflows: hide unsaved new workflows until first save A brand-new "+ New" workflow is created with unsaved=true and kept out of the hub's scheduled/unscheduled lists while the user is still building it in the Edit Agent. The first commit (Save) clears the flag and the workflow appears. Every other create path stays visible immediately. * [aidan] ux/workflows: remove redundant save workflow button The Edit Agent already has Discard/Save controls in its strip, so the header "Save Workflow" button was a duplicate save path. Remove it and its pulse/edit-session-id wiring; the model/time subtitle stays. * [aidan] ux/workflows: animate title on auto-rename Wrap the workflow card title in the same Typewriter the chat card uses, so when the auto-generated name replaces the placeholder after Save it retypes letter-by-letter. Gated on a real (non-placeholder) title so it never animates on mount or for already-named workflows. * [aidan] ux/workflows: animate sidebar title on auto-rename Wrap the calendar hub's sidebar row title in the same Typewriter the workflow card uses, so a title that auto-renames retypes letter-by-letter in the sidebar too. Extract the placeholder/isRealTitle guard into the shared workflowVisuals so the card and sidebar stay in sync. * [aidan] fix/workflows: connect watch tether, keep watched chat open, wire draft run/history * [aidan] ui: grey out chat pill when not selected * [aidan] ui: fix running agent display * [aidan] feat/history-popover: add chat history and scheduled tasks run log tabs (#96) * [aidan] ux/schedule: toast when calendar view already open on expand * [aidan] ui: fix popover descs * [aidan] feat/workflow-runs: add pause, resume, and stop controls for live runs * [aidan] feat/schedule-calendar: add calendar occurrences endpoint and concrete timezones * [aidan] feat/workflows: require at least one step to save a workflow * [aidan] ux/edit-agent: hide Discard for an unsaved new workflow * [aidan] ux/edit-agent: move fix-prefix card below the step list * [aidan] feat/workflows: toast when an unattended scheduled run starts * [aidan] fix/dashboard-tethers: anchor workflow-sidecar tethers to measured card rects * [aidan] feat/workflows: validate steps before scheduling and keep chat tool memory * [aidan] feat/mcp-suggestions: dismissable integration banner with per-session cooldown * [aidan] feat/workflows: add scheduled-run "running now" toast with click-to-view (#97) * [aidan] ux/workflows: surface paused state on card, sidebar, and calendar; tidy run history * [aidan] feat/mcp-suggestions: suggest both Google and Microsoft when provider is ambiguous * [aidan] fix/agent-tokens: friendly out-of-tokens card across all agent surfaces * [aidan] feat/workflow-model: persist edit-agent model on save with switch notice and fresh drafts * [aidan] fix/workflow-chat: force stop on watched run mirrors workflow card stop * [aidan] fix/workflow-cards: keep watched run tethered on finish to avoid duplicate chat * [aidan] feat/schedule-calendar: mark current time with a now line in week view * [aidan] refactor/private-names: rename error and schedule classifiers from _ to p_ * [aidan] feat/scheduled-tasks: agent workflow scheduling and in-chat convert (#98) * [aidan] feat/workflow-suggest: nudge user to convert repeatable chat to workflow Add SuggestConvertToWorkflow MCP tool that agents call at the end of a task when they've completed something worth repeating (daily report, weekly check, recurring data pull). Frontend detects the tool call and glows the "Convert to workflow" button 3 times to draw the eye. When user clicks it, the suggested cadence (e.g. "every weekday at 9am") is stored in the draft and seeded into the scheduling agent's first prompt, so the agent can act on the suggestion rather than asking the user again. Tool is never auto-called — agents decide when a task is genuinely repeatable (not debugging, creative work, one-off lookup). Tool description emphasizes sparse, high-confidence use only (once per session max). Files changed: - backend/apps/agents/schedule_mcp_server.py: add SuggestConvertToWorkflow tool - frontend/src/shared/mcpToolMeta.ts: add label for new tool - frontend/src/app/pages/Dashboard/cards/AgentCard.tsx: detect suggestion in session messages, show+glow "Convert to workflow" button, pass cadence to draft - frontend/src/shared/state/workflowsSlice.ts: add suggested_cadence field to Workflow interface - frontend/src/app/pages/Workflows/SchedulingView.tsx: seed scheduling agent prompt with suggested cadence hint * [aidan] feat/agent-scheduling: route recurring asks through native workflows, deny claude cron skill * [aidan] feat/workflow-convert: in-chat convert popup and auto-open scheduled workflow card * [aidan] ux/calendar-page: schedule calendar restyle + popover fixes (#99) * [aidan] fix/dashboard-delete: remove workflows calendar panel on delete key * [aidan] ux/workflows-calendar: restyle hub, fix today highlight, add toolbar toggle * [aidan] ux/schedule-popover: compact density, fix sticky header bleed, add header spacing * [aidan] ux/schedule-calendar: hollow ring dot for past fires in month view * [aidan] ux/schedule-calendar: clickable +N more opens day's full run list * [aidan] feat/run-log-filters: add success and skipped pills to scheduled task history * [aidan] fix/convert-button: stop drag capture so convert-to-workflow click fires * [aidan] ux/calendar-card: match border color and radius to chat and workflow cards * [aidan] fix/minimap: render missed-runs card on the minimap * [aidan] ux/run-sparkline: simplify tooltip to plain run tally * [aidan] ux/run-history: collapse expanded run view to one clickable line * [aidan] ux/calendar-card: match corner radius to browser cards * [aidan] feat/workflows: launch-time scheduling UX and workflow-card polish (#101) * [aidan] feat/schedule-list: lazy-load list view via scroll sentinel * [aidan] feat/missed-runs: launch toast with per-workflow counts and pan-to-card * [aidan] fix/dashboard-tethers: keep watching line anchored on canvas zoom * [aidan] feat/scheduled-tasks: review missed runs at launch instead of auto-firing on_missed * [aidan] refactor/workflow-cards: use radius and status design tokens, polish card chrome * [aidan] ux/agent-card: keep convert-to-workflow visible during runs with mid-turn toast * [aidan] ux/mcp-bubble: drop redundant verb label when a workflow label is shown * [aidan] chore/backend: remove stale explanatory comments * [aidan] fix/workflows-hub: load workflows on hub mount so calendar fills at launch * [aidan] feat/workflows: generate title, description, step labels at convert time * [aidan] fix/tidy-layout: include workflows hub in tidy and fit-to-view * [aidan] feat/schedule-list: window long list via measured-height virtualizer * [aidan] ux/workflows-hub: remove time-saved badge from calendar header * [aidan] fix/types: add missing semantic-type labels and drop stray fade arg * [aidan] feat/schedule: pin monthly day-of-month and honor repeat-every intervals * [aidan] feat/schedule: inherit source-session tool surface for scheduled runs * [aidan] ux/calendar: restack hour-cell events as bars with overflow affordance * [aidan] feat/calendar: open the run card when clicking a scheduled occurrence * [aidan] ux/missed-runs: add per-group select-all toggle and rename skip action * [aidan] feat: new scheduled task design ported * [aidan] ui: sidebar reorder, repeat controls on schedule card * [aidan] ui: sidebar, scheduling time * [aidan] feat/schedule: pin monthly last-day-of-month * [aidan] feat/steps: per-step enable toggle * [aidan] feat/workflows: per-workflow color swatch * [aidan] feat/trash: soft-delete workflows with restore and purge * [aidan] feat/run-monitor: live run monitor card on the canvas * [aidan] feat/run-context: attach a run as removable chat context * [aidan] feat/compose: new-workflow landing page and auto-commit build flow * [aidan] ui/workflows: dark mode and design-system cohesion * [aidan] ui/calendar: overflow popover, condensed week view, scroll fix * [aidan] feat/home: ongoing runs, missed review, and accurate Coming-up counts * [aidan] fix/run-status: sync ongoing runs and heal stuck/interrupted runs * [aidan] ux/schedule: last-day-of-month UI, Run-at time typing, interval input * [aidan] ux/workflows: default window size and toolbar icon * [aidan] fix/schedule: measure ran_late from start and anchor recurrences to created_at * [aidan] feat/calendar: render fire times from backend, drop JS recurrence reimpl * [aidan] chore/dashboard: drop dead configure/missed-run cards, refetch on reconnect * [aidan] chore/agent-card: remove unreachable convert-to-workflow action * [aidan] fix/workflows: don't bump updated_at on a no-op draft commit so viewing a workflow doesn't reorder the sidebar * [aidan] feat/schedule: warn when scheduling a workflow that has no steps * [aidan] fix/selection-tool: never select the workflows app, and exit the tool on Escape without dropping selections * [aidan] ux/compose: diversify new-workflow starter prompts across personas * [aidan] ux/run-monitor: spawn the run card a bit farther right of the workflows app * [aidan] fix/schedule: harden run recovery and storage writes against crashes * [aidan] ux/compose: restyle new-workflow starters as a clean pill cluster with rich prompts * [aidan] fix/workflows: optimistically apply edits so the schedule banner updates instantly * [aidan] ui/workflows: three-tone surface depth so the window lifts off the canvas in both themes * [aidan] ui/workflows: close buttons turn red on hover, matching the chat card * [aidan] test/schedule: cover executor pipeline, storage durability, and recurrence gaps * [aidan] fix: remove package-lock json * [aidan] fix/workflows-compose: keep compose view until edit agent replies * [aidan] feat/workflows: auto-generate workflow + step titles with typewriter animation * [eric] deps: restore frontend/package-lock.json (PR #105 deletion broke npm ci) --------- Co-authored-by: Eric <ciregenz@berkeley.edu> Co-authored-by: cire <134991075+ciregenz@users.noreply.github.com>
1088 lines
34 KiB
Python
1088 lines
34 KiB
Python
# mypy: allow-untyped-defs
|
|
"""Per-test stdout/stderr capturing mechanism."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import abc
|
|
import collections
|
|
import contextlib
|
|
import io
|
|
from io import UnsupportedOperation
|
|
import os
|
|
import sys
|
|
from tempfile import TemporaryFile
|
|
from types import TracebackType
|
|
from typing import Any
|
|
from typing import AnyStr
|
|
from typing import BinaryIO
|
|
from typing import Final
|
|
from typing import final
|
|
from typing import Generator
|
|
from typing import Generic
|
|
from typing import Iterable
|
|
from typing import Iterator
|
|
from typing import Literal
|
|
from typing import NamedTuple
|
|
from typing import TextIO
|
|
from typing import TYPE_CHECKING
|
|
|
|
|
|
if TYPE_CHECKING:
|
|
from typing_extensions import Self
|
|
|
|
from _pytest.config import Config
|
|
from _pytest.config import hookimpl
|
|
from _pytest.config.argparsing import Parser
|
|
from _pytest.deprecated import check_ispytest
|
|
from _pytest.fixtures import fixture
|
|
from _pytest.fixtures import SubRequest
|
|
from _pytest.nodes import Collector
|
|
from _pytest.nodes import File
|
|
from _pytest.nodes import Item
|
|
from _pytest.reports import CollectReport
|
|
|
|
|
|
_CaptureMethod = Literal["fd", "sys", "no", "tee-sys"]
|
|
|
|
|
|
def pytest_addoption(parser: Parser) -> None:
|
|
group = parser.getgroup("general")
|
|
group._addoption(
|
|
"--capture",
|
|
action="store",
|
|
default="fd",
|
|
metavar="method",
|
|
choices=["fd", "sys", "no", "tee-sys"],
|
|
help="Per-test capturing method: one of fd|sys|no|tee-sys",
|
|
)
|
|
group._addoption(
|
|
"-s",
|
|
action="store_const",
|
|
const="no",
|
|
dest="capture",
|
|
help="Shortcut for --capture=no",
|
|
)
|
|
|
|
|
|
def _colorama_workaround() -> None:
|
|
"""Ensure colorama is imported so that it attaches to the correct stdio
|
|
handles on Windows.
|
|
|
|
colorama uses the terminal on import time. So if something does the
|
|
first import of colorama while I/O capture is active, colorama will
|
|
fail in various ways.
|
|
"""
|
|
if sys.platform.startswith("win32"):
|
|
try:
|
|
import colorama # noqa: F401
|
|
except ImportError:
|
|
pass
|
|
|
|
|
|
def _windowsconsoleio_workaround(stream: TextIO) -> None:
|
|
"""Workaround for Windows Unicode console handling.
|
|
|
|
Python 3.6 implemented Unicode console handling for Windows. This works
|
|
by reading/writing to the raw console handle using
|
|
``{Read,Write}ConsoleW``.
|
|
|
|
The problem is that we are going to ``dup2`` over the stdio file
|
|
descriptors when doing ``FDCapture`` and this will ``CloseHandle`` the
|
|
handles used by Python to write to the console. Though there is still some
|
|
weirdness and the console handle seems to only be closed randomly and not
|
|
on the first call to ``CloseHandle``, or maybe it gets reopened with the
|
|
same handle value when we suspend capturing.
|
|
|
|
The workaround in this case will reopen stdio with a different fd which
|
|
also means a different handle by replicating the logic in
|
|
"Py_lifecycle.c:initstdio/create_stdio".
|
|
|
|
:param stream:
|
|
In practice ``sys.stdout`` or ``sys.stderr``, but given
|
|
here as parameter for unittesting purposes.
|
|
|
|
See https://github.com/pytest-dev/py/issues/103.
|
|
"""
|
|
if not sys.platform.startswith("win32") or hasattr(sys, "pypy_version_info"):
|
|
return
|
|
|
|
# Bail out if ``stream`` doesn't seem like a proper ``io`` stream (#2666).
|
|
if not hasattr(stream, "buffer"): # type: ignore[unreachable,unused-ignore]
|
|
return
|
|
|
|
raw_stdout = stream.buffer.raw if hasattr(stream.buffer, "raw") else stream.buffer
|
|
|
|
if not isinstance(raw_stdout, io._WindowsConsoleIO): # type: ignore[attr-defined,unused-ignore]
|
|
return
|
|
|
|
def _reopen_stdio(f, mode):
|
|
if not hasattr(stream.buffer, "raw") and mode[0] == "w":
|
|
buffering = 0
|
|
else:
|
|
buffering = -1
|
|
|
|
return io.TextIOWrapper(
|
|
open(os.dup(f.fileno()), mode, buffering),
|
|
f.encoding,
|
|
f.errors,
|
|
f.newlines,
|
|
f.line_buffering,
|
|
)
|
|
|
|
sys.stdin = _reopen_stdio(sys.stdin, "rb")
|
|
sys.stdout = _reopen_stdio(sys.stdout, "wb")
|
|
sys.stderr = _reopen_stdio(sys.stderr, "wb")
|
|
|
|
|
|
@hookimpl(wrapper=True)
|
|
def pytest_load_initial_conftests(early_config: Config) -> Generator[None]:
|
|
ns = early_config.known_args_namespace
|
|
if ns.capture == "fd":
|
|
_windowsconsoleio_workaround(sys.stdout)
|
|
_colorama_workaround()
|
|
pluginmanager = early_config.pluginmanager
|
|
capman = CaptureManager(ns.capture)
|
|
pluginmanager.register(capman, "capturemanager")
|
|
|
|
# Make sure that capturemanager is properly reset at final shutdown.
|
|
early_config.add_cleanup(capman.stop_global_capturing)
|
|
|
|
# Finally trigger conftest loading but while capturing (issue #93).
|
|
capman.start_global_capturing()
|
|
try:
|
|
try:
|
|
yield
|
|
finally:
|
|
capman.suspend_global_capture()
|
|
except BaseException:
|
|
out, err = capman.read_global_capture()
|
|
sys.stdout.write(out)
|
|
sys.stderr.write(err)
|
|
raise
|
|
|
|
|
|
# IO Helpers.
|
|
|
|
|
|
class EncodedFile(io.TextIOWrapper):
|
|
__slots__ = ()
|
|
|
|
@property
|
|
def name(self) -> str:
|
|
# Ensure that file.name is a string. Workaround for a Python bug
|
|
# fixed in >=3.7.4: https://bugs.python.org/issue36015
|
|
return repr(self.buffer)
|
|
|
|
@property
|
|
def mode(self) -> str:
|
|
# TextIOWrapper doesn't expose a mode, but at least some of our
|
|
# tests check it.
|
|
return self.buffer.mode.replace("b", "")
|
|
|
|
|
|
class CaptureIO(io.TextIOWrapper):
|
|
def __init__(self) -> None:
|
|
super().__init__(io.BytesIO(), encoding="UTF-8", newline="", write_through=True)
|
|
|
|
def getvalue(self) -> str:
|
|
assert isinstance(self.buffer, io.BytesIO)
|
|
return self.buffer.getvalue().decode("UTF-8")
|
|
|
|
|
|
class TeeCaptureIO(CaptureIO):
|
|
def __init__(self, other: TextIO) -> None:
|
|
self._other = other
|
|
super().__init__()
|
|
|
|
def write(self, s: str) -> int:
|
|
super().write(s)
|
|
return self._other.write(s)
|
|
|
|
|
|
class DontReadFromInput(TextIO):
|
|
@property
|
|
def encoding(self) -> str:
|
|
assert sys.__stdin__ is not None
|
|
return sys.__stdin__.encoding
|
|
|
|
def read(self, size: int = -1) -> str:
|
|
raise OSError(
|
|
"pytest: reading from stdin while output is captured! Consider using `-s`."
|
|
)
|
|
|
|
readline = read
|
|
|
|
def __next__(self) -> str:
|
|
return self.readline()
|
|
|
|
def readlines(self, hint: int | None = -1) -> list[str]:
|
|
raise OSError(
|
|
"pytest: reading from stdin while output is captured! Consider using `-s`."
|
|
)
|
|
|
|
def __iter__(self) -> Iterator[str]:
|
|
return self
|
|
|
|
def fileno(self) -> int:
|
|
raise UnsupportedOperation("redirected stdin is pseudofile, has no fileno()")
|
|
|
|
def flush(self) -> None:
|
|
raise UnsupportedOperation("redirected stdin is pseudofile, has no flush()")
|
|
|
|
def isatty(self) -> bool:
|
|
return False
|
|
|
|
def close(self) -> None:
|
|
pass
|
|
|
|
def readable(self) -> bool:
|
|
return False
|
|
|
|
def seek(self, offset: int, whence: int = 0) -> int:
|
|
raise UnsupportedOperation("redirected stdin is pseudofile, has no seek(int)")
|
|
|
|
def seekable(self) -> bool:
|
|
return False
|
|
|
|
def tell(self) -> int:
|
|
raise UnsupportedOperation("redirected stdin is pseudofile, has no tell()")
|
|
|
|
def truncate(self, size: int | None = None) -> int:
|
|
raise UnsupportedOperation("cannot truncate stdin")
|
|
|
|
def write(self, data: str) -> int:
|
|
raise UnsupportedOperation("cannot write to stdin")
|
|
|
|
def writelines(self, lines: Iterable[str]) -> None:
|
|
raise UnsupportedOperation("Cannot write to stdin")
|
|
|
|
def writable(self) -> bool:
|
|
return False
|
|
|
|
def __enter__(self) -> Self:
|
|
return self
|
|
|
|
def __exit__(
|
|
self,
|
|
type: type[BaseException] | None,
|
|
value: BaseException | None,
|
|
traceback: TracebackType | None,
|
|
) -> None:
|
|
pass
|
|
|
|
@property
|
|
def buffer(self) -> BinaryIO:
|
|
# The str/bytes doesn't actually matter in this type, so OK to fake.
|
|
return self # type: ignore[return-value]
|
|
|
|
|
|
# Capture classes.
|
|
|
|
|
|
class CaptureBase(abc.ABC, Generic[AnyStr]):
|
|
EMPTY_BUFFER: AnyStr
|
|
|
|
@abc.abstractmethod
|
|
def __init__(self, fd: int) -> None:
|
|
raise NotImplementedError()
|
|
|
|
@abc.abstractmethod
|
|
def start(self) -> None:
|
|
raise NotImplementedError()
|
|
|
|
@abc.abstractmethod
|
|
def done(self) -> None:
|
|
raise NotImplementedError()
|
|
|
|
@abc.abstractmethod
|
|
def suspend(self) -> None:
|
|
raise NotImplementedError()
|
|
|
|
@abc.abstractmethod
|
|
def resume(self) -> None:
|
|
raise NotImplementedError()
|
|
|
|
@abc.abstractmethod
|
|
def writeorg(self, data: AnyStr) -> None:
|
|
raise NotImplementedError()
|
|
|
|
@abc.abstractmethod
|
|
def snap(self) -> AnyStr:
|
|
raise NotImplementedError()
|
|
|
|
|
|
patchsysdict = {0: "stdin", 1: "stdout", 2: "stderr"}
|
|
|
|
|
|
class NoCapture(CaptureBase[str]):
|
|
EMPTY_BUFFER = ""
|
|
|
|
def __init__(self, fd: int) -> None:
|
|
pass
|
|
|
|
def start(self) -> None:
|
|
pass
|
|
|
|
def done(self) -> None:
|
|
pass
|
|
|
|
def suspend(self) -> None:
|
|
pass
|
|
|
|
def resume(self) -> None:
|
|
pass
|
|
|
|
def snap(self) -> str:
|
|
return ""
|
|
|
|
def writeorg(self, data: str) -> None:
|
|
pass
|
|
|
|
|
|
class SysCaptureBase(CaptureBase[AnyStr]):
|
|
def __init__(
|
|
self, fd: int, tmpfile: TextIO | None = None, *, tee: bool = False
|
|
) -> None:
|
|
name = patchsysdict[fd]
|
|
self._old: TextIO = getattr(sys, name)
|
|
self.name = name
|
|
if tmpfile is None:
|
|
if name == "stdin":
|
|
tmpfile = DontReadFromInput()
|
|
else:
|
|
tmpfile = CaptureIO() if not tee else TeeCaptureIO(self._old)
|
|
self.tmpfile = tmpfile
|
|
self._state = "initialized"
|
|
|
|
def repr(self, class_name: str) -> str:
|
|
return "<{} {} _old={} _state={!r} tmpfile={!r}>".format(
|
|
class_name,
|
|
self.name,
|
|
hasattr(self, "_old") and repr(self._old) or "<UNSET>",
|
|
self._state,
|
|
self.tmpfile,
|
|
)
|
|
|
|
def __repr__(self) -> str:
|
|
return "<{} {} _old={} _state={!r} tmpfile={!r}>".format(
|
|
self.__class__.__name__,
|
|
self.name,
|
|
hasattr(self, "_old") and repr(self._old) or "<UNSET>",
|
|
self._state,
|
|
self.tmpfile,
|
|
)
|
|
|
|
def _assert_state(self, op: str, states: tuple[str, ...]) -> None:
|
|
assert (
|
|
self._state in states
|
|
), "cannot {} in state {!r}: expected one of {}".format(
|
|
op, self._state, ", ".join(states)
|
|
)
|
|
|
|
def start(self) -> None:
|
|
self._assert_state("start", ("initialized",))
|
|
setattr(sys, self.name, self.tmpfile)
|
|
self._state = "started"
|
|
|
|
def done(self) -> None:
|
|
self._assert_state("done", ("initialized", "started", "suspended", "done"))
|
|
if self._state == "done":
|
|
return
|
|
setattr(sys, self.name, self._old)
|
|
del self._old
|
|
self.tmpfile.close()
|
|
self._state = "done"
|
|
|
|
def suspend(self) -> None:
|
|
self._assert_state("suspend", ("started", "suspended"))
|
|
setattr(sys, self.name, self._old)
|
|
self._state = "suspended"
|
|
|
|
def resume(self) -> None:
|
|
self._assert_state("resume", ("started", "suspended"))
|
|
if self._state == "started":
|
|
return
|
|
setattr(sys, self.name, self.tmpfile)
|
|
self._state = "started"
|
|
|
|
|
|
class SysCaptureBinary(SysCaptureBase[bytes]):
|
|
EMPTY_BUFFER = b""
|
|
|
|
def snap(self) -> bytes:
|
|
self._assert_state("snap", ("started", "suspended"))
|
|
self.tmpfile.seek(0)
|
|
res = self.tmpfile.buffer.read()
|
|
self.tmpfile.seek(0)
|
|
self.tmpfile.truncate()
|
|
return res
|
|
|
|
def writeorg(self, data: bytes) -> None:
|
|
self._assert_state("writeorg", ("started", "suspended"))
|
|
self._old.flush()
|
|
self._old.buffer.write(data)
|
|
self._old.buffer.flush()
|
|
|
|
|
|
class SysCapture(SysCaptureBase[str]):
|
|
EMPTY_BUFFER = ""
|
|
|
|
def snap(self) -> str:
|
|
self._assert_state("snap", ("started", "suspended"))
|
|
assert isinstance(self.tmpfile, CaptureIO)
|
|
res = self.tmpfile.getvalue()
|
|
self.tmpfile.seek(0)
|
|
self.tmpfile.truncate()
|
|
return res
|
|
|
|
def writeorg(self, data: str) -> None:
|
|
self._assert_state("writeorg", ("started", "suspended"))
|
|
self._old.write(data)
|
|
self._old.flush()
|
|
|
|
|
|
class FDCaptureBase(CaptureBase[AnyStr]):
|
|
def __init__(self, targetfd: int) -> None:
|
|
self.targetfd = targetfd
|
|
|
|
try:
|
|
os.fstat(targetfd)
|
|
except OSError:
|
|
# FD capturing is conceptually simple -- create a temporary file,
|
|
# redirect the FD to it, redirect back when done. But when the
|
|
# target FD is invalid it throws a wrench into this lovely scheme.
|
|
#
|
|
# Tests themselves shouldn't care if the FD is valid, FD capturing
|
|
# should work regardless of external circumstances. So falling back
|
|
# to just sys capturing is not a good option.
|
|
#
|
|
# Further complications are the need to support suspend() and the
|
|
# possibility of FD reuse (e.g. the tmpfile getting the very same
|
|
# target FD). The following approach is robust, I believe.
|
|
self.targetfd_invalid: int | None = os.open(os.devnull, os.O_RDWR)
|
|
os.dup2(self.targetfd_invalid, targetfd)
|
|
else:
|
|
self.targetfd_invalid = None
|
|
self.targetfd_save = os.dup(targetfd)
|
|
|
|
if targetfd == 0:
|
|
self.tmpfile = open(os.devnull, encoding="utf-8")
|
|
self.syscapture: CaptureBase[str] = SysCapture(targetfd)
|
|
else:
|
|
self.tmpfile = EncodedFile(
|
|
TemporaryFile(buffering=0),
|
|
encoding="utf-8",
|
|
errors="replace",
|
|
newline="",
|
|
write_through=True,
|
|
)
|
|
if targetfd in patchsysdict:
|
|
self.syscapture = SysCapture(targetfd, self.tmpfile)
|
|
else:
|
|
self.syscapture = NoCapture(targetfd)
|
|
|
|
self._state = "initialized"
|
|
|
|
def __repr__(self) -> str:
|
|
return (
|
|
f"<{self.__class__.__name__} {self.targetfd} oldfd={self.targetfd_save} "
|
|
f"_state={self._state!r} tmpfile={self.tmpfile!r}>"
|
|
)
|
|
|
|
def _assert_state(self, op: str, states: tuple[str, ...]) -> None:
|
|
assert (
|
|
self._state in states
|
|
), "cannot {} in state {!r}: expected one of {}".format(
|
|
op, self._state, ", ".join(states)
|
|
)
|
|
|
|
def start(self) -> None:
|
|
"""Start capturing on targetfd using memorized tmpfile."""
|
|
self._assert_state("start", ("initialized",))
|
|
os.dup2(self.tmpfile.fileno(), self.targetfd)
|
|
self.syscapture.start()
|
|
self._state = "started"
|
|
|
|
def done(self) -> None:
|
|
"""Stop capturing, restore streams, return original capture file,
|
|
seeked to position zero."""
|
|
self._assert_state("done", ("initialized", "started", "suspended", "done"))
|
|
if self._state == "done":
|
|
return
|
|
os.dup2(self.targetfd_save, self.targetfd)
|
|
os.close(self.targetfd_save)
|
|
if self.targetfd_invalid is not None:
|
|
if self.targetfd_invalid != self.targetfd:
|
|
os.close(self.targetfd)
|
|
os.close(self.targetfd_invalid)
|
|
self.syscapture.done()
|
|
self.tmpfile.close()
|
|
self._state = "done"
|
|
|
|
def suspend(self) -> None:
|
|
self._assert_state("suspend", ("started", "suspended"))
|
|
if self._state == "suspended":
|
|
return
|
|
self.syscapture.suspend()
|
|
os.dup2(self.targetfd_save, self.targetfd)
|
|
self._state = "suspended"
|
|
|
|
def resume(self) -> None:
|
|
self._assert_state("resume", ("started", "suspended"))
|
|
if self._state == "started":
|
|
return
|
|
self.syscapture.resume()
|
|
os.dup2(self.tmpfile.fileno(), self.targetfd)
|
|
self._state = "started"
|
|
|
|
|
|
class FDCaptureBinary(FDCaptureBase[bytes]):
|
|
"""Capture IO to/from a given OS-level file descriptor.
|
|
|
|
snap() produces `bytes`.
|
|
"""
|
|
|
|
EMPTY_BUFFER = b""
|
|
|
|
def snap(self) -> bytes:
|
|
self._assert_state("snap", ("started", "suspended"))
|
|
self.tmpfile.seek(0)
|
|
res = self.tmpfile.buffer.read()
|
|
self.tmpfile.seek(0)
|
|
self.tmpfile.truncate()
|
|
return res
|
|
|
|
def writeorg(self, data: bytes) -> None:
|
|
"""Write to original file descriptor."""
|
|
self._assert_state("writeorg", ("started", "suspended"))
|
|
os.write(self.targetfd_save, data)
|
|
|
|
|
|
class FDCapture(FDCaptureBase[str]):
|
|
"""Capture IO to/from a given OS-level file descriptor.
|
|
|
|
snap() produces text.
|
|
"""
|
|
|
|
EMPTY_BUFFER = ""
|
|
|
|
def snap(self) -> str:
|
|
self._assert_state("snap", ("started", "suspended"))
|
|
self.tmpfile.seek(0)
|
|
res = self.tmpfile.read()
|
|
self.tmpfile.seek(0)
|
|
self.tmpfile.truncate()
|
|
return res
|
|
|
|
def writeorg(self, data: str) -> None:
|
|
"""Write to original file descriptor."""
|
|
self._assert_state("writeorg", ("started", "suspended"))
|
|
# XXX use encoding of original stream
|
|
os.write(self.targetfd_save, data.encode("utf-8"))
|
|
|
|
|
|
# MultiCapture
|
|
|
|
|
|
# Generic NamedTuple only supported since Python 3.11.
|
|
if sys.version_info >= (3, 11) or TYPE_CHECKING:
|
|
|
|
@final
|
|
class CaptureResult(NamedTuple, Generic[AnyStr]):
|
|
"""The result of :method:`caplog.readouterr() <pytest.CaptureFixture.readouterr>`."""
|
|
|
|
out: AnyStr
|
|
err: AnyStr
|
|
|
|
else:
|
|
|
|
class CaptureResult(
|
|
collections.namedtuple("CaptureResult", ["out", "err"]), # noqa: PYI024
|
|
Generic[AnyStr],
|
|
):
|
|
"""The result of :method:`caplog.readouterr() <pytest.CaptureFixture.readouterr>`."""
|
|
|
|
__slots__ = ()
|
|
|
|
|
|
class MultiCapture(Generic[AnyStr]):
|
|
_state = None
|
|
_in_suspended = False
|
|
|
|
def __init__(
|
|
self,
|
|
in_: CaptureBase[AnyStr] | None,
|
|
out: CaptureBase[AnyStr] | None,
|
|
err: CaptureBase[AnyStr] | None,
|
|
) -> None:
|
|
self.in_: CaptureBase[AnyStr] | None = in_
|
|
self.out: CaptureBase[AnyStr] | None = out
|
|
self.err: CaptureBase[AnyStr] | None = err
|
|
|
|
def __repr__(self) -> str:
|
|
return (
|
|
f"<MultiCapture out={self.out!r} err={self.err!r} in_={self.in_!r} "
|
|
f"_state={self._state!r} _in_suspended={self._in_suspended!r}>"
|
|
)
|
|
|
|
def start_capturing(self) -> None:
|
|
self._state = "started"
|
|
if self.in_:
|
|
self.in_.start()
|
|
if self.out:
|
|
self.out.start()
|
|
if self.err:
|
|
self.err.start()
|
|
|
|
def pop_outerr_to_orig(self) -> tuple[AnyStr, AnyStr]:
|
|
"""Pop current snapshot out/err capture and flush to orig streams."""
|
|
out, err = self.readouterr()
|
|
if out:
|
|
assert self.out is not None
|
|
self.out.writeorg(out)
|
|
if err:
|
|
assert self.err is not None
|
|
self.err.writeorg(err)
|
|
return out, err
|
|
|
|
def suspend_capturing(self, in_: bool = False) -> None:
|
|
self._state = "suspended"
|
|
if self.out:
|
|
self.out.suspend()
|
|
if self.err:
|
|
self.err.suspend()
|
|
if in_ and self.in_:
|
|
self.in_.suspend()
|
|
self._in_suspended = True
|
|
|
|
def resume_capturing(self) -> None:
|
|
self._state = "started"
|
|
if self.out:
|
|
self.out.resume()
|
|
if self.err:
|
|
self.err.resume()
|
|
if self._in_suspended:
|
|
assert self.in_ is not None
|
|
self.in_.resume()
|
|
self._in_suspended = False
|
|
|
|
def stop_capturing(self) -> None:
|
|
"""Stop capturing and reset capturing streams."""
|
|
if self._state == "stopped":
|
|
raise ValueError("was already stopped")
|
|
self._state = "stopped"
|
|
if self.out:
|
|
self.out.done()
|
|
if self.err:
|
|
self.err.done()
|
|
if self.in_:
|
|
self.in_.done()
|
|
|
|
def is_started(self) -> bool:
|
|
"""Whether actively capturing -- not suspended or stopped."""
|
|
return self._state == "started"
|
|
|
|
def readouterr(self) -> CaptureResult[AnyStr]:
|
|
out = self.out.snap() if self.out else ""
|
|
err = self.err.snap() if self.err else ""
|
|
# TODO: This type error is real, need to fix.
|
|
return CaptureResult(out, err) # type: ignore[arg-type]
|
|
|
|
|
|
def _get_multicapture(method: _CaptureMethod) -> MultiCapture[str]:
|
|
if method == "fd":
|
|
return MultiCapture(in_=FDCapture(0), out=FDCapture(1), err=FDCapture(2))
|
|
elif method == "sys":
|
|
return MultiCapture(in_=SysCapture(0), out=SysCapture(1), err=SysCapture(2))
|
|
elif method == "no":
|
|
return MultiCapture(in_=None, out=None, err=None)
|
|
elif method == "tee-sys":
|
|
return MultiCapture(
|
|
in_=None, out=SysCapture(1, tee=True), err=SysCapture(2, tee=True)
|
|
)
|
|
raise ValueError(f"unknown capturing method: {method!r}")
|
|
|
|
|
|
# CaptureManager and CaptureFixture
|
|
|
|
|
|
class CaptureManager:
|
|
"""The capture plugin.
|
|
|
|
Manages that the appropriate capture method is enabled/disabled during
|
|
collection and each test phase (setup, call, teardown). After each of
|
|
those points, the captured output is obtained and attached to the
|
|
collection/runtest report.
|
|
|
|
There are two levels of capture:
|
|
|
|
* global: enabled by default and can be suppressed by the ``-s``
|
|
option. This is always enabled/disabled during collection and each test
|
|
phase.
|
|
|
|
* fixture: when a test function or one of its fixture depend on the
|
|
``capsys`` or ``capfd`` fixtures. In this case special handling is
|
|
needed to ensure the fixtures take precedence over the global capture.
|
|
"""
|
|
|
|
def __init__(self, method: _CaptureMethod) -> None:
|
|
self._method: Final = method
|
|
self._global_capturing: MultiCapture[str] | None = None
|
|
self._capture_fixture: CaptureFixture[Any] | None = None
|
|
|
|
def __repr__(self) -> str:
|
|
return (
|
|
f"<CaptureManager _method={self._method!r} _global_capturing={self._global_capturing!r} "
|
|
f"_capture_fixture={self._capture_fixture!r}>"
|
|
)
|
|
|
|
def is_capturing(self) -> str | bool:
|
|
if self.is_globally_capturing():
|
|
return "global"
|
|
if self._capture_fixture:
|
|
return f"fixture {self._capture_fixture.request.fixturename}"
|
|
return False
|
|
|
|
# Global capturing control
|
|
|
|
def is_globally_capturing(self) -> bool:
|
|
return self._method != "no"
|
|
|
|
def start_global_capturing(self) -> None:
|
|
assert self._global_capturing is None
|
|
self._global_capturing = _get_multicapture(self._method)
|
|
self._global_capturing.start_capturing()
|
|
|
|
def stop_global_capturing(self) -> None:
|
|
if self._global_capturing is not None:
|
|
self._global_capturing.pop_outerr_to_orig()
|
|
self._global_capturing.stop_capturing()
|
|
self._global_capturing = None
|
|
|
|
def resume_global_capture(self) -> None:
|
|
# During teardown of the python process, and on rare occasions, capture
|
|
# attributes can be `None` while trying to resume global capture.
|
|
if self._global_capturing is not None:
|
|
self._global_capturing.resume_capturing()
|
|
|
|
def suspend_global_capture(self, in_: bool = False) -> None:
|
|
if self._global_capturing is not None:
|
|
self._global_capturing.suspend_capturing(in_=in_)
|
|
|
|
def suspend(self, in_: bool = False) -> None:
|
|
# Need to undo local capsys-et-al if it exists before disabling global capture.
|
|
self.suspend_fixture()
|
|
self.suspend_global_capture(in_)
|
|
|
|
def resume(self) -> None:
|
|
self.resume_global_capture()
|
|
self.resume_fixture()
|
|
|
|
def read_global_capture(self) -> CaptureResult[str]:
|
|
assert self._global_capturing is not None
|
|
return self._global_capturing.readouterr()
|
|
|
|
# Fixture Control
|
|
|
|
def set_fixture(self, capture_fixture: CaptureFixture[Any]) -> None:
|
|
if self._capture_fixture:
|
|
current_fixture = self._capture_fixture.request.fixturename
|
|
requested_fixture = capture_fixture.request.fixturename
|
|
capture_fixture.request.raiseerror(
|
|
f"cannot use {requested_fixture} and {current_fixture} at the same time"
|
|
)
|
|
self._capture_fixture = capture_fixture
|
|
|
|
def unset_fixture(self) -> None:
|
|
self._capture_fixture = None
|
|
|
|
def activate_fixture(self) -> None:
|
|
"""If the current item is using ``capsys`` or ``capfd``, activate
|
|
them so they take precedence over the global capture."""
|
|
if self._capture_fixture:
|
|
self._capture_fixture._start()
|
|
|
|
def deactivate_fixture(self) -> None:
|
|
"""Deactivate the ``capsys`` or ``capfd`` fixture of this item, if any."""
|
|
if self._capture_fixture:
|
|
self._capture_fixture.close()
|
|
|
|
def suspend_fixture(self) -> None:
|
|
if self._capture_fixture:
|
|
self._capture_fixture._suspend()
|
|
|
|
def resume_fixture(self) -> None:
|
|
if self._capture_fixture:
|
|
self._capture_fixture._resume()
|
|
|
|
# Helper context managers
|
|
|
|
@contextlib.contextmanager
|
|
def global_and_fixture_disabled(self) -> Generator[None]:
|
|
"""Context manager to temporarily disable global and current fixture capturing."""
|
|
do_fixture = self._capture_fixture and self._capture_fixture._is_started()
|
|
if do_fixture:
|
|
self.suspend_fixture()
|
|
do_global = self._global_capturing and self._global_capturing.is_started()
|
|
if do_global:
|
|
self.suspend_global_capture()
|
|
try:
|
|
yield
|
|
finally:
|
|
if do_global:
|
|
self.resume_global_capture()
|
|
if do_fixture:
|
|
self.resume_fixture()
|
|
|
|
@contextlib.contextmanager
|
|
def item_capture(self, when: str, item: Item) -> Generator[None]:
|
|
self.resume_global_capture()
|
|
self.activate_fixture()
|
|
try:
|
|
yield
|
|
finally:
|
|
self.deactivate_fixture()
|
|
self.suspend_global_capture(in_=False)
|
|
|
|
out, err = self.read_global_capture()
|
|
item.add_report_section(when, "stdout", out)
|
|
item.add_report_section(when, "stderr", err)
|
|
|
|
# Hooks
|
|
|
|
@hookimpl(wrapper=True)
|
|
def pytest_make_collect_report(
|
|
self, collector: Collector
|
|
) -> Generator[None, CollectReport, CollectReport]:
|
|
if isinstance(collector, File):
|
|
self.resume_global_capture()
|
|
try:
|
|
rep = yield
|
|
finally:
|
|
self.suspend_global_capture()
|
|
out, err = self.read_global_capture()
|
|
if out:
|
|
rep.sections.append(("Captured stdout", out))
|
|
if err:
|
|
rep.sections.append(("Captured stderr", err))
|
|
else:
|
|
rep = yield
|
|
return rep
|
|
|
|
@hookimpl(wrapper=True)
|
|
def pytest_runtest_setup(self, item: Item) -> Generator[None]:
|
|
with self.item_capture("setup", item):
|
|
return (yield)
|
|
|
|
@hookimpl(wrapper=True)
|
|
def pytest_runtest_call(self, item: Item) -> Generator[None]:
|
|
with self.item_capture("call", item):
|
|
return (yield)
|
|
|
|
@hookimpl(wrapper=True)
|
|
def pytest_runtest_teardown(self, item: Item) -> Generator[None]:
|
|
with self.item_capture("teardown", item):
|
|
return (yield)
|
|
|
|
@hookimpl(tryfirst=True)
|
|
def pytest_keyboard_interrupt(self) -> None:
|
|
self.stop_global_capturing()
|
|
|
|
@hookimpl(tryfirst=True)
|
|
def pytest_internalerror(self) -> None:
|
|
self.stop_global_capturing()
|
|
|
|
|
|
class CaptureFixture(Generic[AnyStr]):
|
|
"""Object returned by the :fixture:`capsys`, :fixture:`capsysbinary`,
|
|
:fixture:`capfd` and :fixture:`capfdbinary` fixtures."""
|
|
|
|
def __init__(
|
|
self,
|
|
captureclass: type[CaptureBase[AnyStr]],
|
|
request: SubRequest,
|
|
*,
|
|
_ispytest: bool = False,
|
|
) -> None:
|
|
check_ispytest(_ispytest)
|
|
self.captureclass: type[CaptureBase[AnyStr]] = captureclass
|
|
self.request = request
|
|
self._capture: MultiCapture[AnyStr] | None = None
|
|
self._captured_out: AnyStr = self.captureclass.EMPTY_BUFFER
|
|
self._captured_err: AnyStr = self.captureclass.EMPTY_BUFFER
|
|
|
|
def _start(self) -> None:
|
|
if self._capture is None:
|
|
self._capture = MultiCapture(
|
|
in_=None,
|
|
out=self.captureclass(1),
|
|
err=self.captureclass(2),
|
|
)
|
|
self._capture.start_capturing()
|
|
|
|
def close(self) -> None:
|
|
if self._capture is not None:
|
|
out, err = self._capture.pop_outerr_to_orig()
|
|
self._captured_out += out
|
|
self._captured_err += err
|
|
self._capture.stop_capturing()
|
|
self._capture = None
|
|
|
|
def readouterr(self) -> CaptureResult[AnyStr]:
|
|
"""Read and return the captured output so far, resetting the internal
|
|
buffer.
|
|
|
|
:returns:
|
|
The captured content as a namedtuple with ``out`` and ``err``
|
|
string attributes.
|
|
"""
|
|
captured_out, captured_err = self._captured_out, self._captured_err
|
|
if self._capture is not None:
|
|
out, err = self._capture.readouterr()
|
|
captured_out += out
|
|
captured_err += err
|
|
self._captured_out = self.captureclass.EMPTY_BUFFER
|
|
self._captured_err = self.captureclass.EMPTY_BUFFER
|
|
return CaptureResult(captured_out, captured_err)
|
|
|
|
def _suspend(self) -> None:
|
|
"""Suspend this fixture's own capturing temporarily."""
|
|
if self._capture is not None:
|
|
self._capture.suspend_capturing()
|
|
|
|
def _resume(self) -> None:
|
|
"""Resume this fixture's own capturing temporarily."""
|
|
if self._capture is not None:
|
|
self._capture.resume_capturing()
|
|
|
|
def _is_started(self) -> bool:
|
|
"""Whether actively capturing -- not disabled or closed."""
|
|
if self._capture is not None:
|
|
return self._capture.is_started()
|
|
return False
|
|
|
|
@contextlib.contextmanager
|
|
def disabled(self) -> Generator[None]:
|
|
"""Temporarily disable capturing while inside the ``with`` block."""
|
|
capmanager: CaptureManager = self.request.config.pluginmanager.getplugin(
|
|
"capturemanager"
|
|
)
|
|
with capmanager.global_and_fixture_disabled():
|
|
yield
|
|
|
|
|
|
# The fixtures.
|
|
|
|
|
|
@fixture
|
|
def capsys(request: SubRequest) -> Generator[CaptureFixture[str]]:
|
|
r"""Enable text capturing of writes to ``sys.stdout`` and ``sys.stderr``.
|
|
|
|
The captured output is made available via ``capsys.readouterr()`` method
|
|
calls, which return a ``(out, err)`` namedtuple.
|
|
``out`` and ``err`` will be ``text`` objects.
|
|
|
|
Returns an instance of :class:`CaptureFixture[str] <pytest.CaptureFixture>`.
|
|
|
|
Example:
|
|
|
|
.. code-block:: python
|
|
|
|
def test_output(capsys):
|
|
print("hello")
|
|
captured = capsys.readouterr()
|
|
assert captured.out == "hello\n"
|
|
"""
|
|
capman: CaptureManager = request.config.pluginmanager.getplugin("capturemanager")
|
|
capture_fixture = CaptureFixture(SysCapture, request, _ispytest=True)
|
|
capman.set_fixture(capture_fixture)
|
|
capture_fixture._start()
|
|
yield capture_fixture
|
|
capture_fixture.close()
|
|
capman.unset_fixture()
|
|
|
|
|
|
@fixture
|
|
def capsysbinary(request: SubRequest) -> Generator[CaptureFixture[bytes]]:
|
|
r"""Enable bytes capturing of writes to ``sys.stdout`` and ``sys.stderr``.
|
|
|
|
The captured output is made available via ``capsysbinary.readouterr()``
|
|
method calls, which return a ``(out, err)`` namedtuple.
|
|
``out`` and ``err`` will be ``bytes`` objects.
|
|
|
|
Returns an instance of :class:`CaptureFixture[bytes] <pytest.CaptureFixture>`.
|
|
|
|
Example:
|
|
|
|
.. code-block:: python
|
|
|
|
def test_output(capsysbinary):
|
|
print("hello")
|
|
captured = capsysbinary.readouterr()
|
|
assert captured.out == b"hello\n"
|
|
"""
|
|
capman: CaptureManager = request.config.pluginmanager.getplugin("capturemanager")
|
|
capture_fixture = CaptureFixture(SysCaptureBinary, request, _ispytest=True)
|
|
capman.set_fixture(capture_fixture)
|
|
capture_fixture._start()
|
|
yield capture_fixture
|
|
capture_fixture.close()
|
|
capman.unset_fixture()
|
|
|
|
|
|
@fixture
|
|
def capfd(request: SubRequest) -> Generator[CaptureFixture[str]]:
|
|
r"""Enable text capturing of writes to file descriptors ``1`` and ``2``.
|
|
|
|
The captured output is made available via ``capfd.readouterr()`` method
|
|
calls, which return a ``(out, err)`` namedtuple.
|
|
``out`` and ``err`` will be ``text`` objects.
|
|
|
|
Returns an instance of :class:`CaptureFixture[str] <pytest.CaptureFixture>`.
|
|
|
|
Example:
|
|
|
|
.. code-block:: python
|
|
|
|
def test_system_echo(capfd):
|
|
os.system('echo "hello"')
|
|
captured = capfd.readouterr()
|
|
assert captured.out == "hello\n"
|
|
"""
|
|
capman: CaptureManager = request.config.pluginmanager.getplugin("capturemanager")
|
|
capture_fixture = CaptureFixture(FDCapture, request, _ispytest=True)
|
|
capman.set_fixture(capture_fixture)
|
|
capture_fixture._start()
|
|
yield capture_fixture
|
|
capture_fixture.close()
|
|
capman.unset_fixture()
|
|
|
|
|
|
@fixture
|
|
def capfdbinary(request: SubRequest) -> Generator[CaptureFixture[bytes]]:
|
|
r"""Enable bytes capturing of writes to file descriptors ``1`` and ``2``.
|
|
|
|
The captured output is made available via ``capfd.readouterr()`` method
|
|
calls, which return a ``(out, err)`` namedtuple.
|
|
``out`` and ``err`` will be ``byte`` objects.
|
|
|
|
Returns an instance of :class:`CaptureFixture[bytes] <pytest.CaptureFixture>`.
|
|
|
|
Example:
|
|
|
|
.. code-block:: python
|
|
|
|
def test_system_echo(capfdbinary):
|
|
os.system('echo "hello"')
|
|
captured = capfdbinary.readouterr()
|
|
assert captured.out == b"hello\n"
|
|
|
|
"""
|
|
capman: CaptureManager = request.config.pluginmanager.getplugin("capturemanager")
|
|
capture_fixture = CaptureFixture(FDCaptureBinary, request, _ispytest=True)
|
|
capman.set_fixture(capture_fixture)
|
|
capture_fixture._start()
|
|
yield capture_fixture
|
|
capture_fixture.close()
|
|
capman.unset_fixture()
|