관리-도구
편집 파일: tester.cpython-38.pyc
U /?�f � @ s� d dl mZ d dlmZmZ d dlmZmZ d dl mZm Z d dl m Z ddl mZ d dlmZ d d lmZ d d lmZ d d lmZ d d lmZ d d l mZ e�� dd� Zdd� Z G dd� de�Z dS )� )�md5)�load�dumps)� b64decode� b64encode)r �loads)�path� )� OneServer)�pickling_support)�exc_info)�reraise)� OrderedDict)�open)�environc C s t | d�}t|�}|�� |S )N�rt)r r �close)� fixture_file�f�ret� r �:/opt/hc_python/lib/python3.8/site-packages/pyone/tester.py�read_fixture_file s r c C s( t | d�}|�t|��� � |�� d S )N�wb)r �write� json_dumps�encoder )r �objr r r r �write_fixture_file' s r c @ sb e Zd ZdZddd�Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dd� Z dd� Z dS )�OneServerTesteraQ This class extends the OneServer to facilitate unit testing The idea is to be able to "record" fixtures while testing with a live OpenNebula platform. Those recordings can later be use in "replay" mode to simulate an OpenNebula platform. XMLAPI method calls are recorded as test_base/unit_test/method_signature_instance The signature is generated as the md5 of the parameters if several calls with the same signature are doing during the same unit test, instance is incremented. The order in which calls happen within the same unit_test must be deterministic. Nc K s� |d krt �dd �}|d kr0t �dd��� dk}|d krDt �dd�}t�|�rZt|�| _nt� | _|| _|| _ | � |� t j | |||f|� d S )NZPYONE_TEST_FIXTURE_FILEZPYONE_TEST_FIXTURE_REPLAY�True)�1�yes�trueZPYONE_TEST_FIXTURE_UNIT�init) r �get�lowerr �isfiler � _fixtures�dict�_fixture_replay� _fixture_file�set_fixture_unit_testr �__init__)�self�uri�sessionr Z fixture_unit�timeoutZfixture_replay�optionsr r r r- 6 s zOneServerTester.__init__c C s& || j krt� | j |< | j | | _dS )z� Set the name of the unit test. creates an entry in the fixture tree if it does not exist. :param name: :return: N)r( r) �_fixture_unit_test)r. �namer r r r, P s z%OneServerTester.set_fixture_unit_testc C s( t � }t|dd��� }|�|� |�� S )NT)� sort_keys)r r r �update� hexdigest)r. � methodname�paramsZ signature_md5Zsparmsr r r �_fixture_signature\ s z"OneServerTester._fixture_signaturec C s, | j | | �||� �d�}|s(td��|S )z� returns the next fixture for a given call. :param methodname: XMlRPC method :param params: the paramters passed :return: file name were to store to or read from the fixture data r zHCould not read fixture, if the tests changed you must re-record fixtures)r3 r: �pop� Exception)r. r8 r9 r r r r � _get_fixtureb s z OneServerTester._get_fixturec C sV | � ||�}|| jkr"t� | j|< || j| kr>g | j| |< | j| | �|� dS )z} Will create the fixture for a given call :param methodname: :param params: :return: N)r: r3 r) �append)r. r8 r9 �object� signaturer r r � _set_fixtureq s z OneServerTester._set_fixturec C s� d}| j r4| �||�}d|kr�ttt|d ��� nhzVzt�| ||�}W n> tk r� } z dt t t � d��� � i}|�W 5 d}~X Y nX W 5 | �|||� X |S )z� Intercepts requests. In record mode they are recorded before being returned. In replay mode they are read from fixtures instead Exceptions are also captured and reraised N� exception� ) r* r= r r r rA r � _do_requestr< r r r �decode)r. �methodr9 r rB r r r rD � s �zOneServerTester._do_requestc C s> t |�}t|�D ] \}}t|t�r| �|�||< qt�| |�S )a� Parameters will be used to generate the signature of the method, an md5. So we need signatures to be deterministic. There are two sources of randomness - Python version, in particular differences in dealing with encodings - Unsorted sets. This method will add casting transformations to fix those, only required for testing. :param params: :return: a list of parameters that should generate a deterministic md5 signature. )�list� enumerate� isinstancer) �_to_ordered_dictr � _cast_parms)r. r9 Zlparams�i�paramr r r rK � s zOneServerTester._cast_parmsc C sH t |t�rDtt|�� ��}|�� D ] \}}t |t�r"| �|�||< q"|S )zQ deep orders a dictionary :param param: :return: )rI r) r �sorted�itemsrJ )r. rM �key�valuer r r rJ � s z OneServerTester._to_ordered_dictc C s dS )Ng{�G�z�?r �r. r r r �server_retry_interval� s z%OneServerTester.server_retry_intervalc C s | j st| j| j� dS )zQ Unpdates the fixture data if we are recording. :return: N)r* r r+ r( rR r r r � server_close� s z OneServerTester.server_close)NNNN)�__name__� __module__� __qualname__�__doc__r- r, r: r= rA rD rK rJ rS rT r r r r r , s r N) �hashlibr �jsonr r r �base64r r �pickler �osr � r Ztblibr �sysr �sixr � collectionsr �gzipr r �installr r r r r r r �<module> s