
    ?g                         d dl Z ddlmZ ddlmZ d dlmZ 	 d dlZ ej                          d
dZ	 G d de j                        Zed	k(  r e j                          yy# e$ r Y 8w xY w)    N   )get_websocket_url)websocket_proxycare)Configurationc                 6    	 | |   }|S # t         $ r |}Y |S w xY wN)KeyError)dictkeydefaultvals       U/var/www/openai/venv/lib/python3.12/site-packages/kubernetes/stream/ws_client_test.pydictvalr      s2    3i J  Js   	 c                       e Zd Zd Zd Zy)WSClientTestc                 N    dD ]   \  }}| j                  t        |      |       " y )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_urls      r   test_websocket_clientz"WSClientTest.test_websocket_client$   s*    	KC .s3V<	    c                    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 ]  \  }}}}}}}t               }|t        |d|       |+t        |dt        j                  j	                  |             |t        |d|       t        i |d d       }	| j                  t        |	d      |       | j                  t        |	d      |       | j                  t        |	d      |       | j                  t        |	d      |        y )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_opts
             r   test_websocket_proxycarez%WSClientTest.test_websocket_proxycare1   sl    J J T T2Kbuw{  ~M  PS  OT  U2Kbuw{  ~M  P^  O_  `2KA`cvx|  N  Pu  v_ZE68[+{N #_F /!1J1J\b1J1cd#
H5 .r64FKgk2CDkRgk2CDkRgk2CDkRgk/BNS/_r   N)__name__
__module____qualname__r   r8    r   r   r   r   "   s    =Tr   r   __main__r   )unittest	ws_clientr   r   kubernetes.client.configurationr   r.   disable_warningsImportErrorr   TestCaser   r9   mainr<   r   r   <module>rE      sr     ( * 9	G'T8$$ 'TR zHMMO g  		s   A A&%A&