σ
O'^c           @@  sζ   d  d l  m Z d  d l m Z m Z m Z m Z m Z d  d l m	 Z	 d  d l
 m Z d  d l m Z m Z m Z d  d l m Z m Z d e f d     YZ d	 e f d
     YZ d e f d     YZ d e f d     YZ d S(   i    (   t   absolute_import(   t
   Connectiont   Consumert   Exchanget   Producert   Queue(   t   text_t(   t   Message(   t
   StdChannelt	   Transportt
   Management(   t   Caset   Mockt   test_StdChannelc           B@  s5   e  Z d    Z d   Z d   Z d   Z d   Z RS(   c         C@  sH   t  d  |  _ |  j j   |  _ |  j j j   |  j j j j   d  S(   Ns	   memory://(   R   t   connt   channelt   queuest   cleart
   connectiont   state(   t   self(    (    sA   /tmp/pip-unpacked-wheel-UAnTfW/kombu/tests/transport/test_base.pyt   setUp   s    c         C@  s\   t  d t d   } |  j j GH|  j j |  } |  j | t  |  j | j |  j  d  S(   Nt   foo(   R   R   R   R   R   t   assertIsInstancet   assertIs(   R   t   qt   cons(    (    sA   /tmp/pip-unpacked-wheel-UAnTfW/kombu/tests/transport/test_base.pyt   test_Consumer   s
    c         C@  s9   |  j  j   } |  j | t  |  j | j  |  j   d  S(   N(   R   R   R   R   (   R   t   prod(    (    sA   /tmp/pip-unpacked-wheel-UAnTfW/kombu/tests/transport/test_base.pyt   test_Producer   s    c         C@  s'   |  j  t   t   j   Wd  QXd  S(   N(   t   assertRaisest   NotImplementedErrorR   t   get_bindings(   R   (    (    sA   /tmp/pip-unpacked-wheel-UAnTfW/kombu/tests/transport/test_base.pyt   test_interface_get_bindings   s    c         C@  s#   |  j  t   j t d    d  S(   NR   (   t   assertIsNoneR   t   after_reply_message_receivedR   (   R   (    (    sA   /tmp/pip-unpacked-wheel-UAnTfW/kombu/tests/transport/test_base.pyt+   test_interface_after_reply_message_received#   s    (   t   __name__t
   __module__R   R   R   R!   R$   (    (    (    sA   /tmp/pip-unpacked-wheel-UAnTfW/kombu/tests/transport/test_base.pyR      s
   				t   test_Messagec           B@  sY   e  Z d    Z d   Z d   Z d   Z d   Z d   Z d   Z d   Z	 d   Z
 RS(	   c         C@  s=   t  d  |  _ |  j j   |  _ t |  j d d |  _ d  S(   Ns	   memory://t   delivery_tagi9  (   R   R   R   R   t   message(   R   (    (    sA   /tmp/pip-unpacked-wheel-UAnTfW/kombu/tests/transport/test_base.pyR   +   s    c         C@  sL   t  |  j t d  d d } |  j t   | j   Wd  QX| j   d  S(   Nt   FOOt
   postencodet   ccyzz(   R   R   R   R   t   LookupErrort   _reraise_errort   ack(   R   t   m(    (    sA   /tmp/pip-unpacked-wheel-UAnTfW/kombu/tests/transport/test_base.pyt   test_postencode0   s    c         C@  so   t  d g  |  j _ d |  j j d <t   } |  j _ |  j j   |  j |  j j	 d  |  j
 | j  d  S(   Nt   abct   consumer_tagt   ACK(   t   setR   t   no_ack_consumersR)   t   delivery_infoR   t	   basic_ackR/   t   assertNotEqualt   _statet   assertFalset   called(   R   R/   (    (    sA   /tmp/pip-unpacked-wheel-UAnTfW/kombu/tests/transport/test_base.pyt"   test_ack_respects_no_ack_consumers6   s    c         C@  sX   t  d g  |  j _ i  |  j _ t   } |  j _ |  j j   | j |  j j	  d  S(   NR2   (
   R5   R   R6   R)   R7   R   R8   R/   t   assert_called_withR(   (   R   R/   (    (    sA   /tmp/pip-unpacked-wheel-UAnTfW/kombu/tests/transport/test_base.pyt   test_ack_missing_consumer_tag?   s
    c         C@  sV   t    |  j _ d |  j j d <t   } |  j _ |  j j   | j |  j j	  d  S(   NR2   R3   (
   R5   R   R6   R)   R7   R   R8   R/   R>   R(   (   R   R/   (    (    sA   /tmp/pip-unpacked-wheel-UAnTfW/kombu/tests/transport/test_base.pyt   test_ack_not_no_ackG   s
    c         C@  s7   t    } |  j _ |  j j t    t  | j   d  S(   N(   R   R)   R/   t   ack_log_errort   KeyErrorR>   (   R   R/   (    (    sA   /tmp/pip-unpacked-wheel-UAnTfW/kombu/tests/transport/test_base.pyt    test_ack_log_error_when_no_errorO   s    c         C@  s}   t    } |  j _ t d  | _ t    } |  j j | t  | j   |  j | j j	  |  j
 d | j j d d  d  S(   NR   s   Couldn't acki    (   R   R)   R/   RB   t   side_effectRA   R>   t
   assertTruet   criticalR<   t   assertInt	   call_args(   R   R/   t   logger(    (    sA   /tmp/pip-unpacked-wheel-UAnTfW/kombu/tests/transport/test_base.pyt   test_ack_log_error_when_errorT   s    	
c         C@  sC   t    } |  j _ |  j j t    t d t | j d t  d  S(   Nt   requeue(   R   R)   t   rejectt   reject_log_errorRB   t   TrueR>   (   R   RL   (    (    sA   /tmp/pip-unpacked-wheel-UAnTfW/kombu/tests/transport/test_base.pyt#   test_reject_log_error_when_no_error]   s    c         C@  s   t    } |  j _ t d  | _ t    } |  j j | t  | j d t  |  j | j	 j
  |  j d | j	 j d d  d  S(   NR   RK   s   Couldn't rejecti    (   R   R)   RL   RB   RD   RM   R>   t   FalseRE   RF   R<   RG   RH   (   R   RL   RI   (    (    sA   /tmp/pip-unpacked-wheel-UAnTfW/kombu/tests/transport/test_base.pyt    test_reject_log_error_when_errorb   s    	(   R%   R&   R   R1   R=   R?   R@   RC   RJ   RO   RQ   (    (    (    sA   /tmp/pip-unpacked-wheel-UAnTfW/kombu/tests/transport/test_base.pyR'   )   s   										t   test_interfacec           B@  sY   e  Z d    Z d   Z d   Z d   Z d   Z d   Z d   Z d   Z	 d   Z
 RS(	   c         C@  s*   |  j  t   t d   j   Wd  QXd  S(   N(   R   R   R	   t   Nonet   establish_connection(   R   (    (    sA   /tmp/pip-unpacked-wheel-UAnTfW/kombu/tests/transport/test_base.pyt   test_establish_connectionn   s    c         C@  s-   |  j  t   t d   j d   Wd  QXd  S(   N(   R   R   R	   RS   t   close_connection(   R   (    (    sA   /tmp/pip-unpacked-wheel-UAnTfW/kombu/tests/transport/test_base.pyt   test_close_connectionr   s    c         C@  s-   |  j  t   t d   j d   Wd  QXd  S(   N(   R   R   R	   RS   t   create_channel(   R   (    (    sA   /tmp/pip-unpacked-wheel-UAnTfW/kombu/tests/transport/test_base.pyt   test_create_channelv   s    c         C@  s-   |  j  t   t d   j d   Wd  QXd  S(   N(   R   R   R	   RS   t   close_channel(   R   (    (    sA   /tmp/pip-unpacked-wheel-UAnTfW/kombu/tests/transport/test_base.pyt   test_close_channelz   s    c         C@  s-   |  j  t   t d   j d   Wd  QXd  S(   N(   R   R   R	   RS   t   drain_events(   R   (    (    sA   /tmp/pip-unpacked-wheel-UAnTfW/kombu/tests/transport/test_base.pyt   test_drain_events~   s    c         C@  s    t  d   j t d d   d  S(   Nt   nameR   (   R	   RS   t   heartbeat_checkR   (   R   (    (    sA   /tmp/pip-unpacked-wheel-UAnTfW/kombu/tests/transport/test_base.pyt   test_heartbeat_check   s    c         C@  s   |  j  t d   j    d  S(   N(   RE   R	   RS   t   driver_version(   R   (    (    sA   /tmp/pip-unpacked-wheel-UAnTfW/kombu/tests/transport/test_base.pyt   test_driver_version   s    c         C@  s    t  d   j t d d   d  S(   NR^   t   loop(   R	   RS   t   register_with_event_loopR   (   R   (    (    sA   /tmp/pip-unpacked-wheel-UAnTfW/kombu/tests/transport/test_base.pyt   test_register_with_event_loop   s    c         C@  s   |  j  t d   j  d  S(   N(   RE   R	   RS   t   manager(   R   (    (    sA   /tmp/pip-unpacked-wheel-UAnTfW/kombu/tests/transport/test_base.pyt   test_manager   s    (   R%   R&   RU   RW   RY   R[   R]   R`   Rb   Re   Rg   (    (    (    sA   /tmp/pip-unpacked-wheel-UAnTfW/kombu/tests/transport/test_base.pyRR   l   s   								t   test_Managementc           B@  s   e  Z d    Z RS(   c         C@  s9   t  t d d   } |  j t   | j   Wd  QXd  S(   NR^   t	   transport(   R
   R   R   R   R    (   R   R0   (    (    sA   /tmp/pip-unpacked-wheel-UAnTfW/kombu/tests/transport/test_base.pyt   test_get_bindings   s    (   R%   R&   Rj   (    (    (    sA   /tmp/pip-unpacked-wheel-UAnTfW/kombu/tests/transport/test_base.pyRh      s   N(   t
   __future__R    t   kombuR   R   R   R   R   t
   kombu.fiveR   t   kombu.messageR   t   kombu.transport.baseR   R	   R
   t   kombu.tests.caseR   R   R   R'   RR   Rh   (    (    (    sA   /tmp/pip-unpacked-wheel-UAnTfW/kombu/tests/transport/test_base.pyt   <module>   s   (C#