U ,c8@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}zBtj td|(tj td| dVW5QRXW5QRXW5|sxtj|tj|Xd||fd||ffD]X\}}}|} |dkr| rt d|| fqt || t j t jBst d|| |fqdS)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)sixStringIOsysr writegetvaluer mockpatchobjectAssertionErrorresearchDOTALL MULTILINE) selfr r captureZ stdout_fpZ stderr_fpnameregexpfpvaluer2/tmp/pip-unpacked-wheel-1xt1w7un/fire/testutils.pyassertOutputMatches%s,z BaseTestCase.assertOutputMatchescs4tjjdkrtt|j||Stt|j||SdS)N)r version_infomajorsuperrZassertRaisesRegexpassertRaisesRegex)rargskwargs __class__rrr%Ks zBaseTestCase.assertRaisesRegexc cs|j|d||tjdz dVWnRtjk rx}z2|j|krVtd|j|f||jtjW5d}~XYnXW5QRXW5QRXdS)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__rrr(rr"s  % rc cs.t}t|z |VW5t|XdS)z>Context manager to mock a directory change and revert on exit.N)osgetcwdchdir) directoryZcwdirrrrChangeDirectoryks   r9)r1 __future__rrrr2r5rr ZunittestZfirerrrr ZTestCaserr3r9mainskipZskipIfrrrrs$     I