|hz dZddlZddlZddlZddlZddlZddlZddlmZddlm Z ddlm Z GddejZ ejdZejZej Zej"Zy) z"Utilities for Python Fire's tests.N)mock)core)tracec\eZdZdZej ddZej ddZy) BaseTestCasez'Shared test case for Python Fire tests.c#Ktj}tj} tjj t d|5tjj t d|5ddddddd|s[t j j|jt jj|j d||fd||ffD]t\}}}|j} || st|d| tj|| tjtjzrct|d| d|y#1swYxYw#1swYxYw#|s[t j j|jt jj|jwwxYww)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 M/var/www/html/test/engine/venv/lib/python3.12/site-packages/fire/testutils.pyassertOutputMatchesz BaseTestCase.assertOutputMatches s I I/ ::  S(I 6 ZZ  sHi 8    ++-. ++-.& :& :< 6fbkkme   $e -. .yy BLL(@A $eV 56 6 6   ++-. ++-.s[)G%%F&E77E+<E7F BG%AG%G%+E4 0E77F<FAG""G%c #K|j|5|jtj5 d ddddddy#tj$rZ}|j|k7rt d|jd||j |jtjd}~wwxYw#1swYxYw#1swYyxYww)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 rassertRaisesFireExitz!BaseTestCase.assertRaisesFireExitFs  ! ! ! 0    T]] +    }}  XX  "%((D"23 3    5?? ;     sPC C B=A C  C B: AB55B::B==C C  CCN).*r)T)r))__name__ __module__ __qualname____doc__ contextlibcontextmanagerr r(rrrs8/ #6#6Jr1rc#Ktj}tj| |tj|y#tj|wxYww)z>Context manager to mock a directory change and revert on exit.N)osgetcwdchdir) directorycwdirs rChangeDirectoryr8`s? ))+%((9 OHHUOBHHUOs*A!AA!AA!)r-r.r r3rrunittestrfirerrTestCaserr/r8mainskipskipIfr0r1rr?sq) @8$$@F }}}} r1