File: //opt/alt/python37/lib/python3.7/site-packages/tests/unit/__pycache__/test_handler.cpython-37.pyc
B
�P�e� � @ s� d Z ddlZddlZddlZddlZddlZddlZddlZddlZddl m
Z
m Z m Z ddl
mZmZmZ ddlmZmZ dZG dd� de�Zejd d
� �Zd d � Zej�d
dd� d�dfdefddgdegfg�dd� �Z dd� Z dd� Z dd� Z ejj!ej"d dkd d �d d � �Z#ejj!ej"d dk d d �d!d"� �Z$d#d$� Z%d%d&� Z&d'd(� Z'd)d*� Z(d+d,� Z)d-d.� Z*d/d0� Z+d1d2� Z,d3d4� Z-d5d6� Z.d7d8� Z/d9d:� Z0dS );z�pytests for the formatting and construction of GELF logs by the graypy
logging handlers
.. note::
These tests mock sending to Graylog and do not require an active graylog
instance to operate.
� N)�BaseGELFHandler�GELFHTTPHandler�GELFTLSHandler)�handler�logger�formatted_logger)�MOCK_LOG_RECORD�MOCK_LOG_RECORD_NAMEu �c @ s e Zd Zdd� ZdS )� TestClassc C s dS )Nz <TestClass>� )�selfr r �H/opt/alt/python37/lib/python3.7/site-packages/tests/unit/test_handler.py�__repr__ s zTestClass.__repr__N)�__name__�
__module__� __qualname__r r r r r
r
s r
c c s^ y$t j�| d�� }|V W d Q R X W n4 tk
rX t j�| d�� }|V W d Q R X Y nX d S )N�send�emit)�mock�patch�object� Exception)r � mock_sendr r r
r $ s r c C s� | j g kst�| j \\\}}t|tj�rDt�tdd��|�� d��S yt�t
� |�� d��S t
j k
r� yt�|� d��S t
k
r� t�|d d� � d��S X Y nX d S )NF)�compresszutf-8���)Zcall_args_list�AssertionError�
isinstance�logging� LogRecord�json�loadsr �
makePickle�decode�zlib�
decompress�errorr )r �arg�_r r r
�get_mock_send_arg. s r( zmessage,expected)u €u €u €zutf-8� ��ac C s" |t �t�| ��d��ks t�d S )Nzutf-8)r r r Z_pack_gelf_dictr" r )�message�expectedr r r
� test_pack@ s
r- c C s~ y t d��W n" t k
r. | jddd� Y nX t|�}d|d ksHt�|d �d�sZt�|d �d�sz|d �d �szt�d
S ) zyCheck that a the ``full_message`` traceback info is passed when
the ``exc_info=1`` flag is given within a log messagez Syntax error�Failed� )�exc_info�
short_message� full_messagez"Traceback (most recent call last):zSyntaxError: Syntax errorzSyntaxError: Syntax error
N)� SyntaxErrorr% r( r �
startswith�endswith)r r r& r r r
� test_manual_exc_info_handlerO s r6 c C sz y t d��W n t k
r* | �d� Y nX t|�}d|d ksDt�|d �d�sVt�|d �d�sv|d �d�svt�d S )Nz Syntax errorr. r1 r2 z"Traceback (most recent call last):zSyntaxError: Syntax errorzSyntaxError: Syntax error
)r3 � exceptionr( r r4 r5 )r r r& r r r
� test_normal_exception_handlera s r8 c C s& | � d� t|�}d|d ks"t�d S )Nu Mensaje de registro españolr1 )r% r( r )r r r& r r r
� test_unicodeq s
r9 � z python2 only)�reasonc C s* | � d� t|�}dt |d ks&t�d S )Ns Broken � log messagezBroken %s log messager1 )r% r( �UNICODE_REPLACEMENTr )r r �decodedr r r
�test_broken_unicode_python2w s
r> z python3 onlyc C s& | � d� t|�}d|d ks"t�d S )Ns Broken � log messagezb'Broken \xde log message'r1 )r% r( r )r r r= r r r
�test_broken_unicode_python3� s
r? c C s> | j dddid� t|�}d|d ks*t�d|d ks:t�d S )Nz Log message�foo�bar)�extrar1 �_foo)r% r( r )r r r= r r r
�test_extra_field� s rD c C sF | j ddddgid� t|�}d|d ks.t�ddg|d ksBt�d S )Nz Log messager@ rA Zbaz)rB r1 rC )r% r( r )r r r= r r r
� test_list� s rE c C s@ | j ddt� id� t|�}d|d ks,t�d|d ks<t�d S )Nz Log messager@ )rB r1 z <TestClass>rC )r% r
r( r )r r r= r r r
�test_arbitrary_object� s rF c C sX t � ddddddd�}| jdd |id
� t|�}d |d ks@t�|�� |d ksTt�d S )
Ni� � r: � � � � z Log message�ts)rB zdatetime.datetimeZ_ts)�datetimer% r( r � isoformat)r r � timestampr= r r r
�Rtest_message_to_pickle_serializes_datetime_objects_instead_of_blindly_repring_them� s
rP c C s> | j dddid� t|�}d|d ks*t�d|d ks:t�d S )Nz Log message�status�OK)rB r1 Z_status)r% r( r )r r r= r r r
�test_status_field_issue� s rS c C s& t � } t�| t� d| d ks"t�d S )N�INFO�
level_name)�dictr Z_add_level_namesr r )� gelf_dictr r r
�test_add_level_name� s rX c C st t �� t�dd�kst�t �� t�dd�ks0t�t �� t�dd�ksHt�dt�dd�ks\t�dt�dd�kspt�dS )zETest all posible resolutions of :meth:`BaseGELFHandler._resolve_host`FNT� localhost� )�socket� gethostnamer Z
_resolve_hostr �getfqdnr r r r
�test_resolve_host� s
r^ c C s< t � } d}t�| |t� t| d ks(t�d| d ks8t�d S )Nz
test facilityZ_logger�facility)rV r Z_set_custom_facilityr r r )rW r_ r r r
�test_set_custom_facility� s
r` c C sJ x"| j D ]}t|t�rt�d� qW | �d� t|�}d|d ksFt�dS )z�Test the ability to set and modify the graypy handler's
:class:`logging.Formatter` and have the resultant ``short_message`` be
formatted by the set :class:`logging.Formatter`z)formatting not mocked for GELFHTTPHandlerz Log messagezERROR : Log messager1 N)�handlersr r �pytest�skipr% r( r )r r r r= r r r
�test_formatted_logger� s
rd c C s& t �t�� tddd� W dQ R X dS )z~Test constructing :class:`graypy.handler.BaseGELFHandler` with
specifying conflicting arguments ``fqdn`` and ``localname``TrY )Zfqdn� localnameN)rb �raises�
ValueErrorr r r r r
�test_invalid_fqdn_localhost� s rh c C s& t �t�� tddd� W dQ R X dS )z|Test constructing :class:`graypy.handler.GELFTLSHandler` with
incorrect arguments specifying server ca cert verificationz 127.0.0.1T)�validateN)rb rf rg r r r r r
�test_invalid_ca_certs� s rj c C s& t �t�� tddd� W dQ R X dS )z}Test constructing :class:`graypy.handler.GELFTLSHandler` with
incorrect arguments specifying client cert/key verificationz 127.0.0.1z /dev/null)�keyfileN)rb rf rg r r r r r
�test_invalid_client_certs� s rl )1�__doc__rM r r r[ �sysr# r rb Zgraypy.handlerr r r Z tests.helperr r r Ztests.unit.helperr r r< r r
Z
yield_fixturer r( �markZ parametrize�encoder- r6 r8 r9 Zskipif� version_infor> r? rD rE rF rP rS rX r^ r` rd rh rj rl r r r r
�<module> sL
"
"