σ O'—^c@@s©ddlmZmZddlZddlmZddlmZmZm Z m Z ddd„Z d efd „ƒYZ d efd „ƒYZ d efd„ƒYZdS(i(tabsolute_importtunicode_literalsN(t ConsumerMixini(tCasetMockt ContextMocktpatchu text/plainuutf-8cC@s.tddƒ}||_||_||_|S(NtnameuMessage(Rtbodyt content_typetcontent_encoding(RR R tm((s9/tmp/pip-unpacked-wheel-UAnTfW/kombu/tests/test_mixins.pytMessage s    tConscB@seZd„ZRS(cC@s―tddƒ|_||jj_|jj|_|jj|_|jj|_|jj|_|jj|_|jj|_|jj |_ t ddƒ|_ |j |j _dS(NRu ConsumerMixinu extra_context( Rtcallst get_consumerst return_valueton_connection_revivedton_consume_readyton_consume_endt on_iterationton_decode_errorton_connection_errorRt extra_context(tselft consumers((s9/tmp/pip-unpacked-wheel-UAnTfW/kombu/tests/test_mixins.pyt__init__s(t__name__t __module__R(((s9/tmp/pip-unpacked-wheel-UAnTfW/kombu/tests/test_mixins.pyR sttest_ConsumerMixincB@s5eZd„Zd„Zd„Zd„Zd„ZRS(cC@srtddƒ}tddƒ}t||gƒ}tddƒ}|_tddƒ}|_||_|||fS(NRu consumerAu consumerBu connectionuest_connection(RR t connectionRtestablish_connectionR(RtAconstBconstct_conntest((s9/tmp/pip-unpacked-wheel-UAnTfW/kombu/tests/test_mixins.pyt_context$s cC@sε|jƒ\}}}t|_|jdtƒ}t|ƒ|jjƒ|jjƒ|jjjƒ|j |j j ƒ|j jƒ|j jjddƒt|ƒt|ƒt|ƒt|_|jtƒt|ƒWdQXdS(Ntno_ackttimeouti(R%tFalset should_stoptconsumetTruetnextt __enter__tassert_called_withRt assertTrueRtcalledRRt drain_eventst assertRaisest StopIteration(RR"R R!tit((s9/tmp/pip-unpacked-wheel-UAnTfW/kombu/tests/test_mixins.pyt test_consume-s          c@sΉ|jƒ\‰}}tˆ_ˆjdtƒ}tjˆjj_ |j tjƒt |ƒWdQX‡fd†}|ˆjj_ ˆjdtƒ}|j t ƒt |ƒWdQXdS(NR&c@stˆ_tjƒ‚dS(N(R+R)tsocketterror(targstkwargs(R"(s9/tmp/pip-unpacked-wheel-UAnTfW/kombu/tests/test_mixins.pytse2Gs ( R%R(R)R*R+R6R7RR1t side_effectR2R,R3(RR R!R4R:((R"s9/tmp/pip-unpacked-wheel-UAnTfW/kombu/tests/test_mixins.pyt&test_consume_drain_raises_socket_error?s c@s{|jƒ\‰}}tˆ_ˆjdtddƒ}‡fd†}|ˆjj_|jt j ƒt |ƒWdQXdS(NR&R'ic@stˆ_tjƒ‚dS(N(R+R)R6R'(R8R9(R"(s9/tmp/pip-unpacked-wheel-UAnTfW/kombu/tests/test_mixins.pytseTs ( R%R(R)R*R+RR1R;R2R6R7R,(RR R!R4R=((R"s9/tmp/pip-unpacked-wheel-UAnTfW/kombu/tests/test_mixins.pyt(test_consume_drain_raises_socket_timeoutOs c C@sσ|jƒ\}}}|jƒΉ\}}}|j||jƒ|j||jƒ|jjƒ|j|jj ƒ|jj dd}|ƒ}|j|j |j ƒ|j|j |jƒ|j jƒ|j jƒWdQX|jj||ƒdS(Ni(R%tConsumertassertIsRtdefault_channelRR.R/RR0t call_argsRtchannelR-R( RR"R R!tconnRCtconsumertclstsubcons((s9/tmp/pip-unpacked-wheel-UAnTfW/kombu/tests/test_mixins.pyttest_Consumer_context[s   (RRR%R5R<R>RH(((s9/tmp/pip-unpacked-wheel-UAnTfW/kombu/tests/test_mixins.pyR"s    ttest_ConsumerMixin_interfacecB@s‘eZd„Zd„Zd„Zd„Zd„Zd„Zd„Zd„Z d„Z d „Z d „Z d „Z d „Zd „Zd„Zd„Zd„ZRS(cC@stƒ|_dS(N(RR"(R((s9/tmp/pip-unpacked-wheel-UAnTfW/kombu/tests/test_mixins.pytsetUposcC@s3|jtƒ|jjtƒtƒƒWdQXdS(N(R2tNotImplementedErrorR"RR(R((s9/tmp/pip-unpacked-wheel-UAnTfW/kombu/tests/test_mixins.pyttest_get_consumersrscC@s|j|jjƒƒdS(N(t assertIsNoneR"R(R((s9/tmp/pip-unpacked-wheel-UAnTfW/kombu/tests/test_mixins.pyttest_on_connection_revivedvscC@s)|j|jjtƒtƒgƒƒdS(N(RMR"RR(R((s9/tmp/pip-unpacked-wheel-UAnTfW/kombu/tests/test_mixins.pyttest_on_consume_readyyscC@s&|j|jjtƒtƒƒƒdS(N(RMR"RR(R((s9/tmp/pip-unpacked-wheel-UAnTfW/kombu/tests/test_mixins.pyttest_on_consume_end~scC@s|j|jjƒƒdS(N(RMR"R(R((s9/tmp/pip-unpacked-wheel-UAnTfW/kombu/tests/test_mixins.pyttest_on_iterationscC@s[tdƒ}tdƒ=}|jj|tdƒƒ|j|jƒ|jjƒWdQXdS(Nufooukombu.mixins.error( R RR"RtKeyErrorR/R0tackR.(RtmessageR7((s9/tmp/pip-unpacked-wheel-UAnTfW/kombu/tests/test_mixins.pyttest_on_decode_error„s  cC@sBtdƒ0}|jjtdƒdƒ|j|jƒWdQXdS(Nukombu.mixins.warnufooi(RR"RRRR/R0(Rtwarn((s9/tmp/pip-unpacked-wheel-UAnTfW/kombu/tests/test_mixins.pyttest_on_connection_error‹scC@s&|jjtƒtƒƒWdQXdS(N(R"RR(R((s9/tmp/pip-unpacked-wheel-UAnTfW/kombu/tests/test_mixins.pyttest_extra_contextscC@s|j|jjƒdS(N(R/R"t restart_limit(R((s9/tmp/pip-unpacked-wheel-UAnTfW/kombu/tests/test_mixins.pyttest_restart_limit”scC@sitddƒ}||j_tf|_|j|jj|jƒtf|_|j|jj|jƒdS(NRu connection(RR"RRRtconnection_errorstassertTupleEqualt ValueErrortchannel_errors(RRD((s9/tmp/pip-unpacked-wheel-UAnTfW/kombu/tests/test_mixins.pyttest_connection_errors—s    cC@s|tddƒ}tddƒ}tddƒ|_tddƒ|_|jj||ƒWdQX|jjƒ|jjƒdS(NRuAuBu A.__enter__u B.__enter__(RRR-R"t _consume_fromR.(Rtatb((s9/tmp/pip-unpacked-wheel-UAnTfW/kombu/tests/test_mixins.pyttest__consume_fromŸs cC@sutddƒ}||j_||j_d|j_|jjƒ}|j|ƒWdQX|jj |jj dƒdS(NRu connectioni( RtcloneRR"Rtconnect_max_retriesRR/tensure_connectionR.R(RRD((s9/tmp/pip-unpacked-wheel-UAnTfW/kombu/tests/test_mixins.pyttest_establish_connection©s    cC@sMtddƒ}tf|_d|_||j_d„}|jj|ƒdS(NRu connectioncS@stdƒ‚dS(Nufoo(RR(((s9/tmp/pip-unpacked-wheel-UAnTfW/kombu/tests/test_mixins.pytraisesΌs((RRRR[R^R"Rtmaybe_conn_error(RRDRh((s9/tmp/pip-unpacked-wheel-UAnTfW/kombu/tests/test_mixins.pyttest_maybe_conn_error΅s     c@s~tddƒ}|ˆj_tf|_d|_tddƒ}ˆj_‡fd†}tˆj_ ||_ ˆjj ƒdS(NRu connectionu c.consumec@stˆj_dgS(Ni(R+R"R)(R8R9(R(s9/tmp/pip-unpacked-wheel-UAnTfW/kombu/tests/test_mixins.pyR=Ηs (( RR"RRRR[R^RR*R(R)R;trun(RRDR*R=((Rs9/tmp/pip-unpacked-wheel-UAnTfW/kombu/tests/test_mixins.pyttest_runΐs     c@s―tddƒ}|ˆj_tf|_d|_tddƒ}ˆj_tdƒT}dg‰‡‡fd†}t ˆj_ ||_ ˆjj ƒˆj |jƒWdQXdS(NRu connectionu c.consumeukombu.mixins.sleepic@s3ˆddkrtˆj_nˆdcd7<ˆS(Nii(R+R"R)(R8R9(tcounterR(s9/tmp/pip-unpacked-wheel-UAnTfW/kombu/tests/test_mixins.pyR=Χs((RR"RRRR[R^RR*RR(R)R;RkR/R0(RRDR*tsleepR=((RmRs9/tmp/pip-unpacked-wheel-UAnTfW/kombu/tests/test_mixins.pyttest_run_restart_rate_limitedΞs       c@s£tddƒ}|ˆj_tf|_d|_tddƒ}ˆj_tdƒH}‡fd†}t ˆj_ ||_ ˆjj ƒˆj |jƒWdQXdS(NRu connectionu c.consumeukombu.mixins.warnc@stˆj_tdƒ‚dS(Nufoo(R+R"R)RR(R8R9(R(s9/tmp/pip-unpacked-wheel-UAnTfW/kombu/tests/test_mixins.pyt se_raisesιs ((RR"RRRR[R^RR*RR(R)R;RkR/R0(RRDR*RVRp((Rs9/tmp/pip-unpacked-wheel-UAnTfW/kombu/tests/test_mixins.pyttest_run_raisesαs      (RRRJRLRNRORPRQRURWRXRZR_RcRgRjRlRoRq(((s9/tmp/pip-unpacked-wheel-UAnTfW/kombu/tests/test_mixins.pyRIms"             (t __future__RRR6t kombu.mixinsRtcaseRRRRR R RRI(((s9/tmp/pip-unpacked-wheel-UAnTfW/kombu/tests/test_mixins.pyts "K