iB <ddlZddlZddlZddlZddlZddlZddlZddlZddlZddl m Z m Z m Z m Z mZmZdZdZGddejZGddejZGd d ejZGd d ejZd edededejefdZdS)N)consterrorsnet server_lib transformstypeszsignkey-prod-000000zsignkey-prod-111111cheZdZd fd Zd dZd dZd dZd dZd dZd d Z d d Z d d Z d d Z xZ S)Test_create_serve_urlreturnNcttjt jjjdtjt jj jddSN) supersetUposenvironpoprEnvKey SERVE_ORIGINvalue SERVE_PATHself __class__s W/root/projects/qq-shell/venv/lib/python3.11/site-packages/inngest/_internal/net_test.pyrzTest_create_serve_url.setUpsV   u|06=== u|.4d;;;;;cJtjdddd}d}||ksJdS)Nhttps://foo.test/api/inngest public_path request_url serve_origin serve_pathrcreate_serve_urlractualexpecteds rtest_only_request_urlz+Test_create_serve_url.test_only_request_urlsC%6     2!!!!!!rcJtjdddd}d}||ksJdS)Nrhttps://bar.testrhttps://bar.test/api/inngestr#r%s rtest_serve_originz'Test_create_serve_url.test_serve_origin"sC%6+     2!!!!!!rcdtjtjjj<t jdddd}d}||ksJdS)Nzhttps://bar-env.testrr*rr+)rrrrrrrr$r%s rtest_serve_origin_env_varz/Test_create_serve_url.test_serve_origin_env_var,sW6L 5<,23%6+     2!!!!!!rcJtjdddd}d}||ksJdS)Nrzbar.testrr+r#r%s r test_serve_origin_missing_schemez6Test_create_serve_url.test_serve_origin_missing_scheme8sC%6#     2!!!!!!rcJtjdddd}d}||ksJdS)Nrzhttps://bar.test:8080rz!https://bar.test:8080/api/inngestr#r%s rtest_serve_origin_portz,Test_create_serve_url.test_serve_origin_portBsC%60     7!!!!!!rcJtjdddd}d}||ksJdS)Nr /custom/pathrhttps://foo.test/custom/pathr#r%s rtest_serve_pathz%Test_create_serve_url.test_serve_pathLsC%6%     2!!!!!!rcdtjtjjj<t jdddd}d}||ksJdS)Nz /env/pathrr4rr5)rrrrrrrr$r%s rtest_serve_path_env_varz-Test_create_serve_url.test_serve_path_env_varVsW4? 5<*01%6%     2!!!!!!rcJtjdddd}d}||ksJdS)Nrr*r4rzhttps://bar.test/custom/pathr#r%s rtest_serve_origin_and_pathz0Test_create_serve_url.test_serve_origin_and_pathbsC%6+%     2!!!!!!rcJtjdddd}d}||ksJdS)z> public_path takes precedence over serve_path z /public/pathrNz /serve/pathrzhttps://foo.test/public/pathr#r%s rtest_public_pathz&Test_create_serve_url.test_public_pathlsE %&6$     2!!!!!!rr N)__name__ __module__ __qualname__rr(r,r.r0r2r6r8r:r< __classcell__rs@rr r s<<<<<< """""""" " " " """"""""""""" " " " """"" " " " " " " " "rr c6eZdZddZddZddZddZddZdS) Test_RequestSignaturer Nc tjddid}tt jdz}t |t |}t|trJtj j j d|d|i}ttj||tjjt dtrJdS)Nmsghiutf-8s=&t=bodyheadersmode signing_keysigning_key_fallback)jsondumpsencoderoundtime_sign _signing_key isinstance Exceptionr HeaderKey SIGNATURErrvalidate_request_sig ServerKindCLOUDrrMunix_mssigrNs r test_successz"Test_RequestSignature.test_success|sz5$-((//88 d*++D,00c9-----  * 02Hs2H2Hw2H2H   $*0(%)        rc fttjdz}tdt|}t |t rJt jjj d|d|i}t tj d|t j j tdt rJdS)NrIs{"msg":"a & b"}rJrKs{"msg":"a \u0026 b"}rL)rUrVrWrXrYrZrr[r\rrr]r^r_)rrarbrNs rtest_escape_sequencesz+Test_RequestSignature.test_escape_sequencess d*++& g>>c9-----  * 02Hs2H2Hw2H2H   $-*0(%)        rctjddid}tt jdz}t |t |}t|trJtj j j d|d|i}tjddid}tj||tjjt d }t|t"jsJdS) zR Validation fails if the body is changed after signature creation rFbarrHrIrJrKzyou've been hackedNrL)rRrSrTrUrVrWrXrYrZrr[r\rrr]r^r_rSigVerificationFailedError)rrMrarbrN validations rtest_body_tamperz&Test_RequestSignature.test_body_tampers z5%.))0099 d*++D,00c9-----  * 02Hs2H2Hw2H2H z5"6788??HH-&,$!%    *f&GHHHHHHHrc tjddid}tt jdz}t |t |}t|trJtj j j d|d|i}ttj||tjjt"t trJdS) zt Validation succeeds if the primary signing key fails but the fallback signing key succeeds rFrGrHrIrJrKrLN)rRrSrTrUrVrW_signing_key_fallbackrYrZrr[r\rrr]r^r_rXr`s r test_rotationz#Test_RequestSignature.test_rotations z5$-((//88 d*++D/99c9-----  * 02Hs2H2Hw2H2H   $*0(%:        rc tjddid}tt jdz}t |d|}t |trJtj j j d|d|i}t tj ||tjjt t"tsJd S) zE Validation fails after trying both the signing keys rFrGrHrIzsomething-elserJrKrLN)rRrSrTrUrVrWrYrZrr[r\rrr]r^r_rXrlr`s r test_fails_for_both_signing_keysz6Test_RequestSignature.test_fails_for_both_signing_keyss z5$-((//88 d*++D*G44c9-----  * 02Hs2H2Hw2H2H   $*0(%:        rr=)r>r?r@rcrerjrmrorrrDrD{s|    (    &IIII0    2      rrDceZdZd fd Zdejejgejfdej fdZ dejejgejfdej fdZ d dZ d dZd d Zd d ZxZS) Test_fetch_with_auth_fallbackr Nc|ttjdd|_dS)NGETzhttp://localhost)rrhttpxRequest_reqrs rrz#Test_fetch_with_auth_fallback.setUps,  M%);<< rhandlercLGfddtj}|S)Nc8eZdZdejdejffd ZdS)HTest_fetch_with_auth_fallback._create_async_transport..Transportrequestr cK|Sr rprr|rxs rhandle_async_requestz]Test_fetch_with_auth_fallback._create_async_transport..Transport.handle_async_requestsww'''rN)r>r?r@rurvResponserrxsr Transportr{sI ( ( ( ( ( ( ( ( ( (rr)ruAsyncBaseTransportrrxrs ` r_create_async_transportz5Test_fetch_with_auth_fallback._create_async_transportsH ( ( ( ( ( ( (0 ( ( (y{{rcLGfddtj}|S)Nc8eZdZdejdejffd ZdS)BTest_fetch_with_auth_fallback._create_transport..Transportr|r c|Sr rpr~s rhandle_requestzQTest_fetch_with_auth_fallback._create_transport..Transport.handle_requestsww'''rN)r>r?r@rurvrrrsrrrsE (em ( ( ( ( ( ( ( ( (rr)ru BaseTransportrs ` r_create_transportz/Test_fetch_with_auth_fallback._create_transportsH ( ( ( ( ( ( (+ ( ( (y{{rcKddtjdtjffd }tjtj||tj| ||j ttd{V}t|trJ|jdksJd ksJdtjtj| ||j tt}t|trJ|jdksJd ksJdS) zF The signing key is valid, so the fallback isn't used rr|r cdz |jd}|tjdd|Sdt jt }||krtjdd|Stjdd|SN authorizationrcontentr|zBearer )rNgetrurrhash_signing_keyrXr| actual_tokenexpected_token req_counts rrxzETest_fetch_with_auth_fallback.test_signing_key_works..handlers NI"?..??L#~c3HHHHE*5lCCEE ~--~c3HHHH>#sGDDD Dr transportrPrQNrrrurvrrfetch_with_auth_fallbackThreadAwareAsyncHTTPClientr initializeClientrrwrXrlrYrZ status_codefetch_with_auth_fallback_syncrrxresrs @rtest_signing_key_worksz4Test_fetch_with_auth_fallback.test_signing_key_works    EU] Eu~ E E E E E E 0  *66w??   jll L4#9#9'#B#B C C C I$!6         c9-----#%%%%A~~~~ / L4#9#9'#B#B C C C I$!6     c9-----#%%%%A~~~~~~rcKddtjdtjffd }tjtj||tj| ||j ttd{V}t|trJ|jdksJd ksJdtjtj| ||j tt}t|trJ|jdksJd ksJdS) zE The signing key is invalid, so the fallback is used rr|r cdz |jd}|tjdd|Sdt jt }||krtjdd|Stjdd|SrrNrrurrrrlrs rrxzNTest_fetch_with_auth_fallback.test_signing_key_fallback_works..handler? NI"?..??L#~c3HHHHN*56KLLNN ~--~c3HHHH>#sGDDD DrrrNrrrs @rtest_signing_key_fallback_worksz=Test_fetch_with_auth_fallback.test_signing_key_fallback_works8rrcKddtjdtjffd }tjtj||tj| ||j ddd {V}t|trJ|j d ksJd ksJdtjtj| ||j dd}t|trJ|j d ksJd ksJd S) z/ Both signing keys are invalid rr|r cdz |jd}|tjdd|Sdt jt }||krtjdd|Stjdd|Srrrs rrxzPTest_fetch_with_auth_fallback.test_signing_key_fallback_invalid..handlernrrrzsignkey-prod-aaaaaazsignkey-prod-bbbbbbrNrrrurvrrrrrrrrrwrYrZrrrs @r!test_signing_key_fallback_invalidz?Test_fetch_with_auth_fallback.test_signing_key_fallback_invalidgs   EU] Eu~ E E E E E E 0  *66w??   jll L4#9#9'#B#B C C C I-!6         c9-----#%%%%A~~~~ / L4#9#9'#B#B C C C I-!6     c9-----#%%%%A~~~~~~rcKddtjdtjffd }tjtj||tj| ||j ddd{V}t|trJ|j dksJd ksJdtjtj| ||j dd}t|trJ|j dksJd ksJdS) z Still send a request when we don't have a signing key. This is necessary to work with the Dev Server rr|r cdz |jd}|tjdd|Stjdd|S)Nrrirrr)rNrrur)r|rrs rrxzBTest_fetch_with_auth_fallback.test_no_signing_key..handlersU NI"?..??L'~c3HHHH>#sGDDD DrrNrrrrrs @rtest_no_signing_keyz1Test_fetch_with_auth_fallback.test_no_signing_keys   EU] Eu~ E E E E E E0  *66w??   jll L4#9#9'#B#B C C C I!%         c9-----#%%%%A~~~~ / L4#9#9'#B#B C C C I!%     c9-----#%%%%A~~~~~~rr=)r>r?r@rtypingCallablerurvrrrrrrrrrrArBs@rrrrrs====== %-%.!@A   !    %-%.!@A  ----^----^----^((((((((rrrc.eZdZddZddZddZddZdS)Test_parse_urlr NctjdtjjdksJtjdtjjdksJdS)Nzfoo.testrOzhttps://foo.testzhttp://foo.test)r parse_urlrr^r_ DEV_SERVERrs rtest_no_schemezTest_parse_url.test_no_schemesl M*:+@+F G G G! " " " " M*:+@+K L L L  ! ! ! ! ! !rcDtjdtjjdksJt tjdtjjt sJt tjdtjjt sJdS)Nzhttp://foo:8080rfooz http://foo)rrrr^r_rYrZrs rtest_no_domainzTest_parse_url.test_no_domains M+*2G2M N N N  ! ! ! ! M%j&;&A B B BI      M,Z-B-H I I I        rc\tjdtjjdksJdS)Nzhttp://foo:8080/barr)rrrr^r_rs r test_pathzTest_parse_url.test_pathsD M%J,A,G   % % % % % % %rcJttjdtjjt sJttjdtjjt sJttjdtjjt sJttjdtjjt sJttjdtjjt sJdS)Ntruer1false0)rYrrrr^r_rZrs rtest_boolean_stringsz#Test_parse_url.test_boolean_stringss" M&z'<'B C C CY      M#J$9$? @ @ @)      M' (=(C D D Di      M#J$9$? @ @ @)      M":#8#> ? ? ?       rr=)r>r?r@rrrrrprrrrsd                   rrrMrPrar cxtj|}t|tr|Stj|}t j|d|tj }| t|d| S)NrH) r canonicalizerYrZremove_signing_key_prefixhmacnewrThashlibsha256updatestr hexdigest)rMrPra canonicalizedmacs rrWrWs+D11M-++6{CCK (7##  C JJs7||""7++,,, ==??r)rrrRrrVrunittest unittest.mockruinngest._internalrrrrrrrXrlTestCaser rDIsolatedAsyncioTestCaserrrbytesrint MaybeErrorrWrprrrs  OOOOOOOOOOOOOOOO$ -f"f"f"f"f"H-f"f"f"Ro o o o o H-o o o dQQQQQH$DQQQh/ / / / / X&/ / / d  C # %:J3:O      r