σ
P'^c           @@  st   d  d l  m Z d  d l m Z m Z m Z m Z m Z d  d l m	 Z	 m
 Z
 m Z m Z d e	 f d     YZ d S(   i    (   t   absolute_import(   t	   AMQPAdmint   AMQShellt   dump_messaget   amqpt   main(   t   AppCaset   Mockt
   WhateverIOt   patcht   test_AMQShellc           B@  s
  e  Z d    Z d   Z d   Z d   Z d   Z d   Z d   Z d   Z	 d   Z
 d	   Z d
   Z d   Z d   Z d   Z d   Z d   Z d   Z d   Z d   Z d   Z d   Z d   Z d   Z d   Z e d  d    Z e d  d    Z RS(   c         C@  s@   t    |  _ |  j   |  _ t d |  j j d |  j  |  _ d  S(   Nt   connectt   out(   R   t   fht
   create_admt   admR   R   t   shell(   t   self(    (    s<   /tmp/pip-unpacked-wheel-gV1wwp/celery/tests/bin/test_amqp.pyt   setup   s    c         O@  s   t  d |  j d |  j | |  S(   Nt   appR   (   R   R   R   (   R   t   argst   kwargs(    (    s<   /tmp/pip-unpacked-wheel-gV1wwp/celery/tests/bin/test_amqp.pyR      s    c         C@  s-   |  j  j d  |  j d |  j j    d  S(   Ns   queue.declare foot   ok(   R   t   onecmdt   assertInR   t   getvalue(   R   (    (    s<   /tmp/pip-unpacked-wheel-gV1wwp/celery/tests/bin/test_amqp.pyt   test_queue_declare   s    c         C@  s-   |  j  j d  |  j d |  j j    d  S(   Ns   foo foos   unknown syntax(   R   R   R   R   R   (   R   (    (    s<   /tmp/pip-unpacked-wheel-gV1wwp/celery/tests/bin/test_amqp.pyt   test_missing_command   s    c         C@  s   t  |  j j     d  S(   N(   t	   ExceptionR   R   (   R   (    (    s<   /tmp/pip-unpacked-wheel-gV1wwp/celery/tests/bin/test_amqp.pyt   RV    s    c         C@  sF   |  j  j d } |  j | j d   d  |  j | j d  d  d  S(   Ns   exchange.declares   ok.t   NO(   R   R   t   assertEqualt   format_responset   None(   R   t   spec(    (    s<   /tmp/pip-unpacked-wheel-gV1wwp/celery/tests/bin/test_amqp.pyt   test_spec_format_response#   s    c         C@  s-   |  j  j d  |  j d |  j j    d  S(   Ns
   ns.cmd args   unknown syntax(   R   R   R   R   R   (   R   (    (    s<   /tmp/pip-unpacked-wheel-gV1wwp/celery/tests/bin/test_amqp.pyt   test_missing_namespace(   s    c         C@  s-   |  j  j d  |  j d |  j j    d  S(   Nt   helps   Example:(   R   R   R   R   R   (   R   (    (    s<   /tmp/pip-unpacked-wheel-gV1wwp/celery/tests/bin/test_amqp.pyt	   test_help,   s    c         C@  s-   |  j  j d  |  j d |  j j    d  S(   Ns   help queue.declares
   passive:no(   R   R   R   R   R   (   R   (    (    s<   /tmp/pip-unpacked-wheel-gV1wwp/celery/tests/bin/test_amqp.pyt   test_help_command0   s    c         C@  s-   |  j  j d  |  j d |  j j    d  S(   Ns   help foo.bazs   unknown syntax(   R   R   R   R   R   (   R   (    (    s<   /tmp/pip-unpacked-wheel-gV1wwp/celery/tests/bin/test_amqp.pyt   test_help_unknown_command4   s    c         C@  s   t    |  j _ t   |  j j _ t    |  j _ |  j |  j j  |  j j d  |  j	 |  j j j
  |  j	 |  j j  d  S(   Nt   hello(   R   R   t   dispatcht   MemoryErrort   side_effectt   sayt   assertFalset   needs_reconnectR   t
   assertTruet   called(   R   (    (    s<   /tmp/pip-unpacked-wheel-gV1wwp/celery/tests/bin/test_amqp.pyt   test_onecmd_error8   s    c         C@  sC   |  j  t   |  j j d  Wd  QX|  j d |  j j    d  S(   Nt   exits   don't leave!(   t   assertRaisest
   SystemExitR   R   R   R   R   (   R   (    (    s<   /tmp/pip-unpacked-wheel-gV1wwp/celery/tests/bin/test_amqp.pyt	   test_exitA   s    c         C@  s9   t  |  j _ |  j j d  |  j d |  j j    d  S(   Ns   foo bar(   t   TrueR   t   silentt   notet   assertNotInR   R   (   R   (    (    s<   /tmp/pip-unpacked-wheel-gV1wwp/celery/tests/bin/test_amqp.pyt   test_note_silentF   s    c         C@  s0   |  j  j d  t |  j  _ |  j  j d  d  S(   Ns   queue.declare foos   queue.delete foo(   R   R   R7   R/   (   R   (    (    s<   /tmp/pip-unpacked-wheel-gV1wwp/celery/tests/bin/test_amqp.pyt   test_reconnectK   s    c         C@  sQ   |  j  |  j j d  d g  |  j  t |  j j d   t d d g   d  S(   Ns	   queue.decs   queue.declaret   declares   exchange.declare(   R   R   t   completenamest   sorted(   R   (    (    s<   /tmp/pip-unpacked-wheel-gV1wwp/celery/tests/bin/test_amqp.pyt   test_completenamesP   s    
c         C@  se   t    |  j _ t    |  j _ |  j j d  |  j j j   |  j j d  |  j j j d  d  S(   Nt    t   foo(   R   R   t	   emptylinet   defaultR   t   assert_called_with(   R   (    (    s<   /tmp/pip-unpacked-wheel-gV1wwp/celery/tests/bin/test_amqp.pyt   test_empty_lineZ   s    c         C@  s4   |  j  j i d d 6 |  j d |  j j    d  S(   Nt   barRB   (   R   t   respondR   R   R   (   R   (    (    s<   /tmp/pip-unpacked-wheel-gV1wwp/celery/tests/bin/test_amqp.pyt   test_respondb   s    c         C@  s   |  j  |  j j  d  S(   N(   R0   R   t   prompt(   R   (    (    s<   /tmp/pip-unpacked-wheel-gV1wwp/celery/tests/bin/test_amqp.pyt   test_promptf   s    c         C@  sD   |  j  j d  |  j  j d  |  j  j d  |  j  j d  d  S(   Ns   queue.declare foos   exchange.declare bar direct yess   queue.bind foo bar bazs   basic.ack 1(   R   R   (   R   (    (    s<   /tmp/pip-unpacked-wheel-gV1wwp/celery/tests/bin/test_amqp.pyt   test_no_returnsi   s    c         C@  sI   t    } d | _ i d d 6| _ i d d 6| _ |  j t |   d  S(   Ns   the quick brown foxi   t   aRG   t   exchange(   R   t   bodyt
   propertiest   delivery_infoR0   R   (   R   t   m(    (    s<   /tmp/pip-unpacked-wheel-gV1wwp/celery/tests/bin/test_amqp.pyt   test_dump_messageo   s
    		c         C@  s   |  j  d t d    d  S(   Ns   No messages in queue(   R   R   R!   (   R   (    (    s<   /tmp/pip-unpacked-wheel-gV1wwp/celery/tests/bin/test_amqp.pyt   test_dump_message_no_messagev   s    c         C@  s9   t  |  j _ |  j j d  |  j d |  j j    d  S(   Nt   FOO(   R7   R   R8   R9   R:   R   R   (   R   (    (    s<   /tmp/pip-unpacked-wheel-gV1wwp/celery/tests/bin/test_amqp.pyt	   test_notey   s    c         C@  s6   |  j  d  } | j   |  j d |  j j    d  S(   Ns   queue.declare fooR   (   R   t   runR   R   R   (   R   RM   (    (    s<   /tmp/pip-unpacked-wheel-gV1wwp/celery/tests/bin/test_amqp.pyt   test_run~   s    
c         C@  s   |  j    } t   | _ t   } | j _ t   | _ | j   | j j   t   | j _ | j   |  j	 d |  j
 j    d  S(   Nt   bibi(   R   R   t   Shellt   return_valuet   cmdloopRW   RE   t   KeyboardInterruptR,   R   R   R   (   R   RM   R   (    (    s<   /tmp/pip-unpacked-wheel-gV1wwp/celery/tests/bin/test_amqp.pyt   test_run_loop   s    

s   celery.bin.amqp.amqpc         C@  s(   t    } | _ t   | j j   d  S(   N(   R   R[   R   t   execute_from_commandlineRE   (   R   t   Commandt   c(    (    s<   /tmp/pip-unpacked-wheel-gV1wwp/celery/tests/bin/test_amqp.pyt	   test_main   s    s   celery.bin.amqp.AMQPAdminc         C@  s>   t  d |  j  } | j   |  j | j d d |  j  d  S(   NR   i   (   R   R   RW   t   assertIst	   call_args(   R   t   clst   x(    (    s<   /tmp/pip-unpacked-wheel-gV1wwp/celery/tests/bin/test_amqp.pyt   test_command   s    
(   t   __name__t
   __module__R   R   R   R   R   R#   R$   R&   R'   R(   R2   R6   R;   R<   R@   RF   RI   RK   RL   RS   RT   RV   RX   R^   R	   Rb   Rg   (    (    (    s<   /tmp/pip-unpacked-wheel-gV1wwp/celery/tests/bin/test_amqp.pyR
      s4   																
									N(   t
   __future__R    t   celery.bin.amqpR   R   R   R   R   t   celery.tests.caseR   R   R   R	   R
   (    (    (    s<   /tmp/pip-unpacked-wheel-gV1wwp/celery/tests/bin/test_amqp.pyt   <module>   s   ("