o Ďiz @sdZddlZddlZddlZddlZddlZddlZddlmZddlm Z ddlm Z Gdddej Z ej dd ZejZejZejZdS) z"Utilities for Python Fire's tests.N)mock)core)tracec@s0eZdZdZejd ddZejd ddZdS) BaseTestCasez'Shared test case for Python Fire tests..*Tc cs<t}t}zGtjtd|#tjtd| dVWdn1s)wYWdn1s8wYW|sPtj|tj |n|sdtj|tj |wwd||fd||ffD],\}}}|} |dur| rt d|| fqot || t j t jBst d|| |fqodS)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%s: Expected no output. Got: %rz%s: Expected %r to match %r)ioStringIOrpatchobjectsysrwritegetvaluerAssertionErrorresearchDOTALL MULTILINE) selfrrcapture stdout_fp stderr_fpnameregexpfpvaluerJ/home/jeff/fluffinator/venv/lib/python3.10/site-packages/fire/testutils.pyassertOutputMatches sFz BaseTestCase.assertOutputMatchesc cs|j|dQ|tj2zdVWn$tjy8}z|j|kr+td|j|f||jtjd}~wwWdn1sCwYWddSWddS1s[wYdS)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. )rNzIncorrect exit code: %r != %r) r assertRaisesrFireExitcoderassertIsInstancer FireTrace)rr"rexcrrrassertRaisesFireExitFs&  "z!BaseTestCase.assertRaisesFireExitN)rrT)r)__name__ __module__ __qualname____doc__ contextlibcontextmanagerrr&rrrrrs  %rc cs8t}t|z |VWt|dSt|w)z>Context manager to mock a directory change and revert on exit.N)osgetcwdchdir) directorycwdirrrrChangeDirectory`s  r2)r*r+r r-rr unittestrfirerrTestCaserr,r2mainskipskipIfrrrrs    C