B 5d8@sdZddlmZddlmZddlmZddlZddlZddlZddlZddl Z ddl m Z ddl m Z ddl Z ddlZGdd d e jZejd d Ze jZe jZe jZdS) z"Utilities for Python Fire's tests.)absolute_import)division)print_functionN)core)tracecs@eZdZdZejd ddZfddZejd dd ZZ S) BaseTestCasez'Shared test case for Python Fire tests..*Tc cst}t}zBtjtd|(tjtd| dVWdQRXWdQRXWd|sxtj|tj |Xxpd||fd||fgD]X\}}}|} |dkr| rt d|| fqt || t j t jBst d|| |fqWdS)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)sixStringIOmockpatchobjectsysr writegetvaluer AssertionErrorresearchDOTALL MULTILINE) selfr r captureZ stdout_fpZ stderr_fpnameregexpfpvaluer2/tmp/pip-unpacked-wheel-1kz5bvl4/fire/testutils.pyassertOutputMatches%s& z BaseTestCase.assertOutputMatchescs4tjjdkrtt|j||Stt|j||SdS)N)r version_infomajorsuperrZassertRaisesRegexpassertRaisesRegex)rargskwargs) __class__rrr%Ks zBaseTestCase.assertRaisesRegexc cs|j|d||tjdy dVWnRtjk rx}z2|j|krVtd|j|f||jtjWdd}~XYnXWdQRXWdQRXdS)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: %r != %r) r assertRaisesrZFireExitcoderZassertIsInstancerZ FireTrace)rr*rexcrrrassertRaisesFireExitQs  z!BaseTestCase.assertRaisesFireExit)rrT)r) __name__ __module__ __qualname____doc__ contextlibcontextmanagerr r%r, __classcell__rr)r(rr"s  % rc cs.t}t|z |VWdt|XdS)z>Context manager to mock a directory change and revert on exit.N)osgetcwdchdir) directoryZcwdirrrrChangeDirectoryks   r8)r0 __future__rrrr1r4rrZunittestZfirerrr r ZTestCaserr2r8mainskipZskipIfrrrrs"     I