o Zb8@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 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)sixStringIOmockpatchobjectsysr writegetvaluer AssertionErrorresearchDOTALL MULTILINE) selfr r captureZ stdout_fpZ stderr_fpnameregexpfpvaluer>D:\Projects\ConvertPro\env\Lib\site-packages\fire\testutils.pyassertOutputMatches%sFz BaseTestCase.assertOutputMatchescs8tjjdkrtt|j|i|Stt|j|i|S)N)r version_infomajorsuperrassertRaisesRegexpassertRaisesRegex)rargskwargs __class__rrr&Ks zBaseTestCase.assertRaisesRegexc 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. )r NzIncorrect exit code: %r != %r) r assertRaisesrZFireExitcoderassertIsInstancerZ FireTrace)rr,rexcrrrassertRaisesFireExitQs&  "z!BaseTestCase.assertRaisesFireExit)rrT)r) __name__ __module__ __qualname____doc__ contextlibcontextmanagerr r&r/ __classcell__rrr)rr"s  %rc cs8t}t|z |VWt|dSt|w)z>Context manager to mock a directory change and revert on exit.N)osgetcwdchdir) directoryZcwdirrrrChangeDirectoryks  r;)r3Z __future__rrrr4r7rrZunittestfirerrr r TestCaserr5r;mainskipskipIfrrrrs&     I