`jz dZddlZddlZddlZddlZddlZddlZddlmZddlm Z ddlm Z Gddej Z ej dZejZejZejZdS) z"Utilities for Python Fire's tests.N)mock)core)tracecVeZdZdZejddZejddZdS) BaseTestCasez'Shared test case for Python Fire tests..*Tc#Ktj}tj} tjt d|5tjt d|5dVdddn #1swxYwYdddn #1swxYwY|sbt j|t j |nj#|sct j|t j |wwxYwd||fd||ffD]w\}}}|} || rt|d| 3tj || tj tjzst|d| d|xdS)aAsserts that the context generates stdout and stderr matching regexps. Note: If wrapped code raises an exception, stdout and stderr will not be checked. Args: stdout: (str) regexp to match against stdout (None will check no stdout) stderr: (str) regexp to match against stderr (None will check no stderr) capture: (bool, default True) do not bubble up stdout or stderr Yields: Yields to the wrapped context. stdoutstderrNz: Expected no output. Got: z : Expected z to match )ioStringIOrpatchobjectsysr writegetvaluer AssertionErrorresearchDOTALL MULTILINE) selfr r capture stdout_fp stderr_fpnameregexpfpvalues R/home/jworkman/hermes-web-chat/venv/lib/python3.11/site-packages/fire/testutils.pyassertOutputMatchesz BaseTestCase.assertOutputMatches sz I I/ :  S(I 6 6 Z  sHi 8 8   %%%               / ++--... ++--.../ ++--... ++--..../' :& :< 6 6fbkkmme   . $ee -.. . .y BL(@AA 6 $eeeVV 566 6 6 6 6sT&D'B7B< BB BB B DB##D&B#'DA'E6c #K||5|tj5 dVnb#tj$rP}|j|krt d|jd|||jtjd}~wwxYw dddn #1swxYwYddddS#1swxYwYdS)a^Asserts that a FireExit error is raised in the context. Allows tests to check that Fire's wrapper around SystemExit is raised and that a regexp is matched in the output. Args: code: The status code that the FireExit should contain. regexp: stdout must match this regex. Yields: Yields to the wrapped context. )r NzIncorrect exit code: z != ) r! assertRaisesrFireExitcoderassertIsInstancer FireTrace)rr%rexcs r assertRaisesFireExitz!BaseTestCase.assertRaisesFireExitFst  ! ! ! 0 0     T] + +  %%%%}    X   ."%(((DD"233 3    5? ; ; ;                      sW CB/AB/BA BBB/# C/B3 3C6B3 7CC C N)rrT)r)__name__ __module__ __qualname____doc__ contextlibcontextmanagerr!r)r rrsa// #6#6#6#6Jr1rc#Ktj}tj| |Vtj|dS#tj|wxYw)z>Context manager to mock a directory change and revert on exit.N)osgetcwdchdir) directorycwdirs r ChangeDirectoryr8`sT )++%(9 OOOHUOOOOOBHUOOOOs AA)r-r.r r3rrunittestrfirerrTestCaserr/r8mainskipskipIfr0r1r r?s)( @@@@@8$@@@F }} r1