o
    QD©iR  ã                   @   sP   d dl Z d dlmZmZmZmZ d dlmZ edƒdd„ ƒZedƒdd„ ƒZ	dS )	é    N)ÚANYÚ	MagicMockÚcallÚpatch)Ú"SetupPrometheusEndpointOnPortRangez$django_prometheus.exports.HTTPServerc                 C   sP   t jtƒ g| _ddg}t|ƒ}||v sJ ‚tdtƒtdtƒg}| j|ks&J ‚dS )z-Test port range setup with an available port.é@  éA  ©Ú r   ©r
   r   N)ÚsocketÚerrorr   Úside_effectr   r   r   Ú
mock_calls©Úhttpserver_mockÚ
port_rangeÚport_chosenÚexpected_calls© r   ú]/var/www/Datamplify/venv/lib/python3.10/site-packages/django_prometheus/tests/test_exports.pyÚtest_port_range_available   s   r   c                 C   sP   t jt jg| _ddg}t|ƒ}tdtƒtdtƒg}| j|ks J ‚|du s&J ‚dS )z.Test port range setup with no available ports.r   r   r	   r   N)r   r   r   r   r   r   r   r   r   r   r   Útest_port_range_unavailable   s   r   )
r   Úunittest.mockr   r   r   r   Údjango_prometheus.exportsr   r   r   r   r   r   r   Ú<module>   s   
