B q\;@sddlZddlZddlZddlZddlmZddlmZddlm Z m Z ddl m Z m Z ejZejZy eZWnek rdZYnXddZeZd d Zd d Zd dZddZddZddZddZddZddZej !e"dddZ#ej $e"dddZ%ej $e"dd d!Z&ej j'd"d#d$d%Z(ej )d&dd'd(d)d*gd+d,Z*d-d.Z+d/d0Z,d1d2Z-ej )d&dd'd(d)d*gd3d4Z.d5d6Z/d7d8Z0d9d:Z1dS);N)catch_warnings)log) LoggingErrorconf)AstropyWarningAstropyUserWarningcCsLttttdt_dt_ttr8t t rHt dS)N) importlibreloadwarningssysr_showwarning_origZ_excepthook_origZ _set_defaultsZwarnings_logging_enableddisable_warnings_loggingZexception_logging_enableddisable_exception_logging)Zfunctionr8lib/python3.7/site-packages/astropy/tests/test_logger.pysetup_functions  rc Cs6tt}tWdQRX|jjddks2tdS)Nrz%Warnings logging has not been enabled)pytestraisesrrrvalueargsAssertionError)errr'test_warnings_logging_disable_no_enable4s rc Cs>ttt}tWdQRX|jjddks:tdS)Nrz)Warnings logging has already been enabled)renable_warnings_loggingrrrrrr)rrrr"test_warnings_logging_enable_twice:s rc CsHtddt_tt}tWdQRX|jj ddksDt dS)NcSsdS)NrrrrrCsz2test_warnings_logging_overridden..rzhCannot disable warnings logging: warnings.showwarning was not set by this logger, or has been overridden) rrr showwarningrrrrrrr)rrrr test_warnings_logging_overriddenAs   rc CsBt&}t}tdtWdQRXWdQRXt|dksBtt|dksRt|djj ddksjtt6}t t}tdtWdQRXt WdQRXt|dkstt|dkst|dj dkst|dj dst|djdkstt@}t t}tdttdWdQRXt WdQRXt|dksdtt|dksvt|dj dkst|dj dst|djdkst|djj ddkstt&}t}tdtWdQRXWdQRXt|dkstt|dks$t|djj ddks>tdS)NzThis is a warningrrWARNINGzastropy.tests.test_loggerz)This is another warning, not from Astropy)rr log_to_listr warnrlenrmessagerrr levelname startswithorigin) warn_listlog_listrrrtest_warnings_loggingIsD       r)c CsGdddt}t6}tt}td|WdQRXtWdQRXt|dksbt t|dksrt |dj dkst |dj dst |dj dkst dS) Nc@s eZdZdS)zJtest_warnings_logging_with_custom_class..CustomAstropyWarningClassN)__name__ __module__ __qualname__rrrrCustomAstropyWarningClasswsr-zThis is a warningrrrz,CustomAstropyWarningClass: This is a warningzastropy.tests.test_logger)rrrrr r r!rr"rr$r#r%r&)r-r'r(rrr'test_warnings_logging_with_custom_classvs r.c Csddlm}m}t4}tt}||dWdQRXtWdQRXt|dks`t t|dkspt |dj dkst |dj d}|st |dj dkst dS)Nr)W02vo_warn)abrrzGW02: ?:?:?: W02: a attribute 'b' is invalid. Must be a standard XML idzastropy.tests.test_logger)Zastropy.io.votable.exceptionsr/r0rrrr rr"rr$r#r%r&)r/r0r'r(xrrr,test_warning_logging_with_io_votable_warnings r4cCsJGddd}t|tjd<zttdtddWdtjd=XdS)z Regression test for https://github.com/astropy/astropy/issues/2671 This test actually puts a goofy fake module into ``sys.modules`` to test this problem. c@seZdZddZdS)z8test_import_error_in_warning_logging..FakeModulecSs tddS)Nz._showwarning should ignore any exceptions here) ImportError)selfattrrrr __getattr__szDtest_import_error_in_warning_logging..FakeModule.__getattr__N)r*r+r,r8rrrr FakeModulesr9zzRegression test for #2671zrN)rrr modulesr rr)r9rrr$test_import_error_in_warning_loggings  r;c Cs6tt}tWdQRX|jjddks2tdS)Nrz&Exception logging has not been enabled)rrrrrrrr)rrrr(test_exception_logging_disable_no_enables r<c Cs>ttt}tWdQRX|jjddks:tdS)Nrz*Exception logging has already been enabled)renable_exception_loggingrrrrrr)rrrr#test_exception_logging_enable_twices r>zip is not Nonec CsHtddt_tt}tWdQRX|jj ddksDt dS)NcSsdS)Nr)etypeZevaluetbrrrrsz3test_exception_logging_overridden..rzcCannot disable exception logging: sys.excepthook was not set by this logger, or has been overridden) rr=r excepthookrrrrrrr)rrrr!test_exception_logging_overriddens   rBc Csy t}tdWdQRXWnBtk rb}z$tjt|jddksRtWdd}~XYn Xdsltt|dks|ty(t t}tdWdQRXWnBtk r}z$tjt|jddkstWdd}~XYn Xdstt|dkst|dj dkst|dj ds,t|dj dks@tty t}tdWdQRXWnFtk r}z&tjt|jddkstWdd}~XYn Xdstt|dkstdS)NzThis is an ExceptionrFrERRORzException: This is an Exceptionzastropy.tests.test_logger)rr Exceptionr rAexc_inforrr"r=r$r#r%r&r)r(excrrrtest_exception_loggings: $ $ & rGc Csddlm}|t}y*tt}|dWdQRXWnDtk r}z&tj t |j d dsrt Wdd}~XYn Xdst t|dkst |djdkst |dj dst |djd kst dS) Nr)HomogeneousListZfooz3homogeneous list must contain only objects of type FrrCz>TypeError: homogeneous list must contain only objects of type zastropy.utils.collections)Zastropy.utils.collectionsrHintrr=r append TypeErrorr rArErr%rr"r$r#r&)rHlr(rFrrrtest_exception_logging_origins      rMz8Infinite recursion on Python 3.5+, probably a real issue)reasonc Csy&tt }tWdQRXWn0tk rV}ztjtWdd}~XYn Xds`tt|dkspt|dj dkst|dj dkst|dj dkstdS)z Regression test for a crash that occurred on Python 3 when logging an exception that was instantiated with no arguments (no message, etc.) Regression test for https://github.com/astropy/astropy/pull/4056 NFrrrCz%Exception [astropy.tests.test_logger]zastropy.tests.test_logger) rr=r rDr rArErr"r$r#r&)r(rFrrr(test_exception_logging_argless_exception s   rOlevelDEBUGINFOWARNrCc Cstj}zR|dk rt|t.}tdtdtdtdWdQRXWdt|X|dkrttj }|dkrt |dkst nL|dkrt |dkst n2|d krt |d kst n|d krt |d kst |d j d kst |d j dst |d jdkst t |d krb|d j dks8t |d j dsNt |d jdksbt t |dkr|d j dkst |d j dst |d jdkst t |dkr|dj dkst |dj dst |djdkst dS)Nz Error messagezWarning messagezInformation messagez Debug messagerQrRrSrCrrzastropy.tests.test_loggerr)rrPsetLevelr errorwarninginfodebugr log_levelr"rr$r#r%r&)rP orig_levelr(rrrtest_log_to_list"sF      r^c CsNtjdd}tdtdWdQRXt|dkrF|djdksJtdS)NrC) filter_levelz Error messagezWarning messagerr)rr rXrYr"r$r)r(rrrtest_log_to_list_levelWs r`c Cs@tjdd}tdtdWdQRXt|dksZmarkZskipifrirBZxfailrGrMskiprOZ parametrizer^r`rbrcrurwrxryrrrrsH    -  * 5   :