Mh tdZddlZddlZddlZddlZddlZddlZddlZddlm Z m Z m Z ddl m Z mZddlmZddlZddlmZmZdZd gZd eDZejejd d ejd Zejdedede eefddfdZ ejdedede eefddfdZ!ej"ddej#de e eefddffdZ$dZ% ddedede&de e'fdZ(dS)z@Integration tests of authorization running under jupyter-server.N) GeneratorOptionalTuple) HTTPErrorURLError)urlopen) KNOWN_SERVERSextra_node_rootsz 127.0.0.1z /lsp/statuscg|]}d|S)z/lsp/ws/).0lss [/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/jupyter_lsp/tests/test_auth.py rs 5 5 5___ 5 5 5JLSP_TEST_SUBPROCESS_PREFIXz["z", "-m"]routea_server_url_and_tokenreturnc:|\}}t||t||d|d}| Jd|d} tj|dn##tjj$r }|}Yd}~nd}~wwxYw|rJd|d|dS)zBVerify a REST route only provides access to an authenticated user.z?token=Nzno response received from utf-8zthe response for z was not JSON: )verify_responsejsonloadsdecodedecoderJSONDecodeError)rrbase_urltokenraw_body decode_errorerrs rtest_auth_restr%s-OHeHe$$$xE)A)A%)A)A3GGH   !Ee!E!E   L 8??7++,,,, < ' UUUUU|UUUUUUUs'A&&B:BBc2t|d|dS)zFVerify a WebSocket does not provide access to an unauthenticated user.rN)r)rrs rtest_auth_websocketr'.s *1-u55555rmodule)scopetmp_path_factoryc#jKttj}t}|d}|d}|dz }|jddddd}||d}it}|d |i|d } |tj | d gtd d|d} td| ttj} | t|t|t|jt#j| t|| t"j} dt(d|} d}d}|sX|rV t+| d}n>#t,$r1td|dd|dz}t/jdYnwxYw|s|V|st3d| |fV tdt+| d|gnp#t,$rctd| | d | | YnwxYw| | j Jd"d!S)#z+Start a temporary, isolated jupyter server.root_dirhomezetc/jupyter/jupyter_config.jsonT)parentsF) jupyter_lsp jupyterlab nbclassic)jpserver_extensionsr!r!) ServerAppIdentityProviderLanguageServerManagerr)encodingjupyter_serverz--port=z --no-browserz server args)HOME USERPROFILEJUPYTER_CONFIG_DIR)cwdenvstdinzhttp://:z /favicon.ico[z / 20] ...)flushr zthe server did not startzshutting down with API...z/api/shutdown?token=)datazshutting down the hard way...sy Nz%jupyter-server probably still running)struuiduuid4get_unused_portmktempparentmkdirr write_textrdumpsSUBPROCESS_PREFIXprintdictosenvironupdate subprocessPopenPIPE LOCALHOSTrrtimesleep RuntimeError terminate communicatewaitkill returncode)r*r!portr,r- server_conf extensionsapplsm config_dataargsr<procurlretriesoks rrr4s'   E   D&&z22H  " "6 * *D::KT***!%UOOJ",u = =C    C$e,!$K 4:k22WEEE S  S 0 S2BD2B2B SN SD - rz  CJJ YYII{122  Dc(mmJO T T TD &I & & & &CG BW C---..BB    )g))) 6 6 6 6 qLG JqMMMMM W 75666 u* )***333E33"=====  -...         IIKKK ? & &(O & & & & &s% F8GG8%HA*J  J ctjtjtj}|tdf|d|d}||S)zGet an unused port by trying to listen to any random port. Probably could introduce race conditions if inside a tight loop. rr )socketAF_INET SOCK_STREAMbindrUlisten getsocknameclose)sockr^s rrFrFnsg =); < rslFF  ----------,,,,,,,,"""""" 55555555  o 5 5} 5 5 5 DJJNN02S2S2S2STT +..V#VuS#XV4VVV/.V&),,6s6E#s(O6PT666-,6 h6P,6PuS#Xd*+6P6P6P 6Pr   36,/ e_r