ó O'—^c@@sdZddlmZddlZddlmZddlmZddlmZddlm Z m Z ddlZej ddkr–d\Z Zn4ydd lm Z Wnek rÉd\Z ZnXd efd „ƒYZdefd„ƒYZde fd„ƒYZdS(sÓTesting module for the kombu.transport.SQS package. NOTE: The SQSQueueMock and SQSConnectionMock classes originally come from http://github.com/pcsforeducation/sqs-mock-python. They have been patched slightly. i(tabsolute_importN(t Connection(t messaging(tfive(tCasetSkipTestisboto does not support Python 3(tSQSsboto not installedt SQSQueueMockcB@sneZd„Zddd„Zddd„ZeZd„Zd„Zdd d d„Z d d„Z d „Z RS( cC@s||_g|_d|_dS(Ni(tnametmessagest_get_message_calls(tselfR((s@/tmp/pip-unpacked-wheel-UAnTfW/kombu/tests/transport/test_SQS.pyt__init__!s  i cC@s|j g}|j(| S(N(R (R t page_sizetvtimeouttempty((s@/tmp/pip-unpacked-wheel-UAnTfW/kombu/tests/transport/test_SQS.pytclear&scC@s t|jƒS(N(tlenR (R R R((s@/tmp/pip-unpacked-wheel-UAnTfW/kombu/tests/transport/test_SQS.pytcount*scC@sg|j(tS(N(R tTrue(R ((s@/tmp/pip-unpacked-wheel-UAnTfW/kombu/tests/transport/test_SQS.pytdelete.s cC@s-y|jj|ƒWntk r(tSXtS(N(R tremovet ValueErrortFalseR(R tmessage((s@/tmp/pip-unpacked-wheel-UAnTfW/kombu/tests/transport/test_SQS.pytdelete_message2s  icO@s|jd7_|j| S(Ni(R R (R t num_messagestvisibility_timeoutt attributestargstkwargs((s@/tmp/pip-unpacked-wheel-UAnTfW/kombu/tests/transport/test_SQS.pyt get_messages9scC@s|jjdƒS(Ni(R tpop(R R((s@/tmp/pip-unpacked-wheel-UAnTfW/kombu/tests/transport/test_SQS.pytread>scC@s|jj|ƒtS(N(R tappendR(R R((s@/tmp/pip-unpacked-wheel-UAnTfW/kombu/tests/transport/test_SQS.pytwriteAsN( t__name__t __module__R RRt count_slowRRtNoneRR!R#(((s@/tmp/pip-unpacked-wheel-UAnTfW/kombu/tests/transport/test_SQS.pyRs     tSQSConnectionMockcB@sDeZd„Zd„Zdd„Zed„Zd„Zd„ZRS(cC@s i|_dS(N(tqueues(R ((s@/tmp/pip-unpacked-wheel-UAnTfW/kombu/tests/transport/test_SQS.pyR HscC@s|jj|ƒS(N(R)tget(R tqueue((s@/tmp/pip-unpacked-wheel-UAnTfW/kombu/tests/transport/test_SQS.pyt get_queueKstcC@s |jjƒS(N(R)tvalues(R tprefix((s@/tmp/pip-unpacked-wheel-UAnTfW/kombu/tests/transport/test_SQS.pytget_all_queuesNscC@sI|j|ƒ}|rE|jƒr%tS|jƒ|jj|dƒndS(N(R,RRRR)R R'(R R+tforce_deletiontq((s@/tmp/pip-unpacked-wheel-UAnTfW/kombu/tests/transport/test_SQS.pyt delete_queueQs   cC@s |j|ƒS(N(R(R R+R((s@/tmp/pip-unpacked-wheel-UAnTfW/kombu/tests/transport/test_SQS.pyRYscO@st|ƒ}|j|<|S(N(RR)(R RRRR2((s@/tmp/pip-unpacked-wheel-UAnTfW/kombu/tests/transport/test_SQS.pyt create_queue\s( R$R%R R,R0RR3RR4(((s@/tmp/pip-unpacked-wheel-UAnTfW/kombu/tests/transport/test_SQS.pyR(Fs      t test_ChannelcB@s˜eZd„Zd„Zd„Zd„Zd„Zd„Zd„Zd„Z d„Z d „Z d „Z d „Z d „Zd „Zd„Zd„ZRS(cC@s ||_dS(N(tcallback_message(R R((s@/tmp/pip-unpacked-wheel-UAnTfW/kombu/tests/transport/test_SQS.pythandleMessageCallbackcsc@s6td krttƒ‚ntjjjƒdˆ_tƒˆ_ ‡fd†}|ƒtj_ t j dddƒˆ_ t jˆjˆj ˆjƒˆ_tˆjƒˆ_tdtjƒˆ_ˆjjƒˆ_ˆjˆjƒjƒtjˆjˆj dˆjƒˆ_ˆjjˆjdtd ˆjd dƒd S( sMock the back-end SQS classestunittestc@sˆjS(N(t sqs_conn_mock((R (s@/tmp/pip-unpacked-wheel-UAnTfW/kombu/tests/transport/test_SQS.pytmock_sqsvsttest_SQSttypetdirectt transportt routing_keytno_acktcallbackt consumer_tagN(RR'Rt skip_reasontChannelt _queue_cacheRt queue_nameR(R9tsqstkombutExchangetexchangetQueueR+Rtsqs_queue_mockRt Transportt connectiontchanneltdeclareRtProducertproducert basic_consumeRR7(R R:((R s@/tmp/pip-unpacked-wheel-UAnTfW/kombu/tests/transport/test_SQS.pytsetUpfs,       cC@s|j|j|jjƒdS(s;kombu.SQS.Channel instantiates correctly with mocked queuesN(tassertInRFRORE(R ((s@/tmp/pip-unpacked-wheel-UAnTfW/kombu/tests/transport/test_SQS.pyt test_init”scC@s@d}|jj|ƒ|j||jjƒ|jj|ƒdS(Ntnew_unittest_queue(ROt _new_queueRUR9R)t_delete(R RF((s@/tmp/pip-unpacked-wheel-UAnTfW/kombu/tests/transport/test_SQS.pyttest_new_queue˜scC@s@d}|jj|ƒ|jj|ƒ|j||jjƒdS(NRW(RORXRYt assertNotInRE(R RF((s@/tmp/pip-unpacked-wheel-UAnTfW/kombu/tests/transport/test_SQS.pyt test_deleteŸscC@s¬d}|jj|ƒ|jj|jƒ}|jt|ƒdƒx3tdƒD]%}dj|ƒ}|jj|ƒqNW|jj|jddƒ}|jt|ƒdƒdS(Nsmy test messageiis message: {0}R( RRtpublishROt _get_from_sqsRFt assertEqualRtrangetformat(R Rtresultsti((s@/tmp/pip-unpacked-wheel-UAnTfW/kombu/tests/transport/test_SQS.pyttest_get_from_sqs¥scC@s0|jtjƒ|jj|jƒWdQXdS(N(t assertRaisesRtEmptyROt_getRF(R ((s@/tmp/pip-unpacked-wheel-UAnTfW/kombu/tests/transport/test_SQS.pyttest_get_with_empty_list´scC@s0|jtjƒ|jj|jƒWdQXdS(N(ReRRfROt _get_bulkRF(R ((s@/tmp/pip-unpacked-wheel-UAnTfW/kombu/tests/transport/test_SQS.pyttest_get_bulk_raises_empty¸scC@s¨d}x.t|ƒD] }d|}|jj|ƒqW|jj|jd|ƒ}|jj||jƒ}|jt|ƒ|ƒx!|D]}|j d|kƒq‡WdS(Nis message: %sRt properties( R`RRR]ROR^RFt_messages_to_pythonR_Rt assertTrue(R t message_countRcRR tpayloadstp((s@/tmp/pip-unpacked-wheel-UAnTfW/kombu/tests/transport/test_SQS.pyttest_messages_to_python¼s    cC@sEd}|jj|ƒ|j|jƒjƒj}|j||ƒdS(Nsmy test message(RRR]R+ROR*tpayloadR_(R RRb((s@/tmp/pip-unpacked-wheel-UAnTfW/kombu/tests/transport/test_SQS.pyttest_put_and_getÔscC@sux.tdƒD] }d|}|jj|ƒq Wx=tdƒD]/}|jd||j|jƒjƒjƒq>WdS(Nis message: %s(R`RRR]R_R+ROR*Rr(R RcR((s@/tmp/pip-unpacked-wheel-UAnTfW/kombu/tests/transport/test_SQS.pyttest_puts_and_getsÚs   cC@sEd}|jj|ƒ|jj|jƒ}|jdt|ƒƒdS(Nsmy test messagei(RRR]RORiRFR_R(R RRb((s@/tmp/pip-unpacked-wheel-UAnTfW/kombu/tests/transport/test_SQS.pyttest_put_and_get_bulkãscC@s d}d|jj_x.t|ƒD] }d|}|jj|ƒq"W|jj|jƒ}|jdt |ƒƒ|jj|jƒ}|jdt |ƒƒdS(Niis message: %si( ROtqostprefetch_countR`RRR]RiRFR_R(R RnRcRRb((s@/tmp/pip-unpacked-wheel-UAnTfW/kombu/tests/transport/test_SQS.pyttest_puts_and_get_bulkês cC@sBd„}||jj_|jtjƒ|jjƒWdQXdS(NcS@stS(N(R(((s@/tmp/pip-unpacked-wheel-UAnTfW/kombu/tests/transport/test_SQS.pytmock_can_consumes(RORvt can_consumeReRRft drain_events(R Ry((s@/tmp/pip-unpacked-wheel-UAnTfW/kombu/tests/transport/test_SQS.pyt!test_drain_events_with_empty_listÿs cC@sŽd}d}d|jj_x(t|ƒD]}|jjd|ƒq(Wx!t|ƒD]}|jjƒqSW|j||jj|j j ƒdS(Niiis message: %s( RORvRwR`RRR]R{R_RERFR (R Rntexpected_get_message_countRc((s@/tmp/pip-unpacked-wheel-UAnTfW/kombu/tests/transport/test_SQS.pyt!test_drain_events_with_prefetch_5scC@sŽd}d}d|jj_x(t|ƒD]}|jjd|ƒq(Wx!t|ƒD]}|jjƒqSW|j||jj |j j ƒdS(Niis message: %s( R'RORvRwR`RRR]R{R_RERFR (R RnR}Rc((s@/tmp/pip-unpacked-wheel-UAnTfW/kombu/tests/transport/test_SQS.pyt$test_drain_events_with_prefetch_nones(R$R%R7RTRVRZR\RdRhRjRqRsRtRuRxR|R~R(((s@/tmp/pip-unpacked-wheel-UAnTfW/kombu/tests/transport/test_SQS.pyR5as   .            (Nsboto does not support Python 3(Nsboto not installed(t__doc__t __future__RtsysRHRRRtkombu.tests.caseRRt version_infoR'RRCtkombu.transportt ImportErrortobjectRR(R5(((s@/tmp/pip-unpacked-wheel-UAnTfW/kombu/tests/transport/test_SQS.pyts    '