o
    3Ih                     @   s   d dl Z ddlmZ ddlmZ d dlmZ z
d dlZe  W n	 ey)   Y nw dddZ	G dd	 d	e j
Zed
krBe   dS dS )    N   )get_websocket_url)websocket_proxycare)Configurationc                 C   s(   z| | }W |S  t y   |}Y |S w N)KeyError)dictkeydefaultval r   `/home/air/sanwanet/gpt-api/venv/lib/python3.10/site-packages/kubernetes/stream/ws_client_test.pydictval   s   
r   c                   @   s   e Zd Zdd Zdd ZdS )WSClientTestc                 C   s"   dD ]\}}|  t|| qd S )N))zhttp://localhost/apizws://localhost/api)zhttps://localhost/apizwss://localhost/api)zhttps://domain.com/apizwss://domain.com/api)zhttps://api.domain.com/apizwss://api.domain.com/api)zhttp://api.domain.comzws://api.domain.com)zhttps://api.domain.comzwss://api.domain.com)zhttp://api.domain.com/zws://api.domain.com/)zhttps://api.domain.com/zwss://api.domain.com/)assertEqualr   )selfurlws_urlr   r   r   test_websocket_client$   s   
z"WSClientTest.test_websocket_clientc           
      C   s   ddddddddd	d
dgfddddd	d
dgfddddd	d
g dffD ]Z\}}}}}}}t  }|d ur8t|d| |d urGt|dtjj|d |d urQt|d| ti |d d }	| t|	d| | t|	d| | t|	d| | t|	d| q"d S )N)NNNNNNN)http://proxy.example.com:8080/NNproxy.example.com  NN)r   	user:passNr   r   userpassN)r   r    r   r   r   Nr   r   *r   r   r   .example.comzlocalhost,.local,.example.com)	localhostz.localr   proxyproxy_headers)proxy_basic_authno_proxyhttp_proxy_hosthttp_proxy_porthttp_proxy_authhttp_no_proxy)r   setattrurllib3utilmake_headersr   r   r   )
r   r    idpassr#   expect_hostexpect_portexpect_authexpect_noproxyconfigconnect_optr   r   r   test_websocket_proxycare1   s*   
z%WSClientTest.test_websocket_proxycareN)__name__
__module____qualname__r   r3   r   r   r   r   r   "   s    r   __main__r   )unittest	ws_clientr   r   kubernetes.client.configurationr   r)   disable_warningsImportErrorr   TestCaser   r4   mainr   r   r   r   <module>   s   
)