File: //opt/alt/python37/lib/python3.7/site-packages/tests/integration/__pycache__/helper.cpython-37.pyc
B
�P�e� � @ sl d Z ddlmZ ddlmZ ddlZdd� Zddd d
d d d
ddg ZdZddd�Z dd� Z
dd� Z dd� Z dS )zAhelper functions for testing graypy with a local Graylog instance� )�sleep)�uuid4Nc C s
t t� �S )N)�strr � r r �I/opt/alt/python37/lib/python3.7/site-packages/tests/integration/helper.py�get_unique_message s r �messageZ full_message�source�level�func�file�line�moduleZ logger_namezYhttp://127.0.0.1:9000/api/search/universal/relative?query=message:"{0}"&range=300&fields=c C s\ |r|ng }d}xFyt t| |�| d�S tk
rR td� |dkrF� |d7 }Y qX qW dS )zdSearch for a given log message (with possible additional fields)
within a local Graylog instancer )� api_response�wanted_message� � � N)�_parse_api_response�_get_api_response�
ValueErrorr )r �fields�triesr r r �get_graylog_response s r c C s t �| �d�tt| �� S )Nz%2C)� BASE_API_URL�format�join�set�DEFAULT_FIELDS)r r r r r �_build_api_string1 s r c C s" t | |�}tj|dddid�}|S )N)�adminr �acceptzapplication/json)�auth�headers)r �requests�get)r r �urlr r r r r 5 s
r c C s\ | j dkst�t| �� � x,| �� d D ] }|d d |kr(|d S q(W td�|| ���d S )N�� �messagesr z0wanted_message: '{}' not within api_response: {})� status_code�AssertionError�print�jsonr r )r r r r r r r = s r )N)
�__doc__�timer �uuidr r$ r r r r r r r r r r r �<module> s$