Файловый менеджер - Редактировать - /opt/imh-python/lib/python3.9/site-packages/influxdb/tests/__pycache__/client_test.cpython-39.pyc
Ðазад
a p�h,� � @ s� d Z ddlmZ ddlmZ ddlmZ ddlmZ ddlZddlZddlZddl Z ddl Z ddl Z ddl Z ddl Z ddlZddlZddlZddlmZ ddlmZ dd lmZ dd lmZ dd d�Zddd�ZG dd� dej�Z G dd� de�Z dS )ar Unit tests for the InfluxDBClient. NB/WARNING: This module implements tests for the InfluxDBClient class but does so + without any server instance running + by mocking all the expected responses. So any change of (response format from) the server will **NOT** be detected by this module. See client_test_with_server.py for tests against a running server instance. � )�absolute_import)�division)�print_function)�unicode_literalsN)�raises)�HTTPConnection)�InfluxDBClient)� ResultSet�� � c C s t �� }| |_|�d�|_|S )N�utf8)�requests�Response� status_code�encode�_content)r �content�resp� r ��/root/rpmbuild/BUILDROOT/imh-python39-modules-3.9.7-92.el8.x86_64/opt/imh-python/lib/python3.9/site-packages/influxdb/tests/client_test.py�_build_response_object* s r �GETc s, �� � �� ��fdd�}tjj| jd|d�S )Nc s| � }�|� dd�ksJ ��dkr\|� dd�}|dur\t|t�sBJ �|tj|dd�ksXJ �|}t|t�spt�|�}t�|d�S ) z(Request content from the mocked session.�methodr �POST�dataNT)�strict)r r )�get� isinstance�str�json�loads�dumpsr )�args�kwargs�cr �r r r r r �request4 s z _mocked_session.<locals>.requestr& )� side_effect)�upper�mock�patch�object�_session)�clir r r r& r r% r �_mocked_session1 s r. c @ s: e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dd� Z dd� Z dd� Z dd� Z dd� Zdd� Zdd� Zd d � Zee�d d � �Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd(d)� Zee�d*d+� �Zd,d-� Zd.d/� Z d0d1� Z e � d2�d3d4� �Z ee�d5d6� �Z!d7d8� Z"d9d:� Z#d;d<� Z$ee�d=d>� �Z%d?d@� Z&dAdB� Z'dCdD� Z(dEdF� Z)ee�dGdH� �Z*dIdJ� Z+dKdL� Z,dMdN� Z-dOdP� Z.ee�dQdR� �Z/dSdT� Z0dUdV� Z1dWdX� Z2dYdZ� Z3d[d\� Z4ee�d]d^� �Z5d_d`� Z6ee�dadb� �Z7dcdd� Z8e9�:de�dfdg� �Z;e9�:de�dhdi� �Z<e9�:de�djdk� �Z=e9�:de�dldm� �Z>dndo� Z?dpdq� Z@drds� ZAee�dtdu� �ZBdvdw� ZCee�dxdy� �ZDdzd{� ZEee�d|d}� �ZFd~d� ZGee�d�d�� �ZHd�d�� ZIee�d�d�� �ZJd�d�� ZKee�d�d�� �ZLd�d�� ZMee�d�d�� �ZNd�d�� ZOee�d�d�� �ZPd�d�� ZQd�d�� ZRd�d�� ZSd�d�� ZTd�d�� ZUd�d�� ZVd�d�� ZWd�d�� ZXd�S )��TestInfluxDBClientz%Set up the TestInfluxDBClient object.c C sB t �dt� tdddd�| _dddd �d d d id �g| _d| _dS )z4Initialize an instance of TestInfluxDBClient object.�error� localhost� �username�password�cpu_load_short�server01�us-west��host�regionz2009-11-10T23:00:00.123456Z�value�{�G�z�?�� measurement�tags�time�fieldsz%influxdb://uSr:pWd@my.host.fr:1886/dbN)�warnings� simplefilter� FutureWarningr r- � dummy_points� dsn_string��selfr r r �setUpS s ����zTestInfluxDBClient.setUpc C s� t ddddd�}| �d|j� t ddddddd�}| �d |j� t ddddddd d �}| �d |j� t ddddddd d �}| �d |j� t dddddddd �}| �d |j� d S )z5Set up the test schema for TestInfluxDBClient object.r9 r2 r3 r4 �databasezhttp://host:8086T)�sslzhttps://host:8086Zsomepath)rK �pathzhttps://host:8086/somepathNz /somepath)r � assertEqual�_baseurl�rH r- r r r � test_schemei s* � � � �z TestInfluxDBClient.test_schemec C s� t �d�}| �d|j� t �| j�}| �d|j� | �d|j� | �d|j� | �d|j� | �|j � t �d| j �}| � |j � t �d| j �}| �d |j� t jd| j fi d d i��}| �d|j� d S ) z>Set up the test datasource name for TestInfluxDBClient object.zinfluxdb://192.168.0.1:1886zhttp://192.168.0.1:1886zhttp://my.host.fr:1886ZuSrZpWd�dbzudp+zhttps+zhttps://my.host.fr:1886rK FN) r Zfrom_dsnrM rN rF � _username� _password� _database� assertFalseZ_use_udp� assertTruerO r r r �test_dsn� s �zTestInfluxDBClient.test_dsnc C sT t ddd�}| �|jjd� | �t�� t dd�}W d � n1 sF0 Y dS )z=Test mutual TLS authentication for TestInfluxDBClient object.Tz /etc/pki/tls/private/dummy.crt)rK �cert)rX N)r rM r, rX � assertRaises� ValueErrorrO r r r � test_cert� s z TestInfluxDBClient.test_certc C s, t ddddd�}|�d� | �d|j� dS )z2Test switch database in TestInfluxDBClient object.r9 r2 r3 r4 rJ Zanother_databaseN)r Zswitch_databaserM rT rO r r r �test_switch_database� s z'TestInfluxDBClient.test_switch_databasec C s< t ddddd�}|�dd� | �d|j� | �d|j� dS ) z.Test switch user in TestInfluxDBClient object.r9 r2 r3 r4 rJ Zanother_usernameZanother_passwordN)r Z switch_userrM rR rS rO r r r �test_switch_user� s z#TestInfluxDBClient.test_switch_userc C s� t �� �d}|jt jddd� tdd�}|�dddd d d �d d did�gd�� | �|jjd� W d � n1 sr0 Y dS )�(Test write in TestInfluxDBClient object.�http://localhost:8086/write�� �r rQ �rJ �mydb�mypolicyr5 r6 r7 r8 �2009-11-10T23:00:00Zr; r< r= �rJ ZretentionPolicy�points�K cpu_load_short,host=server01,region=us-west value=0.64 1257894000000000000 N) � requests_mock�Mocker� register_urir r �writerM � last_request�body�rH �mr- r r r � test_write� s. � ���� �z TestInfluxDBClient.test_writec C sj t �� �N}|jt jddd� tdd�}|�| j� | �d|jj � d�� W d � n1 s\0 Y dS ) �0Test write points for TestInfluxDBClient object.r_ r` ra rQ rb �Kcpu_load_short,host=server01,region=us-west value=0.64 1257894000123456000 �utf-8N� ri rj rk r r � write_pointsrE rM rm rn �decodero r r r �test_write_points� s � � �z$TestInfluxDBClient.test_write_pointsc C s� t �� ��}|jt jddd� tddd�}|�ddd d d d �d ddid�gd�� t�� }tj d|dd��}|�d� W d � n1 s�0 Y | � |j j |� � � W d � n1 s�0 Y dS )r^ r_ r` ra rQ T�rJ �gziprc rd r5 r6 r7 r8 re r; r< r= rf � �w�� compresslevel�fileobj�moderh N)ri rj rk r r rl �io�BytesIOrz �GzipFilerM rm rn �getvalue�rH rp r- � compressed�fr r r �test_write_gzip� sB � ���� ��"�z"TestInfluxDBClient.test_write_gzipc C s� t �� ��}|jt jddd� tddd�}|�| j� t�� }t j d|dd ��}|� d � W d � n1 sl0 Y | � |j j|�� � W d � n1 s�0 Y d S ) rr r_ r` ra rQ Try r{ r| r} �K cpu_load_short,host=server01,region=us-west value=0.64 1257894000123456000 N)ri rj rk r r rv rE r� r� rz r� rl rM rm rn r� r� r r r �test_write_points_gzip s0 � ���"�z)TestInfluxDBClient.test_write_points_gzipc C sv t �� �Z}|jt jddd� tdd�}|j| jdddid d � | �d |jj � d �� W d � n1 sh0 Y d S )z6Test write points attrs for TestInfluxDBClient object.r_ r` ra rQ rb �testdb�tagZhelloZ somepolicy)rJ r? Zretention_policyzUcpu_load_short,host=server01,region=us-west,tag=hello value=0.64 1257894000123456000 rt Nru ro r r r �%test_write_points_toplevel_attributes s" � � �z8TestInfluxDBClient.test_write_points_toplevel_attributesc C s� dddidddid�dd d iddd id�dd d iddd id�g}d}t �� �D}|jt jddd� tdd�}|j|dddd�dd� W d � n1 s�0 Y | �|jd� | �||jj � d�� dS )z6Test write points batch for TestInfluxDBClient object.� cpu_usage�unit�percentre r; �G�z�(@r= �network� direction�in� �^@�out� (@�Rnetwork,direction=out,host=server01,region=us-west value=12.0 1257894000000000000 r_ r` ra rQ rb r6 r7 r8 � �rg rJ r? Z batch_sizeNrt � ri rj rk r r rv rM Z call_countrm rn rw )rH rE �expected_last_bodyrp r- r r r �test_write_points_batch3 s: ���� � � ��$ �z*TestInfluxDBClient.test_write_points_batchc C s� dddidddid�dd d iddd id�dd d iddd id�g}dd� |D �}d}t �� �D}|jt jddd� tdd�}|j|dddd�dd� W d � n1 s�0 Y | �|jd� | �||jj � d �� dS ) z@Test write points batch from a generator for TestInfluxDBClient.r� r� r� re r; r� r= r� r� r� r� r� r� c s s | ] }|V qd S )Nr )�.0�pointr r r � <genexpr>Z � zGTestInfluxDBClient.test_write_points_batch_generator.<locals>.<genexpr>r� r_ r` ra rQ rb r6 r7 r8 r� r� Nrt r� )rH rE Zdummy_points_generatorr� rp r- r r r �!test_write_points_batch_generatorP s< ����� � ��$ �z4TestInfluxDBClient.test_write_points_batch_generatorc C sn t � t jt j�}t�dd�}|�d|f� tdddddd|d �}|�| j� |� d �\}}| � d |� � � d S ) z4Test write points UDP for TestInfluxDBClient object.� �@ �0.0.0.0r1 r2 �root�testT�Zuse_udpZudp_port� rs N) �socket�AF_INET� SOCK_DGRAM�random�randint�bindr rv rE �recvfromrM rw �rH �s�portr- Z received_data�addrr r r �test_write_points_udpn s � �z(TestInfluxDBClient.test_write_points_udpc C sJ t ddddd�}t|dd�� |�g � W d � n1 s<0 Y dS ) z5Test write points fail for TestInfluxDBClient object.r9 r2 r3 r4 rQ �post�� N)r r. rv rO r r r �test_write_points_fails� s z*TestInfluxDBClient.test_write_points_failsc C s t �� ��}|jt jddd� tdd�}|j| jdd� | �d|jj � |j| jd d� | �d |jj � |j| jd d� | �d |jj � |j| jd d� | �d|jj � |j| jdd� | �d|jj � |j| jdd� | �d|jj � W d � n1 s�0 Y dS )�?Test write points with precision for TestInfluxDBClient object.r_ r` ra rQ rb �n�Ztime_precisionr� �u�H cpu_load_short,host=server01,region=us-west value=0.64 1257894000123456 �ms�E cpu_load_short,host=server01,region=us-west value=0.64 1257894000123 r� �B cpu_load_short,host=server01,region=us-west value=0.64 1257894000 rp �@ cpu_load_short,host=server01,region=us-west value=0.64 20964900 �h�> cpu_load_short,host=server01,region=us-west value=0.64 349415 N) ri rj rk r r rv rE rM rm rn ro r r r � test_write_points_with_precision� sJ � ������z3TestInfluxDBClient.test_write_points_with_precisionc C sr t �� �V}|jt jddd� tdd�}|j| jdd� | �|jj dgdgd�� W d � n1 sd0 Y d S ) zATest write points with consistency for TestInfluxDBClient object.r_ r` ra rQ rb �any�� consistency)rQ r� N) ri rj rk r r rv rE rM rm �qsro r r r �"test_write_points_with_consistency� s � �z5TestInfluxDBClient.test_write_points_with_consistencyc C s@ t � t jt j�}t�dd�}|�d|f� tdddddd|d �}|j| jd d � |� d �\}}| � d |� |j| jdd � |� d �\}}| � d|� |j| jdd � |� d �\}}| � d|� |j| jdd � |� d �\}}| � d|� |j| jdd � |� d �\}}| � d|� |j| jdd � |� d �\}}| � d|� dS )r� r� r� r� r1 r2 r� r� Tr� r� r� r� r� r� r� r� r� r� r� rp r� r� r� N) r� r� r� r� r� r� r rv rE r� rM r� r r r �$test_write_points_with_precision_udp� sV �������z7TestInfluxDBClient.test_write_points_with_precision_udpc C sF t � }| �td�� |j| jdd� W d � n1 s80 Y dS )z<Test write points w/bad precision TestInfluxDBClient object.zHInvalid time precision is given. \(use 'n', 'u', 'ms', 's', 'm' or 'h'\)�gr� N)r ZassertRaisesRegexp� Exceptionrv rE rO r r r � test_write_points_bad_precision s ��z2TestInfluxDBClient.test_write_points_bad_precisionc C sD t � }| �t�� |j| jdd� W d � n1 s60 Y dS )z*Test write points w/bad consistency value.�boor� N)r rY rZ rv rE rO r r r �!test_write_points_bad_consistency s �z4TestInfluxDBClient.test_write_points_bad_consistencyc C sJ t ddddd�}t|dd�� |�g � W d � n1 s<0 Y dS ) zATest write points w/precision fail for TestInfluxDBClient object.r9 r2 r3 r4 rQ r� r� N)r r. Zwrite_points_with_precisionrO r r r �&test_write_points_with_precision_fails s z9TestInfluxDBClient.test_write_points_with_precision_failsc C sn d}t �� �N}|jt jd|d� | j�d�}| �t|d �� �ddd�g� W d � n1 s`0 Y d S ) z0Test query method for TestInfluxDBClient object.��{"results": [{"series": [{"measurement": "sdfsdfsdf", "columns": ["time", "value"], "values": [["2009-11-10T23:00:00Z", 0.64]]}]}, {"series": [{"measurement": "cpu_load_short", "columns": ["time", "value"], "values": [["2009-11-10T23:00:00Z", 0.64]]}]}]}�http://localhost:8086/query��text�select * from foor r< re �r; r@ N) ri rj rk r r- �query�assertListEqual�list� get_points�rH �example_responserp �rsr r r � test_query' s � � �z TestInfluxDBClient.test_queryc C s� t t�d��}t�� �V}|jtjdddiddi|d� | j�d�}| � t |� � �dd d �g� W d � n1 sr0 Y d S ) z.Test query method with a messagepack response.Z�81a7726573756c74739182ac73746174656d656e745f696400a67365726965739183a46e616d65a161a7636f6c756d6e7392a474696d65a176a676616c7565739192c70c05000000005d26178a019096c8cb3ff0000000000000r� �Acceptzapplication/x-msgpackz Content-Type)Zrequest_headers�headersr zselect * from ag �?z2019-07-10T16:51:22.026253Z)�vr@ N) �bytes� bytearray�fromhexri rj rk r r- r� r� r� r� r� r r r �test_query_msgpack> s � � �z%TestInfluxDBClient.test_query_msgpackc C sn d}t �� �N}|jt jd|d� | j�d�}| �t|d �� �ddd�g� W d � n1 s`0 Y d S ) z Test SELECT.*INTO is POSTed.r� r� r� z select * INTO newmeas from foor r< re r� N) ri rj rk r r- r� r� r� r� r� r r r �test_select_into_postU s � � �z(TestInfluxDBClient.test_select_into_postzNot implemented for 0.9c C s� t dd�}g d�g d�g d�g d�g d�gdg d �d �}t�|�t�|� }t�� �<}|jtjd |d � | �|jd dd�||g� W d � n1 s�0 Y dS )z1Test chunked query for TestInfluxDBClient object.rQ rb )l �&�&iA� i� )l ��&i1u � )l Q�&i!N i )l t�&�' i+ )l "&� &r� � Zfoo)r@ Zsequence_number�val)rg r> �columnsz"http://localhost:8086/db/db/seriesr� r� T)�chunkedN) r r r! ri rj rk r r� r� )rH r- Zexample_objectr� rp r r r �test_query_chunkedl s, ��� � �z%TestInfluxDBClient.test_query_chunkedc C s> t | jdd�� | j�d� W d � n1 s00 Y dS )z0Test query failed for TestInfluxDBClient object.r � zselect column_one from foo;N)r. r- r� rG r r r �test_query_fail� s z"TestInfluxDBClient.test_query_failc C sZ t �� �>}|jt jddddid� | j�� }| �|d� W d � n1 sL0 Y dS )z$Test ping querying InfluxDB version.�http://localhost:8086/pingr` �X-Influxdb-Version�1.2.3�r r� N)ri rj rk r r- �pingrM )rH rp �versionr r r � test_ping� s � z TestInfluxDBClient.test_pingc C sb t �� �F}|jt jddd� | j�d� | �|jjd d d� W d � n1 sT0 Y dS ) z3Test create database for TestInfluxDBClient object.r� �{"results":[{}]}r� �new_db�qr zcreate database "new_db"N� ri rj rk r r- �create_databaserM rm r� �rH rp r r r �test_create_database� s � �z'TestInfluxDBClient.test_create_databasec C sb t �� �F}|jt jddd� | j�d� | �|jjd d d� W d � n1 sT0 Y dS ) z<Test create db w/numeric name for TestInfluxDBClient object.r� r� r� �123r� r zcreate database "123"Nr� r� r r r �"test_create_numeric_named_database� s � �z5TestInfluxDBClient.test_create_numeric_named_databasec C s> t | jdd�� | j�d� W d � n1 s00 Y dS )z8Test create database fail for TestInfluxDBClient object.r� r� r� N)r. r- r� rG r r r �test_create_database_fails� s z-TestInfluxDBClient.test_create_database_failsc C sb t �� �F}|jt jddd� | j�d� | �|jjd d d� W d � n1 sT0 Y dS ) z1Test drop database for TestInfluxDBClient object.r� r� r� r� r� r zdrop database "new_db"N� ri rj rk r r- Z drop_databaserM rm r� r� r r r �test_drop_database� s � �z%TestInfluxDBClient.test_drop_databasec C sb t �� �F}|jt jddd� | j�d� | �|jjd d d� W d � n1 sT0 Y dS ) z4Test drop measurement for TestInfluxDBClient object.r� r� r� Znew_measurementr� r z"drop measurement "new_measurement"N) ri rj rk r r- Zdrop_measurementrM rm r� r� r r r �test_drop_measurement� s � �z(TestInfluxDBClient.test_drop_measurementc C sb t �� �F}|jt jddd� | j�d� | �|jjd d d� W d � n1 sT0 Y dS ) z3Test drop numeric db for TestInfluxDBClient object.r� r� r� r� r� r zdrop database "123"Nr r� r r r � test_drop_numeric_named_database� s � �z3TestInfluxDBClient.test_drop_numeric_named_databasec C sz ddddgdggdgd�gigi}t | jdd t�|���. | �| j�� ddiddig� W d � n1 sl0 Y d S ) z9Test get list of databases for TestInfluxDBClient object.�results�seriesZ databasesZnew_db_1Znew_db_2�name)r �valuesr� r r N)r. r- r r! r� �get_list_database�rH r r r r �test_get_list_database� s ���� �z)TestInfluxDBClient.test_get_list_databasec C sF t dddd�}t|dd�� |�� W d � n1 s80 Y dS )z8Test get list of dbs fail for TestInfluxDBClient object.r9 r2 r3 r4 r r� N)r r. r rO r r r � test_get_list_database_fails� s z/TestInfluxDBClient.test_get_list_database_failsc C sz ddddgdgdggd�gigi}t | jdd t�|���. | �| j�� ddiddig� W d � n1 sl0 Y d S ) z<Test get list of measurements for TestInfluxDBClient object.r r � measurementsr �cpuZdisk�r r� r r r N)r. r- r r! r� Zget_list_measurementsr r r r �test_get_list_measurements s ���� �z-TestInfluxDBClient.test_get_list_measurementsc C sp dddgdggdgd�gigi}t | jddt�|���& | �| j�� ddg� W d � n1 sb0 Y d S ) z,Test get a list of series from the database.r r �+cpu_load_short,host=server01,region=us-westz)memory_usage,host=server02,region=us-east�key�r r� r r N�r. r- r r! r� �get_list_seriesr r r r �test_get_list_series s ���� ��z'TestInfluxDBClient.test_get_list_seriesc C sn dddggdgd�gigi}t | jddt�|���( | �| jjdd �dg� W d � n1 s`0 Y d S ) z6Test get a list of series from the database by filter.r r r r r r r r5 )r> Nr r r r r �%test_get_list_series_with_measurement) s ���� �z8TestInfluxDBClient.test_get_list_series_with_measurementc C sr dddggdgd�gigi}t | jddt�|���, | �| jjdd id �dg� W d � n1 sd0 Y d S ) z4Test get a list of series from the database by tags.r r r r r r r r: r7 )r? Nr r r r r � test_get_list_series_with_tags: s ���� �z1TestInfluxDBClient.test_get_list_series_with_tagsc C sF t dddd�}t|dd�� |�� W d � n1 s80 Y dS )z5Test get a list of series from the database but fail.r9 r2 r3 r4 r r� N)r r. r rO r r r �test_get_list_series_failsK s z-TestInfluxDBClient.test_get_list_series_failsc C sp d}t �� �P}|jt jd|d� | jjdddddd � | �|jjd d d � W d � n1 sb0 Y d S )z=Test create default ret policy for TestInfluxDBClient object.r� r� r� �somename�1d� TrQ )�defaultrJ r� r z^create retention policy "somename" on "db" duration 1d replication 4 shard duration 0s defaultN� ri rj rk r r- Zcreate_retention_policyrM rm r� �rH r� rp r r r �$test_create_retention_policy_defaultR s � ��z7TestInfluxDBClient.test_create_retention_policy_defaultc C sn d}t �� �N}|jt jd|d� | jjddddd� | �|jjd d d � W d � n1 s`0 Y d S ) z;Test create retention policy for TestInfluxDBClient object.r� r� r� r r r rQ rb r� r zVcreate retention policy "somename" on "db" duration 1d replication 4 shard duration 0sNr r r r r � test_create_retention_policyf s ���z/TestInfluxDBClient.test_create_retention_policyc C sp d}t �� �P}|jt jd|d� | jjdddddd � | �|jjd d d � W d � n1 sb0 Y d S )z:Test create retention policy with a custom shard duration.r� r� r� Z somename2r r rQ �1h)rJ �shard_durationr� r zWcreate retention policy "somename2" on "db" duration 1d replication 4 shard duration 1hNr r r r r �+test_create_retention_policy_shard_durationz s ���z>TestInfluxDBClient.test_create_retention_policy_shard_durationc C sr d}t �� �R}|jt jd|d� | jjdddddd d � | �|jjd d d � W d � n1 sd0 Y dS )z;Test create retention policy with a default shard duration.r� r� r� Z somename3r r rQ r" T)rJ r# r r� r z_create retention policy "somename3" on "db" duration 1d replication 4 shard duration 1h defaultNr r r r r �3test_create_retention_policy_shard_duration_default� s ���zFTestInfluxDBClient.test_create_retention_policy_shard_duration_defaultc C s� d}t �� ��}|jt jd|d� | jjdddd� | �|jjd d d � | jjddd d � | �|jjd d d � | jjdddd� | �|jjd d d� | jjdddd� | �|jjd d d� W d � n1 s�0 Y dS )z:Test alter retention policy for TestInfluxDBClient object.r� r� r� r rQ Z4d)�durationr� r z5alter retention policy "somename" on "db" duration 4dr )Z replicationz7alter retention policy "somename" on "db" replication 4r" )r# z;alter retention policy "somename" on "db" shard duration 1hT)r z1alter retention policy "somename" on "db" defaultN) ri rj rk r r- �alter_retention_policyrM rm r� r r r r �test_alter_retention_policy� sF � �� �� �� ��z.TestInfluxDBClient.test_alter_retention_policyc C sL t dddd�}t|dd�� | j�dd� W d � n1 s>0 Y d S ) z<Test invalid alter ret policy for TestInfluxDBClient object.r9 r2 r3 r4 r � r rQ N)r r. r- r' rO r r r �#test_alter_retention_policy_invalid� s z6TestInfluxDBClient.test_alter_retention_policy_invalidc C sh d}t �� �H}|jt jd|d� | j�dd� | �|jjd d d� W d � n1 sZ0 Y d S ) z9Test drop retention policy for TestInfluxDBClient object.r� r� r� r rQ r� r z(drop retention policy "somename" on "db"N) ri rj rk r r- �drop_retention_policyrM rm r� r r r r �test_drop_retention_policy� s ��z-TestInfluxDBClient.test_drop_retention_policyc C sJ t dddd�}t|dd�� |�dd� W d � n1 s<0 Y d S ) z:Test failed drop ret policy for TestInfluxDBClient object.r9 r2 r3 r4 �deleter� r rQ N)r r. r+ rO r r r � test_drop_retention_policy_fails� s z3TestInfluxDBClient.test_drop_retention_policy_failsc C s` d}t �� �@}|jt jd|d� | �| j�d�dddd�g� W d � n1 sR0 Y d S ) z:Test get retention policies for TestInfluxDBClient object.zq{"results": [{"series": [{"values": [["fsfdsdf", "24h0m0s", 2]], "columns": ["name", "duration", "replicaN"]}]}]}r� r� rQ Z24h0m0sZfsfdsdfr� )r& r ZreplicaNN)ri rj rk r r� r- Zget_list_retention_policiesr r r r � test_get_list_retention_policies� s � � ��z3TestInfluxDBClient.test_get_list_retention_policieszrequests.Session.requestc C s4 G dd� dt �}|� j|_tdd�}|�| j� dS )z0Test that two connection errors will be handled.c @ s e Zd ZdZdd� Zdd� ZdS )z9TestInfluxDBClient.test_request_retry.<locals>.CustomMock�#Create custom mock object for test.c S s d| _ d S �Nr ��irG r r r �__init__ s zBTestInfluxDBClient.test_request_retry.<locals>.CustomMock.__init__c _ s2 | j d7 _ | j dk r tjj�t�� }d|_|S )�4Handle a connection error for the CustomMock object.� � r` )r3 r � exceptions�ConnectionErrorr r �rH r" r# �rr r r �connection_error s zJTestInfluxDBClient.test_request_retry.<locals>.CustomMock.connection_errorN��__name__� __module__� __qualname__�__doc__r4 r< r r r r � CustomMock s rB rQ rb N)r+ r< r' r rv rE �rH � mock_requestrB r- r r r �test_request_retry� s �z%TestInfluxDBClient.test_request_retryc C sb G dd� dt �}|� j|_tdd�}| �tjj�� |�| j � W d � n1 sT0 Y dS )z4Test that three requests errors will not be handled.c @ s e Zd ZdZdd� Zdd� ZdS )z@TestInfluxDBClient.test_request_retry_raises.<locals>.CustomMockr0 c S s d| _ d S r1 r2 rG r r r r4 s zITestInfluxDBClient.test_request_retry_raises.<locals>.CustomMock.__init__c _ s8 | j d7 _ | j dk r"tjj�nt�� }d|_|S dS )r5 r6 r r N)r3 r r8 � HTTPErrorr r r: r r r r< " s zQTestInfluxDBClient.test_request_retry_raises.<locals>.CustomMock.connection_errorNr= r r r r rB s rB rQ rb N) r+ r< r' r rY r r8 rF rv rE rC r r r �test_request_retry_raises s z,TestInfluxDBClient.test_request_retry_raisesc C sD G dd� dt �}t�dd�}||�j|_td|d�}|�| j� dS )z?Test that a random number of connection errors will be handled.c @ s e Zd ZdZdd� Zdd� ZdS )z@TestInfluxDBClient.test_random_request_retry.<locals>.CustomMockr0 c S s d| _ || _d S r1 �r3 �retries�rH rI r r r r4 : s zITestInfluxDBClient.test_random_request_retry.<locals>.CustomMock.__init__c _ s: | j d7 _ | j | jk r$tjj�nt�� }d|_|S dS )r5 r6 r` N�r3 rI r r8 r9 r r r: r r r r< > s zQTestInfluxDBClient.test_random_request_retry.<locals>.CustomMock.connection_errorNr= r r r r rB 7 s rB r6 � rQ �rJ rI N)r+ r� r� r< r' r rv rE �rH rD rB rI r- r r r �test_random_request_retry4 s z,TestInfluxDBClient.test_random_request_retryc C sr G dd� dt �}t�dd�}||�j|_td|d�}| �tjj �� |� | j � W d � n1 sd0 Y dS )zATest a random number of conn errors plus one will not be handled.c @ s e Zd ZdZdd� Zdd� ZdS )zGTestInfluxDBClient.test_random_request_retry_raises.<locals>.CustomMockr0 c S s d| _ || _d S r1 rH rJ r r r r4 U s zPTestInfluxDBClient.test_random_request_retry_raises.<locals>.CustomMock.__init__c _ s> | j d7 _ | j | jd k r(tjj�nt�� }d|_|S dS )r5 r6 r NrK r: r r r r< Y s zXTestInfluxDBClient.test_random_request_retry_raises.<locals>.CustomMock.connection_errorNr= r r r r rB R s rB r6 rL rQ rM N) r+ r� r� r< r' r rY r r8 r9 rv rE rN r r r � test_random_request_retry_raisesO s z3TestInfluxDBClient.test_random_request_retry_raisesc C s\ d}t �� �<}|jt jd|d� | �| j�� ddd�g� W d � n1 sN0 Y dS )z-Test get users for TestInfluxDBClient object.zQ{"results":[{"series":[{"columns":["user","admin"],"values":[["test",false]]}]}]}r� r� r� F)�userZadminN�ri rj rk r r� r- Zget_list_usersr r r r �test_get_list_usersl s � � �z&TestInfluxDBClient.test_get_list_usersc C sT d}t �� �4}|jt jd|d� | �| j�� g � W d � n1 sF0 Y dS )z6Test get empty userlist for TestInfluxDBClient object.z7{"results":[{"series":[{"columns":["user","admin"]}]}]}r� r� NrR r r r r �test_get_list_users_empty s � �z,TestInfluxDBClient.test_get_list_users_emptyc C sf d}t �� �F}|jt jd|d� | j�d� | �|jjd d d� W d � n1 sX0 Y dS ) z5Test grant admin privs for TestInfluxDBClient object.r� r� r� r� r� r z grant all privileges to "test"N) ri rj rk r r- �grant_admin_privilegesrM rm r� r r r r �test_grant_admin_privileges� s � �z.TestInfluxDBClient.test_grant_admin_privilegesc C sJ t dddd�}t|dd�� | j�d� W d � n1 s<0 Y dS ) z=Test grant invalid admin privs for TestInfluxDBClient object.r9 r2 r3 r4 r r) r N)r r. r- rU rO r r r �#test_grant_admin_privileges_invalid� s z6TestInfluxDBClient.test_grant_admin_privileges_invalidc C sf d}t �� �F}|jt jd|d� | j�d� | �|jjd d d� W d � n1 sX0 Y dS ) z6Test revoke admin privs for TestInfluxDBClient object.r� r� r� r� r� r z!revoke all privileges from "test"N) ri rj rk r r- �revoke_admin_privilegesrM rm r� r r r r � test_revoke_admin_privileges� s � �z/TestInfluxDBClient.test_revoke_admin_privilegesc C sJ t dddd�}t|dd�� | j�d� W d � n1 s<0 Y dS ) z>Test revoke invalid admin privs for TestInfluxDBClient object.r9 r2 r3 r4 r r) r N)r r. r- rX rO r r r �$test_revoke_admin_privileges_invalid� s z7TestInfluxDBClient.test_revoke_admin_privileges_invalidc C sj d}t �� �J}|jt jd|d� | j�ddd� | �|jjd d d � W d � n1 s\0 Y d S ) z/Test grant privs for TestInfluxDBClient object.r� r� r� �readr� r� r� r z grant read on "testdb" to "test"N) ri rj rk r r- �grant_privilegerM rm r� r r r r �test_grant_privilege� s ��z'TestInfluxDBClient.test_grant_privilegec C sN t dddd�}t|dd�� | j�ddd � W d � n1 s@0 Y d S ) z7Test grant invalid privs for TestInfluxDBClient object.r9 r2 r3 r4 r r) r r� r� N)r r. r- r\ rO r r r � test_grant_privilege_invalid� s z/TestInfluxDBClient.test_grant_privilege_invalidc C sj d}t �� �J}|jt jd|d� | j�ddd� | �|jjd d d � W d � n1 s\0 Y d S ) z0Test revoke privs for TestInfluxDBClient object.r� r� r� r[ r� r� r� r z#revoke read on "testdb" from "test"N) ri rj rk r r- �revoke_privilegerM rm r� r r r r �test_revoke_privilege� s ��z(TestInfluxDBClient.test_revoke_privilegec C sN t dddd�}t|dd�� | j�ddd � W d � n1 s@0 Y d S ) z8Test revoke invalid privs for TestInfluxDBClient object.r9 r2 r3 r4 r r) r r� r� N)r r. r- r_ rO r r r � test_revoke_privilege_invalid� s z0TestInfluxDBClient.test_revoke_privilege_invalidc C s� ddddgddgddgd d ggd �gigi}t | jd d t�|���< | �| j�d�ddd�ddd�d d d�g� W d � n1 s�0 Y dS )z5Test get list of privs for TestInfluxDBClient object.r r rJ � privilegeZdb1�READZdb2zALL PRIVILEGESZdb3z NO PRIVILEGES�r� r r r r� )rJ rb N)r. r- r r! r� �get_list_privilegesr r r r �test_get_list_privileges� s$ ���� ��z+TestInfluxDBClient.test_get_list_privilegesc C sH t dddd�}t|dd�� |�d� W d � n1 s:0 Y dS ) z<Test failed get list of privs for TestInfluxDBClient object.r9 r2 r3 r4 r r� r� N)r r. re rO r r r � test_get_list_privileges_fails s z1TestInfluxDBClient.test_get_list_privileges_failsc C s� dddddgddgdd ggd �d ddgd d ggd �dddgd�gd�gi}t | jddt�|���L | �| j�� dddd�dd d�gid d d d�gidg ig� W d � n1 s�0 Y dS )z*Test getting a list of continuous queries.r r Ztestdb01r r� Z testname01Z testquery01Z testname02Z testquery02r Ztestdb02Z testname03Z testquery03Ztestdb03)r r� )Z statement_idr r r )r r� N)r. r- r r! r� �get_list_continuous_queriesr r r r � test_get_list_continuous_queries sF ���������������z3TestInfluxDBClient.test_get_list_continuous_queriesc C s< t | jdd�� | j�� W d � n1 s.0 Y dS )z1Test failing to get a list of continuous queries.r r) N)r. r- rh rG r r r �&test_get_list_continuous_queries_fails8 s z9TestInfluxDBClient.test_get_list_continuous_queries_failsc C s� di gi}t �� �~}|jt jdt�|�d� d}| j�d|d� | �|j j d d d � | j�d|dd � | �|j j d d d � W d � n1 s�0 Y d S ) z Test continuous query creation.r r� r� zOSELECT count("value") INTO "6_months"."events" FROM "events" GROUP BY time(10m)�cq_name�db_namer� r z�create continuous query "cq_name" on "db_name" begin select count("value") into "6_months"."events" from "events" group by time(10m) endzEVERY 10s FOR 2mz�create continuous query "cq_name" on "db_name" resample every 10s for 2m begin select count("value") into "6_months"."events" from "events" group by time(10m) endN) ri rj rk r r r! r- �create_continuous_queryrM rm r� )rH r rp r� r r r � test_create_continuous_query> s( �� ��z/TestInfluxDBClient.test_create_continuous_queryc C sB t | jdd�� | j�ddd� W d � n1 s40 Y dS )z*Test failing to create a continuous query.r r) rk �selectrl N)r. r- rm rG r r r �"test_create_continuous_query_failsY s z5TestInfluxDBClient.test_create_continuous_query_failsc C st di gi}t �� �N}|jt jdt�|�d� | j�dd� | �|j j d d d� W d � n1 sf0 Y d S ) z!Test dropping a continuous query.r r� r� rk rl r� r z,drop continuous query "cq_name" on "db_name"N) ri rj rk r r r! r- �drop_continuous_queryrM rm r� )rH r rp r r r �test_drop_continuous_query_ s ��z-TestInfluxDBClient.test_drop_continuous_queryc C s@ t | jdd�� | j�dd� W d � n1 s20 Y dS )z(Test failing to drop a continuous query.r r) rk rl N)r. r- rq rG r r r � test_drop_continuous_query_failsn s z3TestInfluxDBClient.test_drop_continuous_query_failsc C s< | � t�� tdddd� W d � n1 s.0 Y dS )z5Test invalid port fail for TestInfluxDBClient object.r9 z80/redirr3 r4 N)rY rZ r rG r r r �test_invalid_port_failst s z*TestInfluxDBClient.test_invalid_port_failsc C s� d}t �� ��}|jt jd|d� | jjdddd�}t|�}| �t|�dk� | � |d � � t d d gd gd gdgdggd�gi�� � � | � |d � � t d d gdgdgdgdggd�gi�� � � W d � n1 s�0 Y dS )z4Test chunked response for TestInfluxDBClient object.a {"results":[{"statement_id":0,"series":[{"columns":["key"],"values":[["cpu"],["memory"],["iops"],["network"]],"partial":true}],"partial":true}]} {"results":[{"statement_id":0,"series":[{"columns":["key"],"values":[["qps"],["uptime"],["df"],["mount"]]}]}]} r� r� z show seriesTr )r� � chunk_sizer� r r r r ZmemoryZiopsr� rd r6 ZqpsZuptime�df�mountN) ri rj rk r r- r� r� rV �lenrM �__repr__r )rH r� rp �response�resr r r �test_chunked_responsey s2 � ���� ��z(TestInfluxDBClient.test_chunked_responsec C sf t �� �J}|jt jddddid� t� }|�� | �|jjd d� W d � n1 sX0 Y dS ) z Test auth with default settings.r� r` r� r� r� � AuthorizationzBasic cm9vdDpyb290N� ri rj rk r r r� rM rm r� ro r r r �test_auth_default� s ��z$TestInfluxDBClient.test_auth_defaultc C sl t �� �P}|jt jddddid� tddd�}|�� | �|jjd d � W d � n1 s^0 Y d S ) z,Test auth with custom username and password.r� r` r� r� r� z my-usernamez my-password�r3 r4 r} z&Basic bXktdXNlcm5hbWU6bXktcGFzc3dvcmQ=Nr~ ro r r r �test_auth_username_password� s ���z.TestInfluxDBClient.test_auth_username_passwordc C s� t �� ��}|jt jddddid� tddd�}|�� | �d|jjv � tdd �}|�� | �d|jjv � tdd �}|�� | �d|jjv � W d � n1 s�0 Y dS ) z0Test auth with not defined username or password.r� r` r� r� r� Nr� r} )r3 )r4 ) ri rj rk r r r� rU rm r� ro r r r � test_auth_username_password_none� s � z3TestInfluxDBClient.test_auth_username_password_nonec C sr t �� �V}|jt jddddid� tddddid �}|�� | �|jjd d� W d � n1 sd0 Y dS ) z+Test auth with custom authorization header.r� r` r� r� r� Nr} zmy-token)r3 r4 r� r~ ro r r r �test_auth_token� s ���z"TestInfluxDBClient.test_auth_tokenc C s� t jtjtjdftjtjdftjtjdfg }tdd|d�}| � |j j � d�j |� | � |j j � d�jj� d�|� |j j � d�jjdd �}|�� }| � |j |� dS ) zTest custom socket options.r6 �< � N)r3 r4 �socket_options�http://r� �http://localhost:8086��url)r �default_socket_optionsr� � SOL_SOCKET� SO_KEEPALIVE� IPPROTO_TCP� TCP_KEEPINTVL� TCP_KEEPCNTr � assertEqualsr, �adaptersr r� � poolmanager�connection_pool_kw�connection_from_url� _new_conn)rH Ztest_socket_optionsr- �connection_pool�new_connectionr r r �test_custom_socket_options� s, �������z-TestInfluxDBClient.test_custom_socket_optionsc C sX t ddd�}| �|jj�d�jd� |jj�d�jjdd�}|�� }| �|jt j � dS )z Test default socket options.Nr� r� r� r� ) r r� r, r� r r� r� r� r� r r� )rH r- r� r� r r r �test_none_socket_options� s ���z+TestInfluxDBClient.test_none_socket_optionsN)Yr> r? r@ rA rI rP rW r[ r\ r] rq rx r� r� r� r� r� r� r r� r� r� r� r� r� r� r� r� r� r� �unittest�skipr� r� r� r� r� r r r r r r r r r r r r r! r$ r% r( r* r, r. r/ r) r* rE rG rO rP rS rT rV rW rY rZ r] r^ r` ra rf rg ri rj rn rp rr rs rt r| r r� r� r� r� r� r r r r r/ P s� $ 5; ! ) 0 r/ c s* e Zd ZdZ� fdd�Zddd�Z� ZS ) � FakeClientz0Set up a fake client instance of InfluxDBClient.c s t t| �j|i |�� dS )z0Initialize an instance of the FakeClient object.N)�superr� r4 )rH r" r# �� __class__r r r4 s zFakeClient.__init__Nr c C sR |dkrt d��n<|dkr.| jdkr.t d��n |dkrJ| jdv rJt d��ndS d S ) z&Query data from the FakeClient object.ZFailz Fail onceZhost1z Fail Oncez Fail twicez host1 host2z Fail TwiceZSuccessN)r� Z_host)rH r� �paramsZexpected_response_coderJ r r r r� s zFakeClient.query)Nr N)r> r? r@ rA r4 r� � __classcell__r r r� r r� s �r� )r r )r r r ) rA � __future__r r r r r� r� r� rB r� rz r r) r Zrequests.exceptionsri Z nose.toolsr Zurllib3.connectionr Zinfluxdbr Zinfluxdb.resultsetr r r. ZTestCaser/ r� r r r r �<module> sD <
| ver. 1.1 | |
.
| PHP 8.3.30 | Ð“ÐµÐ½ÐµÑ€Ð°Ñ†Ð¸Ñ Ñтраницы: 0 |
proxy
|
phpinfo
|
ÐаÑтройка