B ܇\f@sddlmZddlmZddlmZddlZddlZddlmZddlZddl m Z dZ dd Z d d Z d d ZeddZddZejdddddZejdddddZejddddZddZdS))absolute_import)division)print_functionN)contextmanager)compatz(-Walways::pytest.RemovedInPytest4Warningc Cs|d}t|dkr&|d|fxt|dkr@|dq(Wdd|D\}}}}}||}||}|ryt|}|dkrtWqttfk r|d|fYqXnd}| |||||d S) zP Copy of the warning._setoption function but does not escape arguments. :ztoo many fields (max 5): %rcSsg|] }|qS)strip).0sr r /lib/python3.7/site-packages/_pytest/warnings.py sz_setoption..rzinvalid lineno %rN) splitlen _OptionErrorappend _getaction _getcategoryint ValueError OverflowErrorfilterwarnings)Zwmodargpartsactionmessagecategorymodulelinenor r r _setoptions"    r!cCs0|d}|jddddd|jddd d dS) Nzpytest-warningsz-Wz--pythonwarningsrz=set which warnings to report, see -W option of python itself.)rhelprZlinelistzcEach line specifies a pattern for warnings.filterwarnings. Processed after -W and --pythonwarnings.)typer")ZgetgroupZ addoptionZaddini)parsergroupr r rpytest_addoption's r&cCs|dddS)NZmarkerszfilterwarnings(warning): add a warning filter to the given test. see https://docs.pytest.org/en/latest/warnings.html#pytest-mark-filterwarnings )Zaddinivalue_line)configr r rpytest_configure8sr(c cs|dp g}|d}tjdd}tjsHtjdtdtjdtdtjdt j dx|D]}t t|q^Wx|D]}t |qvW|dk rx.|j dd D]}x|j D]}t t|qWqWdVx$|D]} |jjt| ||d d qWWdQRXdS) z Context manager that catches warnings generated in the contained execution block. ``item`` can be None if we are not in the context of an item execution. Each warning captured triggers the ``pytest_warning_captured`` hook. ZpythonwarningsrT)recordalways)rerrorN)name)warning_messagewhenitem)kwargs)Z getoptionZgetiniwarningscatch_warningssys warnoptionsrDeprecationWarningPendingDeprecationWarningpytestZRemovedInPytest4Warningr!Z iter_markersargspytest_warning_captured call_historicdict) r'ihookr.r/Zcmdline_filtersZ inifilterslogrZmarkr-r r rcatch_warnings_for_item@s&      r>cCs|j}d}tjrltdd|jDrlg}x.|jD]$}|t|tjrNt|n|q0Wt |j|k}||_t ||j |j |j|j}|rt dt|t|S)zConvert a warnings.WarningMessage to a string, taking in account a lot of unicode shenaningans in Python 2. When Python 2 support is dropped this function can be greatly simplified. Fcss|]}t|tjVqdS)N) isinstancer UNICODE_TYPES)r mr r r psz(warning_record_to_str..z^Warning is using unicode non convertible to ascii, converting to a safe representation: {!r})rrZ_PY2anyr8rr?r@Z ascii_escapedlistr1 formatwarningrfilenamer linewarnformatZsafe_strUnicodeWarning)r-Zwarn_msgZunicode_warningZnew_argsrAmsgr r rwarning_record_to_stris*   rLT) hookwrapperZtryfirstc cs*t|j|jd|d dVWdQRXdS)NZruntest)r'r<r.r/)r>r'r<)r/r r rpytest_runtest_protocolsrNc cs.|j}t||jddd dVWdQRXdS)NZcollect)r'r<r.r/)r'r>hook)Zsessionr'r r rpytest_collectionsrP)rMc cs.|j}t||jddd dVWdQRXdS)Nr')r'r<r.r/)r'r>rO)Zterminalreporterr'r r rpytest_terminal_summarysrQc CsVtjdd$}tdt|tj||dWdQRX|jjt|ddddd dS) a This function should be used instead of calling ``warnings.warn`` directly when we are in the "configure" stage: at this point the actual options might not have been set, so we manually trigger the pytest_warning_captured hook so we can display this warnings in the terminal. This is a hack until we can sort out #2891. :param warning: the warning instance. :param hook: the hook caller :param stacklevel: stacklevel forwarded to warnings.warn T)r)r*) stacklevelNrr')r-r.r/)r0)r1r2 simplefilterr#rHr9r:r;)ZwarningrOrRZrecordsr r r_issue_warning_captureds rT)Z __future__rrrr3r1 contextlibrr7Z_pytestrZSHOW_PYTEST_WARNINGS_ARGr!r&r(r>rLZhookimplrNrPrQrTr r r rs"      )